Added plugins
This commit is contained in:
parent
e5af25adf6
commit
6fd05f526e
2 changed files with 61 additions and 0 deletions
46
paramecio2/libraries/plugins.py
Normal file
46
paramecio2/libraries/plugins.py
Normal file
|
|
@ -0,0 +1,46 @@
|
|||
from flask import g
|
||||
from functools import wraps
|
||||
from paramecio2.libraries.db.webmodel import WebModel
|
||||
|
||||
login_name='login'
|
||||
login_url='.login'
|
||||
|
||||
def db(f):
|
||||
|
||||
@wraps(f)
|
||||
|
||||
def wrapper(*args, **kwds):
|
||||
|
||||
g.connection=WebModel.connection()
|
||||
|
||||
try:
|
||||
|
||||
code=f(*args, **kwds)
|
||||
|
||||
g.connection.close()
|
||||
|
||||
except:
|
||||
|
||||
g.connection.close()
|
||||
|
||||
raise
|
||||
|
||||
return code
|
||||
|
||||
return wrapper
|
||||
|
||||
def login_site(f):
|
||||
|
||||
@wraps(f)
|
||||
|
||||
def wrapper(*args, **kwds):
|
||||
|
||||
if not login_name in session:
|
||||
|
||||
return redirect(url_for(login_url))
|
||||
|
||||
else:
|
||||
|
||||
return f(*args, **kwds)
|
||||
|
||||
return wrapper
|
||||
Loading…
Add table
Add a link
Reference in a new issue