root/websites/appspot.com/blog/trunk/blog/blog.py @ 9474

#!-*- coding:utf-8 -*-
import os
import cgi
import logging
import wsgiref.handlers
from google.appengine.ext import db
from google.appengine.ext import webapp
from google.appengine.api import users
from google.appengine.api import mail
from google.appengine.ext.webapp import template

top_path = '/blog/'

blog_info = {
  'title'       : 'Weblog',
  'description' : 'my short stories',
  'admin'       : 'admin',
  'mail'        : 'admin@example.com',
  'link'        : '',
  'root'        : top_path,
  'css'         : '/blog/static/blog.css',
  'rss'         : '',
  'flavour'     : 'html',
  'num_entries' : 5,
  'plugins'     : [],
}

class Entry(db.Model):
  title = db.StringProperty(required=True)
  link = db.StringProperty(required=True)
  author = db.UserProperty(required=True)
  description = db.StringProperty(required=True, multiline=True)
  timestamp = db.DateTimeProperty(required=True, auto_now=True)

def fixPath(path):
  return path.replace("%25", "%").replace("//", "/")

def authUser(path):
  path = fixPath(path)
  blog_info['author'] = users.get_current_user()
  blog_info['login_url'] = users.create_login_url(path)
  blog_info['logout_url'] = users.create_logout_url(path)
  blog_info['is_admin'] = users.is_current_user_admin()

try:
  plugin_path = os.path.join(os.path.dirname(__file__), "plugins")
  for f in os.listdir(plugin_path):
    name, ext = os.path.splitext(f)
    if ext != '.py' or name == '__init__': continue
    path = top_path[1:-1].replace('/', '.')
    try:
      modname = '%s.plugins.%s' % (path, name)
      logging.info('loading %s' % modname)
      m = __import__(modname, globals(), locals(), [''])
      blog_info['plugins'].append(m)
    except Exception, e:
      logging.error(e)
  logging.info('loaded plugins')
except Exception, e:
  logging.error(e)

class MainPage(webapp.RequestHandler):
  def __init__(self):
    self.blog_info = blog_info.copy()
    webapp.RequestHandler.__init__(self)

  def new_entry(self, **kwds):
    return Entry(
        title = kwds.pop('title', None),
        link = kwds.pop('link', None),
        author = kwds.pop('link', users.get_current_user()),
        description = kwds.pop('description', None),
    )

  def fill_entry(self, entry):
    entry.category = top_path[:-1]
    entry.linkid = entry.link
    entry.linkkey = self.blog_info['link'] + entry.linkid[1:]
    entry.link = self.blog_info['link'] + "%s.%s" % (entry.linkid[1:], self.blog_info['flavour'])

  def init_plugins(self):
    if len(self.blog_info['link']) == 0:
      self.blog_info['link'] = "%s://%s%s" % (self.request.scheme, self.request.host, top_path)
    if len(self.blog_info['rss']) == 0:
      self.blog_info['rss'] = "%s://%s%s%s" % (self.request.scheme, self.request.host, top_path, 'index.rss')
    if self.blog_info['css'][0] == '/':
      self.blog_info['css'] = "%s://%s%s" % (self.request.scheme, self.request.host, self.blog_info['css'])
    for plugin in self.blog_info['plugins']:
      try:
        method = getattr(plugin, 'init')
        if method:
          if method(self): logging.info("initialized plugin: %s" % plugin.__name__)
      except Exception, e:
        self.blog_info['plugins'].remove(plugin)

  def call_plugins(self, method, *args):
    ret = 0
    for plugin in self.blog_info['plugins']:
      try:
        func = getattr(plugin, method)
        if func:
          if func(self, *args): ret = ret + 1
      except AttributeError, e:
        pass
    if ret: logging.info("%s: called %d functions" % (method, ret))
    return ret

  def get_template(self, method) :
    return os.path.join(os.path.dirname(__file__),
            "templates", "%s.%s" % (method, self.blog_info['flavour']))

  def get_path_ext(self):
    path, ext = os.path.splitext(fixPath(self.request.path)[len(top_path)-1:])
    if ext: ext = ext[1:]
    return path, ext

  def post(self):
    authUser(self.request.path)
    path_info, flavour = self.get_path_ext()
    if not flavour: flavour = self.blog_info['flavour']
    else: self.blog_info['flavour'] = flavour

    self.init_plugins()
    entry = db.GqlQuery('SELECT * FROM Entry WHERE link = :path_info', path_info=path_info).get()
    self.call_plugins('post', entry)
    self.call_plugins('finish', entry)
    self.redirect(top_path[:-1] + "%s.%s" % (path_info, flavour));

  def get(self):
    authUser(self.request.path)
    path_info, flavour = self.get_path_ext()
    if not flavour: flavour = self.blog_info['flavour']
    else: self.blog_info['flavour'] = flavour

    self.init_plugins()
    template_values = {
      'blog_info' : self.blog_info,
      'path_info' : path_info,
    }
    name = os.path.basename(path_info)
    if name == 'index':
      path_info = path_info[:-len(name)]
    if path_info and path_info[-1:] != '/':
      self.blog_info['method'] = 'entry'
      entry = db.GqlQuery('SELECT * FROM Entry WHERE link = :path_info', path_info=path_info).get()
      if entry: self.fill_entry(entry)
      self.call_plugins('entry', entry)
      template_values['entry'] = entry
    else:
      self.blog_info['method'] = 'entries'
      entries = []
      if self.call_plugins('entries', entries) == 0:
        for entry in db.GqlQuery('SELECT * FROM Entry ORDER BY timestamp DESC'):
          if entry.link[:len(path_info)] == path_info:
            self.fill_entry(entry)
            entries.append(entry)
            if len(entries) >= int(self.blog_info['num_entries']): break
      for entry in entries:
        self.call_plugins('entry', entry)
      template_values['entries'] = entries
    path = os.path.join(os.path.dirname(__file__), self.get_template(self.blog_info['method']))
    content_type = template.render(self.get_template('content_type'), template_values).rstrip('\n')
    self.response.headers['Content-Type'] = content_type
    self.response.out.write(template.render(path, template_values))

def main():
  application = webapp.WSGIApplication([(top_path + '.*', MainPage)], debug=True)
  wsgiref.handlers.CGIHandler().run(application)

if __name__ == '__main__':
  main()