webapp/handlers_planet.py

   1 #!/usr/bin/python
   2
   3 import tornado.web
   4
   5 from handlers_base import *
   6
   7 class PlanetBaseHandler(BaseHandler):
   8         pass
   9
  10
  11 class PlanetMainHandler(PlanetBaseHandler):
  12         rss_url = "/rss"
  13
  14         def get(self):
  15                 offset = int(self.get_argument("offset", 0))
  16                 limit = int(self.get_argument("limit", 4))
  17
  18                 entries = self.planet.get_entries(offset=offset, limit=limit)
  19
  20                 self.render("planet/index.html", entries=entries, offset=offset + limit, limit=limit)
  21
  22
  23 class PlanetHotEntriesHandler(PlanetBaseHandler):
  24         def get(self):
  25                 days = self.get_argument("days", None)
  26                 try:
  27                         days = int(days)
  28                 except (TypeError, ValueError):
  29                         days = 30
  30
  31                 entries = self.planet.get_hot_entries(days, limit=25)
  32
  33                 self.render("planet/hottest.html", entries=entries)
  34
  35
  36 class PlanetUserHandler(PlanetBaseHandler):
  37         def get(self, author):
  38                 author = self.accounts.get_by_uid(author)
  39                 if not author:
  40                         raise tornado.web.HTTPError(404, "User is unknown")
  41
  42                 offset = int(self.get_argument("offset", 0))
  43                 limit = int(self.get_argument("limit", 4))
  44
  45                 entries = self.planet.get_entries_by_author(author.uid,
  46                         offset=offset, limit=limit)
  47
  48                 self.render("planet/user.html", author=author, entries=entries,
  49                         offset=offset + limit, limit=limit, rss_url="/user/%s/rss" % author.uid)
  50
  51
  52 class PlanetPostingHandler(PlanetBaseHandler):
  53         def get(self, slug):
  54                 self.entry = self.planet.get_entry_by_slug(slug)
  55
  56                 if not self.entry:
  57                         raise tornado.web.HTTPError(404)
  58
  59                 self.render("planet/posting.html",
  60                         author=self.entry.author, entry=self.entry)
  61
  62         def on_finish(self):
  63                 assert self.entry
  64
  65                 # Get the referer and location for statistical purposes
  66                 referer = self.request.headers.get("Referer", None)
  67                 location = self.get_remote_location()
  68                 if location:
  69                         location = location.country
  70
  71                 self.entry.count_view(referer=referer, location=location)
  72
  73
  74 class PlanetSearchHandler(PlanetBaseHandler):
  75         def get(self):
  76                 query = self.get_argument("q", "")
  77
  78                 if query:
  79                         entries = self.planet.search(query)
  80                 else:
  81                         entries = []
  82
  83                 self.render("planet/search.html", entries=entries, query=query)
  84
  85
  86 class PlanetYearHandler(PlanetBaseHandler):
  87         def get(self, year):
  88                 entries = self.planet.get_entries_by_year(year)
  89
  90                 months = {}
  91                 for entry in entries:
  92                         try:
  93                                 months[entry.month].append(entry)
  94                         except KeyError:
  95                                 months[entry.month] = [entry]
  96
  97                 months = months.items()
  98                 months.sort(reverse=True)
  99
 100                 self.render("planet/year.html", months=months, year=year)