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                 years = self.planet.get_years()
  20
  21                 self.render("planet/index.html", entries=entries, years=years,
  22                         offset=offset + limit, limit=limit)
  23
  24
  25 class PlanetUserHandler(PlanetBaseHandler):
  26         def get(self, author):
  27                 author = self.accounts.search(author)
  28                 if not author:
  29                         raise tornado.web.HTTPError(404, "User is unknown")
  30
  31                 offset = int(self.get_argument("offset", 0))
  32                 limit = int(self.get_argument("limit", 4))
  33
  34                 entries = self.planet.get_entries_by_author(author.uid,
  35                         offset=offset, limit=limit)
  36
  37                 self.render("planet/user.html", author=author, entries=entries,
  38                         offset=offset + limit, limit=limit, rss_url="/user/%s/rss" % author.uid)
  39
  40
  41 class PlanetPostingHandler(PlanetBaseHandler):
  42         def get(self, slug):
  43                 entry = self.planet.get_entry_by_slug(slug)
  44
  45                 if not entry:
  46                         raise tornado.web.HTTPError(404)
  47
  48                 self.render("planet/posting.html",
  49                         author=entry.author, entry=entry)
  50
  51
  52 class PlanetSearchHandler(PlanetBaseHandler):
  53         def get(self):
  54                 query = self.get_argument("q", "")
  55
  56                 if query:
  57                         entries = self.planet.search(query)
  58                 else:
  59                         entries = []
  60
  61                 self.render("planet/search.html", entries=entries, query=query)
  62
  63
  64 class PlanetYearHandler(PlanetBaseHandler):
  65         def get(self, year):
  66                 entries = self.planet.get_entries_by_year(year)
  67
  68                 months = {}
  69                 for entry in entries:
  70                         try:
  71                                 months[entry.month].append(entry)
  72                         except KeyError:
  73                                 months[entry.month] = [entry]
  74
  75                 months = months.items()
  76                 months.sort(reverse=True)
  77
  78                 self.render("planet/year.html", months=months, year=year)
  79
  80
  81 class PlanetAPISearchAutocomplete(PlanetBaseHandler):
  82         def get(self):
  83                 query = self.get_argument("q")
  84                 if not query:
  85                         raise tornado.web.HTTPError(400)
  86
  87                 results = self.planet.search_autocomplete(query)
  88
  89                 res = {
  90                         "query"   : query,
  91                         "results" : results,
  92                 }
  93                 self.write(res)