controller.py

import cgi
import datetime
import os
import logging

from urlModel import UrlModel
from urlModel import UrlController
from urlCleaner import UrlCleaner

from google.appengine.ext import webapp
from google.appengine.ext.webapp import template
from google.appengine.ext.webapp.util import run_wsgi_app

def goToHome(self, error = None):
  lasts = urlController.get_lasts()
  self.response.headers['Content-Type'] = 'text/html'
  path = os.path.join(os.path.dirname(__file__), 'home.html')
  self.response.out.write(template.render(path, { 'lasts': lasts, 'error': error, 'url': self.request.get('url') } ))

class MainPage(webapp.RequestHandler):
  def get(self): 
    goToHome(self)

class SavePage(webapp.RequestHandler):
  def post(self):
    url = self.request.get('url')
    potdemiel = self.request.get('potdemiel')
    if len(potdemiel) != 0: # Very simple protection about spammer's bot who filling all field
      goToHome(self, 'An error occured.')
      return

    if not url or len(url) == 0: 
      goToHome(self, 'Please type an URL.')
      return

    try:
      cleanedUrl = urlCleaner.clean(url)
    except ValueError, err:
      goToHome(self, 'An error occured, URL is malformed : ' + err.args[0])
      return

    shortened_url = urlController.save(cleanedUrl)
    self.redirect('/#' + shortened_url.shortname)

class RedirectPage(webapp.RequestHandler):
  def get(self, shortname):
    shortened_url = urlController.get_by_key(shortname)
    self.redirect(shortened_url.url)

class RssPage(webapp.RequestHandler):
  def get(self):
    lasts = urlController.get_lasts()
    self.response.headers['Content-Type'] = 'application/rss+xml'
    path = os.path.join(os.path.dirname(__file__), 'rss.xml')
    self.response.out.write(template.render(path, { 'lasts': lasts } ))


application = webapp.WSGIApplication([
													('/', MainPage),
													('/add', SavePage),
													('/rss.xml', RssPage),
													(r'/(.*)', RedirectPage)
												])

if __name__ == '__main__':
  logging.getLogger().setLevel(logging.ERROR)
  urlController = UrlController()
  urlCleaner = UrlCleaner()
  run_wsgi_app(application)