mirror of
https://github.com/edgewall/genshi.git
synced 2026-02-05 15:46:37 +01:00
4f414c0390
This is a Python 2.6+ feature so it breaks compatibility with Python 2.4/2.5 (assuming this library did work with these ancient releases so far). However this is necessary to use a unified source for Python 2 and Python 3.
139 lines
4.3 KiB
Python
Executable File
139 lines
4.3 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import operator, os, pickle, sys
|
|
|
|
import cherrypy
|
|
from formencode import Invalid
|
|
from genshi.input import HTML
|
|
from genshi.filters import HTMLFormFiller, HTMLSanitizer
|
|
|
|
from geddit.form import LinkForm, CommentForm
|
|
from geddit.lib import ajax, template
|
|
from geddit.model import Link, Comment
|
|
|
|
|
|
class Root(object):
|
|
|
|
def __init__(self, data):
|
|
self.data = data
|
|
|
|
@cherrypy.expose
|
|
@template.output('index.html')
|
|
def index(self):
|
|
links = sorted(self.data.values(), key=operator.attrgetter('time'))
|
|
return template.render(links=links)
|
|
|
|
@cherrypy.expose
|
|
@template.output('submit.html')
|
|
def submit(self, cancel=False, **data):
|
|
if cherrypy.request.method == 'POST':
|
|
if cancel:
|
|
raise cherrypy.HTTPRedirect('/')
|
|
form = LinkForm()
|
|
try:
|
|
data = form.to_python(data)
|
|
link = Link(**data)
|
|
self.data[link.id] = link
|
|
raise cherrypy.HTTPRedirect('/')
|
|
except Invalid as e:
|
|
errors = e.unpack_errors()
|
|
else:
|
|
errors = {}
|
|
|
|
return template.render(errors=errors) | HTMLFormFiller(data=data)
|
|
|
|
@cherrypy.expose
|
|
@template.output('info.html')
|
|
def info(self, id):
|
|
link = self.data.get(id)
|
|
if not link:
|
|
raise cherrypy.NotFound()
|
|
return template.render(link=link)
|
|
|
|
@cherrypy.expose
|
|
@template.output('comment.html')
|
|
def comment(self, id, cancel=False, **data):
|
|
link = self.data.get(id)
|
|
if not link:
|
|
raise cherrypy.NotFound()
|
|
if cherrypy.request.method == 'POST':
|
|
if cancel:
|
|
raise cherrypy.HTTPRedirect('/info/%s' % link.id)
|
|
form = CommentForm()
|
|
try:
|
|
data = form.to_python(data)
|
|
markup = HTML(data['content']) | HTMLSanitizer()
|
|
data['content'] = markup.render('xhtml')
|
|
comment = link.add_comment(**data)
|
|
if not ajax.is_xhr():
|
|
raise cherrypy.HTTPRedirect('/info/%s' % link.id)
|
|
return template.render('_comment.html', comment=comment,
|
|
num=len(link.comments))
|
|
except Invalid as e:
|
|
errors = e.unpack_errors()
|
|
else:
|
|
errors = {}
|
|
|
|
if ajax.is_xhr():
|
|
stream = template.render('_form.html', link=link, errors=errors)
|
|
else:
|
|
stream = template.render(link=link, comment=None, errors=errors)
|
|
return stream | HTMLFormFiller(data=data)
|
|
|
|
@cherrypy.expose
|
|
@template.output('index.xml', method='xml')
|
|
def feed(self, id=None):
|
|
if id:
|
|
link = self.data.get(id)
|
|
if not link:
|
|
raise cherrypy.NotFound()
|
|
return template.render('info.xml', link=link)
|
|
else:
|
|
links = sorted(self.data.values(), key=operator.attrgetter('time'))
|
|
return template.render(links=links)
|
|
|
|
|
|
def main(filename):
|
|
# load data from the pickle file, or initialize it to an empty list
|
|
if os.path.exists(filename):
|
|
fileobj = open(filename, 'rb')
|
|
try:
|
|
data = pickle.load(fileobj)
|
|
finally:
|
|
fileobj.close()
|
|
else:
|
|
data = {}
|
|
|
|
def _save_data():
|
|
# save data back to the pickle file
|
|
fileobj = open(filename, 'wb')
|
|
try:
|
|
pickle.dump(data, fileobj)
|
|
finally:
|
|
fileobj.close()
|
|
if hasattr(cherrypy.engine, 'subscribe'): # CherryPy >= 3.1
|
|
cherrypy.engine.subscribe('stop', _save_data)
|
|
else:
|
|
cherrypy.engine.on_stop_engine_list.append(_save_data)
|
|
|
|
# Some global configuration; note that this could be moved into a
|
|
# configuration file
|
|
cherrypy.config.update({
|
|
'tools.encode.on': True, 'tools.encode.encoding': 'utf-8',
|
|
'tools.decode.on': True,
|
|
'tools.trailing_slash.on': True,
|
|
'tools.staticdir.root': os.path.abspath(os.path.dirname(__file__)),
|
|
})
|
|
|
|
cherrypy.quickstart(Root(data), '/', {
|
|
'/media': {
|
|
'tools.staticdir.on': True,
|
|
'tools.staticdir.dir': 'static'
|
|
}
|
|
})
|
|
|
|
if __name__ == '__main__':
|
|
import formencode
|
|
formencode.api.set_stdtranslation(languages=['en'])
|
|
main(sys.argv[1])
|