134 lines
3.7 KiB
Python
134 lines
3.7 KiB
Python
# -*- Mode: Python; -*-
|
|
# -*- coding: utf-8 -*-
|
|
# vi:si:et:sw=2:sts=2:ts=2
|
|
|
|
from turbogears import controllers, expose, validate, error_handler
|
|
from model import *
|
|
from turbogears import identity, redirect
|
|
from cherrypy import request, response
|
|
# import logging
|
|
# log = logging.getLogger("oilarchive.controllers")
|
|
|
|
# from oilarchive import json
|
|
import oilcache
|
|
from forms import forms
|
|
from sortname import sortname
|
|
|
|
class View:
|
|
@expose(template=".templates.view")
|
|
def view(self, item):
|
|
return dict(item = item)
|
|
|
|
def icon(self, item):
|
|
response.headerMap['Content-Type'] = "image/png"
|
|
return oilcache.icon(item)
|
|
|
|
@expose()
|
|
def default(self, id, *args, **kw):
|
|
try:
|
|
item = ArchiveItem.byHashId(id)
|
|
if not args:
|
|
return self.view(item)
|
|
elif args[0] == 'icon.png':
|
|
return self.icon(item)
|
|
elif args[0] == 'json':
|
|
return item.json
|
|
|
|
except:
|
|
raise redirect("/")
|
|
|
|
class Admin:
|
|
@expose('.templates.admin_index')
|
|
def index(self):
|
|
return dict()
|
|
|
|
@expose('.templates.admin_archives')
|
|
def archives(self, tg_errors=None):
|
|
if tg_errors:
|
|
flash("There was a problem with the form!")
|
|
return dict(
|
|
add_archive = forms.add_archive,
|
|
archives = Archive.select(orderBy = "archiveName"),
|
|
)
|
|
|
|
@expose()
|
|
@validate(form = forms.add_archive)
|
|
@error_handler(archives)
|
|
def archives_add(self, **data):
|
|
new = Archive(
|
|
archiveName = data['archiveName'],
|
|
archiveType = data['archiveType'],
|
|
archiveUrl = data['archiveUrl'],
|
|
ttl = int(data['ttl']),
|
|
)
|
|
raise redirect('archives')
|
|
|
|
@expose('.templates.admin_sortnames')
|
|
def sortnames(self, tg_errors=None):
|
|
if tg_errors:
|
|
flash("There was a problem with the form!")
|
|
q = SortName.select(orderBy="name")
|
|
names = "\n".join([i.name for i in q])
|
|
sortnames_values = dict(names = names)
|
|
return dict(sortnames_values = sortnames_values, sortnames_form = forms.sortnames)
|
|
|
|
@expose()
|
|
@validate(form = forms.sortnames)
|
|
@error_handler(sortnames)
|
|
def sortnames_save(self, **data):
|
|
names = data['names']
|
|
if names:
|
|
for b in SortName.select():
|
|
SortName.delete(b.id)
|
|
names = names.split('\n')
|
|
for name in names:
|
|
SortName(name = name)
|
|
raise redirect('sortnames')
|
|
|
|
class Api:
|
|
@expose()
|
|
def sortname(self, name):
|
|
sname = sortname(name)
|
|
return dict(sortname = sname, name = name)
|
|
|
|
class Root(controllers.RootController):
|
|
view = View()
|
|
admin = Admin()
|
|
api = Api()
|
|
|
|
@expose(template=".templates.welcome")
|
|
# @identity.require(identity.in_group("admin"))
|
|
def index(self):
|
|
import time
|
|
# log.debug("Happy TurboGears Controller Responding For Duty")
|
|
return dict(now=time.ctime())
|
|
|
|
@expose(template=".templates.login")
|
|
def login(self, forward_url=None, previous_url=None, *args, **kw):
|
|
if not identity.current.anonymous \
|
|
and identity.was_login_attempted() \
|
|
and not identity.get_identity_errors():
|
|
raise redirect(forward_url)
|
|
|
|
forward_url=None
|
|
previous_url= request.path
|
|
|
|
if identity.was_login_attempted():
|
|
msg=_("The credentials you supplied were not correct or "
|
|
"did not grant access to this resource.")
|
|
elif identity.get_identity_errors():
|
|
msg=_("You must provide your credentials before accessing "
|
|
"this resource.")
|
|
else:
|
|
msg=_("Please log in.")
|
|
forward_url= request.headers.get("Referer", "/")
|
|
|
|
response.status=403
|
|
return dict(message=msg, previous_url=previous_url, logging_in=True,
|
|
original_parameters=request.params,
|
|
forward_url=forward_url)
|
|
|
|
@expose()
|
|
def logout(self):
|
|
identity.current.logout()
|
|
raise redirect("/")
|