]>
Commit | Line | Data |
---|---|---|
1 | #!/usr/bin/python | |
2 | ||
3 | from __future__ import division | |
4 | ||
5 | import hashlib | |
6 | import logging | |
7 | import operator | |
8 | import socket | |
9 | import textile | |
10 | import tornado.escape | |
11 | import tornado.web | |
12 | ||
13 | from tornado.database import Row | |
14 | ||
15 | import backend | |
16 | import backend.stasy | |
17 | ||
18 | class UIModule(tornado.web.UIModule): | |
19 | @property | |
20 | def accounts(self): | |
21 | return self.handler.accounts | |
22 | ||
23 | @property | |
24 | def banners(self): | |
25 | return self.handler.banners | |
26 | ||
27 | @property | |
28 | def memcached(self): | |
29 | return self.handler.memcached | |
30 | ||
31 | @property | |
32 | def releases(self): | |
33 | return self.handler.releases | |
34 | ||
35 | ||
36 | class MenuModule(UIModule): | |
37 | def render(self): | |
38 | hostname = self.request.host.lower().split(':')[0] | |
39 | ||
40 | menuitems = [] | |
41 | for m in backend.Menu().get(hostname): | |
42 | m.active = False | |
43 | ||
44 | if m.uri and self.request.uri.endswith(m.uri): | |
45 | m.active = True | |
46 | ||
47 | # Translate the description of the link | |
48 | m.description = \ | |
49 | self.locale.translate(m.description) | |
50 | m.description = tornado.escape.xhtml_escape(m.description) | |
51 | ||
52 | menuitems.append(m) | |
53 | ||
54 | return self.render_string("modules/menu.html", menuitems=menuitems) | |
55 | ||
56 | ||
57 | class NewsItemModule(UIModule): | |
58 | def get_author(self, author): | |
59 | # Get name of author | |
60 | author = self.accounts.find(author) | |
61 | if author: | |
62 | return author.cn | |
63 | else: | |
64 | _ = self.locale.translate | |
65 | return _("Unknown author") | |
66 | ||
67 | def render(self, item, uncut=False): | |
68 | # Get author | |
69 | item.author = self.get_author(item.author_id) | |
70 | ||
71 | if not uncut and len(item.text) >= 400: | |
72 | item.text = item.text[:400] + "..." | |
73 | ||
74 | # Render text | |
75 | text_id = "news-%s" % hashlib.md5(item.text.encode("utf-8")).hexdigest() | |
76 | ||
77 | text = self.memcached.get(text_id) | |
78 | if not text: | |
79 | text = textile.textile(item.text) | |
80 | self.memcached.set(text_id, text, 60) | |
81 | ||
82 | item.text = text | |
83 | ||
84 | return self.render_string("modules/news-item.html", item=item, uncut=uncut) | |
85 | ||
86 | ||
87 | class NewsLineModule(NewsItemModule): | |
88 | def render(self, item): | |
89 | return self.render_string("modules/news-line.html", item=item) | |
90 | ||
91 | ||
92 | class MirrorItemModule(UIModule): | |
93 | def render(self, item): | |
94 | return self.render_string("modules/mirror-item.html", item=item) | |
95 | ||
96 | ||
97 | class SidebarItemModule(UIModule): | |
98 | def render(self): | |
99 | return self.render_string("modules/sidebar-item.html") | |
100 | ||
101 | ||
102 | class SidebarReleaseModule(UIModule): | |
103 | def render(self): | |
104 | return self.render_string("modules/sidebar-release.html", | |
105 | latest=self.releases.get_latest()) | |
106 | ||
107 | ||
108 | class ReleaseItemModule(UIModule): | |
109 | def render(self, item): | |
110 | return self.render_string("modules/release-item.html", release=item) | |
111 | ||
112 | ||
113 | class SidebarBannerModule(UIModule): | |
114 | def render(self, item=None): | |
115 | if not item: | |
116 | item = self.banners.get_random() | |
117 | ||
118 | return self.render_string("modules/sidebar-banner.html", item=item) | |
119 | ||
120 | ||
121 | class PlanetEntryModule(UIModule): | |
122 | def render(self, entry, short=False): | |
123 | return self.render_string("modules/planet-entry.html", | |
124 | entry=entry, short=short) | |
125 | ||
126 | ||
127 | class TrackerPeerListModule(UIModule): | |
128 | def render(self, peers, percentages=False): | |
129 | # Guess country code and hostname of the host | |
130 | for peer in peers: | |
131 | country_code = backend.GeoIP().get_country(peer["ip"]) | |
132 | peer["country_code"] = country_code or "unknown" | |
133 | ||
134 | try: | |
135 | peer["hostname"] = socket.gethostbyaddr(peer["ip"])[0] | |
136 | except: | |
137 | peer["hostname"] = "" | |
138 | ||
139 | return self.render_string("modules/tracker-peerlist.html", | |
140 | peers=[Row(p) for p in peers], percentages=percentages) | |
141 | ||
142 | ||
143 | class StasyTableModule(UIModule): | |
144 | def render(self, items, sortby="key", reverse=False): | |
145 | hundred_percent = 0 | |
146 | for v in items.values(): | |
147 | hundred_percent += v | |
148 | ||
149 | keys = [] | |
150 | if sortby == "key": | |
151 | keys = sorted(items.keys(), reverse=reverse) | |
152 | elif sortby == "percentage": | |
153 | keys = [k for k,v in sorted(items.items(), key=operator.itemgetter(1))] | |
154 | if not reverse: | |
155 | keys = reversed(keys) | |
156 | else: | |
157 | raise Exception, "Unknown sortby parameter was provided" | |
158 | ||
159 | if hundred_percent: | |
160 | _items = [] | |
161 | for k in keys: | |
162 | v = items[k] * 100 / hundred_percent | |
163 | _items.append((k, v)) | |
164 | items = _items | |
165 | ||
166 | return self.render_string("modules/stasy-table.html", items=items) | |
167 | ||
168 | ||
169 | class StasyDeviceTableModule(UIModule): | |
170 | def render(self, devices): | |
171 | groups = {} | |
172 | ||
173 | for device in devices: | |
174 | if not groups.has_key(device.cls): | |
175 | groups[device.cls] = [] | |
176 | ||
177 | groups[device.cls].append(device) | |
178 | ||
179 | return self.render_string("modules/stasy-table-devices.html", | |
180 | groups=groups.items()) |