Create a global webapp database connection and create a config class.

[ipfire.org.git] / www / webapp / ui_modules.py

#!/usr/bin/python

import markdown
import tornado.web

from tornado.database import Row

class UIModule(tornado.web.UIModule):
	def render_string(self, *args, **kwargs):
		kwargs.update({
			"link" : self.handler.link,
		})
		return tornado.web.UIModule.render_string(self, *args, **kwargs)

	@property
	def user_db(self):
		return self.handler.application.user_db


class MenuItemModule(UIModule):
	def render(self, item):
		if self.request.uri.endswith(item.uri):
			item.active = True

		if not item.uri.startswith("http://"):
			item.uri = "/%s%s" % (self.locale.code[:2], item.uri,)

		if type(item.name) == type({}):
			item.name = item.name[self.locale.code[:2]]

		return self.render_string("modules/menu-item.html", item=item)


class MenuModule(UIModule):
	def render(self):
		menuclass = self.handler.application.ds.menu
		host = self.request.host.lower().split(':')[0]

		return self.render_string("modules/menu.html", menuitems=menuclass.get(host))


class NewsItemModule(UIModule):
	def render(self, item):
		item = Row(item.copy())
		for attr in ("subject", "content"):
			if type(item[attr]) != type({}):
				continue
			item[attr] = item[attr][self.locale.code[:2]]

		return self.render_string("modules/news-item.html", item=item)


#class SidebarModule(UIModule):
#	def render(self, sidebar):
#		return self.render_string("modules/sidebar.html", items=sidebar.items)


class SidebarItemModule(UIModule):
	def render(self):
		return self.render_string("modules/sidebar-item.html")


class SidebarReleaseModule(UIModule):
	def render(self):
		return self.render_string("modules/sidebar-release.html",
			releases=self.handler.application.ds.releases)


class ReleaseItemModule(UIModule):
	def render(self, item):
		return self.render_string("modules/release-item.html", item=item)


class SidebarBannerModule(UIModule):
	def render(self, item):
		return self.render_string("modules/sidebar-banner.html", item=item)


class BuildModule(UIModule):
	def render(self, build):
		return self.render_string("modules/builds.html", build=build)


class PlanetEntryModule(UIModule):
	def render(self, entry):
		if not getattr(entry, "author", None):
			entry.author = self.user_db.get_user_by_id(entry.author_id)

		entry.markup = markdown.markdown(entry.text)
		entry.published = entry.published.strftime("%Y-%m-%d")
		entry.updated = entry.updated.strftime("%Y-%m-%d %H:%M")

		return self.render_string("modules/planet-entry.html", entry=entry)