332 lines
12 KiB
Python
Executable file
332 lines
12 KiB
Python
Executable file
#!/usr/bin/env python2.6
|
|
|
|
# add the current directory to the python search path
|
|
import sys
|
|
sys.path.insert(0, "./")
|
|
|
|
import forms
|
|
import sqlobject
|
|
import bobo
|
|
from genshi.template import TemplateLoader
|
|
import genshi.filters
|
|
import genshi
|
|
import formencode
|
|
import uuid
|
|
import datetime
|
|
import os
|
|
|
|
db_filename = "database.sqlite"
|
|
db_filename = os.path.abspath(db_filename)
|
|
database = sqlobject.connectionForURI("sqlite://" + db_filename)
|
|
sqlobject.sqlhub.processConnection = database
|
|
loader = TemplateLoader(os.path.join(os.path.dirname(__file__), 'templates'), auto_reload=True)
|
|
|
|
BASE_DICT = {
|
|
"base_url": "/", # the trailing slash is necessary
|
|
"errors": {},
|
|
}
|
|
|
|
POLL_SETTINGS = {
|
|
"show_all_submissions": (bool, True),
|
|
"show_statistics": (bool, True),
|
|
}
|
|
|
|
class ContentSubmission(sqlobject.SQLObject):
|
|
submitter = sqlobject.StringCol()
|
|
content = sqlobject.StringCol()
|
|
poll_id = sqlobject.ForeignKey("Poll")
|
|
timestamp_creation = sqlobject.DateTimeCol()
|
|
|
|
def get_creation_time_string(self):
|
|
return str(self.timestamp_creation)
|
|
|
|
|
|
class PollSetting(sqlobject.SQLObject):
|
|
poll_id = sqlobject.ForeignKey("Poll")
|
|
key = sqlobject.StringCol()
|
|
value = sqlobject.StringCol()
|
|
|
|
|
|
class Poll(sqlobject.SQLObject):
|
|
author = sqlobject.StringCol()
|
|
hash_key = sqlobject.StringCol()
|
|
admin_hash_key = sqlobject.StringCol()
|
|
title = sqlobject.StringCol()
|
|
description = sqlobject.StringCol()
|
|
timestamp_creation = sqlobject.DateTimeCol()
|
|
|
|
def get_settings(self):
|
|
current_dict = {}
|
|
for setting in PollSetting.selectBy(poll_id=self.id):
|
|
if setting.key in POLL_SETTINGS.keys():
|
|
current_dict[setting.key] = self.validate_poll_setting(setting.key, setting.value)
|
|
for key, meta_info in POLL_SETTINGS.items():
|
|
if not key in current_dict.keys():
|
|
current_dict[key] = meta_info[1]
|
|
return current_dict
|
|
|
|
def change_setting(self, key, value):
|
|
validated_value = self.validate_poll_setting(key, value)
|
|
if not validated_value is None:
|
|
poll_setting = PollSetting.selectBy(poll_id=self.id, key=key)
|
|
if poll_setting.count() == 1:
|
|
poll_setting.value = str(validated_value)
|
|
|
|
def get_num_of_submitters(self):
|
|
all_submitters = [submission.submitter for submission in ContentSubmission.selectBy(poll_id=self.id)]
|
|
unique_submitters = []
|
|
for submitter in all_submitters:
|
|
if not submitter in unique_submitters:
|
|
unique_submitters.append(submitter)
|
|
return len(unique_submitters)
|
|
|
|
def get_num_of_submissions(self):
|
|
return ContentSubmission.selectBy(poll_id=self.id).count()
|
|
|
|
def get_submissions(self):
|
|
return ContentSubmission.selectBy(poll_id=self.id)
|
|
|
|
def delete_poll(self):
|
|
submissions = ContentSubmission.selectBy(poll_id=self.id)
|
|
settings = PollSetting.selectBy(poll_id=self.id)
|
|
for submission in submissions:
|
|
submission.destroySelf()
|
|
for setting in settings:
|
|
setting.destroySelf()
|
|
self.destroySelf()
|
|
|
|
def get_url(self):
|
|
return "%spolls/%s" % (BASE_DICT["base_url"], self.hash_key)
|
|
|
|
def get_submit_url(self):
|
|
return "%spolls/%s/submit" % (BASE_DICT["base_url"], self.hash_key)
|
|
|
|
def get_admin_url(self):
|
|
return "%spolls/%s" % (BASE_DICT["base_url"], self.admin_hash_key)
|
|
|
|
def get_edit_url(self):
|
|
return "%spolls/%s/admin" % (BASE_DICT["base_url"], self.admin_hash_key)
|
|
|
|
def get_delete_url(self):
|
|
return "%spolls/%s/delete" % (BASE_DICT["base_url"], self.admin_hash_key)
|
|
|
|
def get_creation_time_string(self):
|
|
return str(self.timestamp_creation)
|
|
|
|
|
|
def validate_poll_setting(key, value):
|
|
if not key in POLL_SETTINGS.keys():
|
|
return None
|
|
setting_type = POLL_SETTINGS[key][0]
|
|
if setting_type in (basestring, unicode, str):
|
|
return value
|
|
elif setting_type == bool:
|
|
text = value.tolower()
|
|
if text in ("0", "false", "no", "off", "disabled"):
|
|
return False
|
|
elif text in ("1", "true", "yes", "on", "enabled"):
|
|
return True
|
|
else:
|
|
return None
|
|
else:
|
|
# all other types (e.g. int, float, ...)
|
|
try:
|
|
return setting_type(value)
|
|
except ValueError:
|
|
return None
|
|
|
|
def get_default_values(**kwargs):
|
|
value_dict = dict(BASE_DICT)
|
|
for key, value in kwargs.items():
|
|
value_dict[key] = value
|
|
return value_dict
|
|
|
|
def render(filename, input_data=None, **values):
|
|
stream = loader.load(filename).generate(**values)
|
|
if not input_data is None:
|
|
stream |= genshi.filters.HTMLFormFiller(data=input_data)
|
|
return stream.render("html", doctype="html")
|
|
|
|
def get_poll_id(hash_key):
|
|
polls = Poll.selectBy(hash_key=hash_key)
|
|
if polls.count() == 1:
|
|
return polls[0].id
|
|
else:
|
|
return None
|
|
|
|
def get_poll_admin_id(hash_key):
|
|
polls = Poll.selectBy(admin_hash_key=hash_key)
|
|
if polls.count() == 1:
|
|
return polls[0].id
|
|
else:
|
|
return None
|
|
|
|
def get_new_hash_key():
|
|
hash_key = uuid.uuid4().get_hex()
|
|
while (not get_poll_id(hash_key) is None) or (not get_poll_admin_id(hash_key) is None):
|
|
hash_key = uuid.uuid4().get_hex()
|
|
return hash_key
|
|
|
|
@bobo.query('/polls/new')
|
|
@bobo.query('/polls/new/:author/:title/:description')
|
|
def new_poll(submit=None, cancel=None, author=None, title=None, description=None):
|
|
value_dict = get_default_values()
|
|
data = {"author": author, "title": title, "description": description}
|
|
if cancel:
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
elif not submit:
|
|
# show the "new poll" form
|
|
return render("poll_new.html", input_data=data, **value_dict)
|
|
else:
|
|
# create the new poll (if it is valid)
|
|
errors = {}
|
|
try:
|
|
data = forms.PollForm.to_python(data)
|
|
except formencode.Invalid, errors_packed:
|
|
errors = errors_packed.unpack_errors()
|
|
if errors:
|
|
value_dict["errors"] = errors
|
|
return render("poll_new.html", input_data=data, **value_dict)
|
|
else:
|
|
# create the new poll
|
|
hash_key = get_new_hash_key()
|
|
admin_hash_key = get_new_hash_key()
|
|
now = datetime.datetime.now()
|
|
new_poll = Poll(hash_key=hash_key, admin_hash_key=admin_hash_key, timestamp_creation=now, **data)
|
|
return bobo.redirect(new_poll.get_admin_url())
|
|
|
|
@bobo.query('/polls/:hash_key/submit')
|
|
def submit_content(hash_key=None, submitter=None, content=None):
|
|
value_dict = get_default_values()
|
|
data = {"submitter": submitter, "content": content}
|
|
poll_id = get_poll_id(hash_key)
|
|
if not poll_id is None:
|
|
poll = Poll.get(poll_id)
|
|
value_dict["poll"] = poll
|
|
errors = {}
|
|
try:
|
|
data = forms.SubmitForm.to_python(data)
|
|
except formencode.Invalid, errors_packed:
|
|
errors = errors_packed.unpack_errors()
|
|
if errors:
|
|
value_dict["errors"] = errors
|
|
return render("poll_details.html", input_data=data, **value_dict)
|
|
else:
|
|
# create the new submission content
|
|
data["timestamp_creation"] = datetime.datetime.now()
|
|
data["poll_id"] = poll.id
|
|
ContentSubmission(**data)
|
|
# remove "content" for the next input
|
|
del data["content"]
|
|
return render("poll_details.html", input_data=data, **value_dict)
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
|
|
@bobo.query('/polls/:admin_hash_key/delete')
|
|
def delete_poll(admin_hash_key=None):
|
|
admin_poll_id = get_poll_admin_id(admin_hash_key)
|
|
if not admin_poll_id is None:
|
|
poll = Poll.get(admin_poll_id)
|
|
poll.delete_poll()
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
|
|
@bobo.query('/polls/:admin_hash_key/admin')
|
|
def admin_poll(cancel=False, submit=None, admin_hash_key=None, author=None, title=None, description=None, **kwargs):
|
|
value_dict = get_default_values()
|
|
data = {"author": author, "title": title, "description": description}
|
|
poll_id = get_poll_admin_id(admin_hash_key)
|
|
if poll_id is None:
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
poll = Poll.get(poll_id)
|
|
# cancel: return to (non-edit) admin page
|
|
if cancel:
|
|
return bobo.redirect(poll.get_admin_url())
|
|
if author is None:
|
|
data["author"] = poll.author
|
|
if title is None:
|
|
data["title"] = poll.title
|
|
if description is None:
|
|
data["description"] = poll.description
|
|
settings = poll.get_settings()
|
|
# override with the given settings (taken from the form input with the prefix "setting_")
|
|
for setting_key, setting_value in kwargs.items():
|
|
if setting_key.startswith("setting_"):
|
|
real_key = setting_key[len("setting_"):]
|
|
if real_key in POLL_SETTINGS.keys():
|
|
valid_value = validate_poll_setting(real_key, setting_value)
|
|
if not valid_value is None:
|
|
settings[real_key] = valid_value
|
|
# store the new settings or create the new poll
|
|
errors = {}
|
|
if submit:
|
|
# check for errors only if the content is submitted (not just rendered)
|
|
try:
|
|
data = forms.PollForm.to_python(data)
|
|
except formencode.Invalid, errors_packed:
|
|
errors = errors_packed.unpack_errors()
|
|
# add "settings" after forms validation - since there is no destination type
|
|
data["settings"] = settings
|
|
# the admin hash should also not be validated - thus we may not add it before
|
|
if errors:
|
|
value_dict["errors"] = errors
|
|
return render("poll_admin.html", input_data=data, **value_dict)
|
|
else:
|
|
if submit:
|
|
# update core attributes of the existing poll
|
|
poll.author = data["author"]
|
|
poll.title = data["title"]
|
|
poll.description = data["description"]
|
|
current_settings = poll.get_settings()
|
|
# update settings
|
|
for key, value in settings.items():
|
|
if current_settings[key] != value:
|
|
poll.change_setting(key, value)
|
|
return bobo.redirect(poll.get_admin_url())
|
|
else:
|
|
return render("poll_admin.html", input_data=data, **value_dict)
|
|
|
|
@bobo.query('')
|
|
def base():
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
|
|
@bobo.query('/polls')
|
|
def base_polls():
|
|
return bobo.redirect(BASE_DICT["base_url"] + "polls/")
|
|
|
|
@bobo.query('/')
|
|
@bobo.query('/polls/')
|
|
def show_polls():
|
|
value_dict = get_default_values()
|
|
value_dict["polls"] = Poll.select()
|
|
return render("polls.html", **value_dict)
|
|
|
|
@bobo.query('/polls/:poll_hash')
|
|
def show_one_poll(poll_hash=None):
|
|
value_dict = get_default_values()
|
|
poll_id = get_poll_id(poll_hash)
|
|
if not poll_id is None:
|
|
value_dict["poll"] = Poll.get(poll_id)
|
|
return render("poll_details.html", **value_dict)
|
|
else:
|
|
admin_poll_id = get_poll_admin_id(poll_hash)
|
|
if not admin_poll_id is None:
|
|
value_dict["poll"] = Poll.get(admin_poll_id)
|
|
return render("poll_admin_details.html", **value_dict)
|
|
else:
|
|
return bobo.redirect(BASE_DICT["base_url"])
|
|
|
|
@bobo.query('/static/:pagename')
|
|
def show_static(pagename=None):
|
|
value_dict = get_default_values()
|
|
return render(pagename, **value_dict)
|
|
|
|
for table in (Poll, ContentSubmission, PollSetting):
|
|
#Poll.dropTable()
|
|
if not table.tableExists():
|
|
table.createTable()
|
|
for poll in Poll.select():
|
|
print poll
|
|
|
|
# this line allows to use wortschlucker with mod_wsgi
|
|
# see: http://groups.google.com/group/bobo-web/msg/2ba55fc381658cd1
|
|
#application = bobo.Application(bobo_resources=__name__)
|
|
|