import difflib
from pprint import pformat
from pylons import c, request
from tg import expose, redirect
from pyforge.app import Application
from pyforge.lib.dispatch import _dispatch
from helloforge import model as M
from helloforge import version
class HelloForgeApp(Application):
'''This is the HelloWorld application for PyForge, showing
all the rich, creamy goodness that is installable apps.
'''
__version__ = version.__version__
default_config = dict(project_name='NoProject',
message='Custom message goes here')
def __init__(self, config):
self.config = config
self.root = RootController()
def install(self, project):
self.config.config['project_name'] = project._id
self.uninstall(project)
p = M.Page.upsert('Root')
p.text = 'This is the root page.'
p.m.save()
def uninstall(self, project):
M.Page.m.remove(dict(project_id=c.project._id))
M.Comment.m.remove(dict(project_id=c.project._id))
class RootController(object):
@expose('helloforge.templates.index')
def index(self):
return dict(message=c.app.config.config['message'])
def _dispatch(self, state, remainder):
return _dispatch(self, state, remainder)
def _lookup(self, pname, *remainder):
return PageController(pname), remainder
class PageController(object):
def __init__(self, title):
self.title = title
self.comments = CommentController(self.title)
def page(self, version=None):
if version is None:
return M.Page.upsert(self.title)
else:
return M.Page.upsert(self.title, version=int(version))
@expose('helloforge.templates.page_view')
def index(self, version=None):
return dict(page=self.page(version))
@expose('helloforge.templates.page_edit')
def edit(self):
return dict(page=self.page())
@expose('helloforge.templates.page_history')
def history(self):
pages = M.Page.history(self.title)
return dict(title=self.title, pages=pages)
@expose('helloforge.templates.page_diff')
def diff(self, v1, v2):
p1 = self.page(int(v1))
p2 = self.page(int(v2))
diff=difflib.unified_diff(
p1.text.split('\n'),
p2.text.split('\n'))
return dict(p1=p1, p2=p2, diff=diff)
@expose(content_type='text/plain')
def raw(self):
return pformat(self.page())
@expose()
def revert(self, version):
orig = self.page(version)
current = self.page()
current.text = orig.text
current.m.save()
redirect('.')
@expose()
def update(self, text):
page = self.page()
page.text = text
page.m.save()
redirect('.')
class CommentController(object):
def __init__(self, page_title, comment_id=None):
self.page_title = page_title
self.comment_id = comment_id
def page(self):
return M.Page.upsert(self.page_title)
def comment(self):
return M.Comment.m.get(_id=self.comment_id)
@expose()
def reply(self, text):
if self.comment_id:
c = self.comment().reply()
c.text = text
else:
c = self.page().reply()
c.text = text
c.m.save()
redirect(request.referer)
@expose()
def delete(self):
self.comment().m.delete()
redirect(request.referer)
def _dispatch(self, state, remainder):
return _dispatch(self, state, remainder)
def _lookup(self, next, *remainder):
if self.comment_id:
return CommentController(
self.page_title,
self.comment_id + '/' + next), remainder
else:
return CommentController(
self.page_title, next), remainder