29 lines
808 B
Python
29 lines
808 B
Python
|
from bottle import redirect, request
|
||
|
|
from db import db
|
||
|
|
from functools import wraps
|
||
|
|
from inspect import Signature
|
||
|
|
from user import User
|
||
|
|
|
||
|
|
|
||
|
|
class SessionPlugin(object):
|
||
|
|
name = 'SessionPlugin'
|
||
|
|
keyword = 'user'
|
||
|
|
api = 2
|
||
|
|
|
||
|
|
def __init__(self, loginpage):
|
||
|
|
self.loginpage = loginpage
|
||
|
|
|
||
|
|
def apply(self, callback, route):
|
||
|
|
if self.keyword in Signature.from_callable(route.callback).parameters:
|
||
|
|
@wraps(callback)
|
||
|
|
def wrapper(*args, **kwargs):
|
||
|
|
uid = request.get_cookie('uid', secret=db.secret)
|
||
|
|
if uid is None:
|
||
|
|
return redirect(self.loginpage)
|
||
|
|
kwargs[self.keyword] = User(db, uid)
|
||
|
|
return callback(*args, **kwargs)
|
||
|
|
|
||
|
|
return wrapper
|
||
|
|
else:
|
||
|
|
return callback
|