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