mirror of
https://github.com/cwinfo/powerdns-admin.git
synced 2024-12-04 19:15:30 +00:00
Created a new model to represent the Flask-Session storage schema sessions
with a method for removing expired sessions.
Added a trigger for the Flask-Session model's session clean-up method to the `before_request` handler of the user router.
This commit is contained in:
parent
45f1ba7b82
commit
4442577b0b
39
powerdnsadmin/models/sessions.py
Normal file
39
powerdnsadmin/models/sessions.py
Normal file
@ -0,0 +1,39 @@
|
|||||||
|
from flask import current_app, session
|
||||||
|
from flask_login import current_user
|
||||||
|
from .base import db
|
||||||
|
|
||||||
|
|
||||||
|
class Sessions(db.Model):
|
||||||
|
id = db.Column(db.Integer, primary_key=True)
|
||||||
|
session_id = db.Column(db.String(255), index=True, unique=True)
|
||||||
|
data = db.Column(db.BLOB)
|
||||||
|
expiry = db.Column(db.DateTime)
|
||||||
|
|
||||||
|
def __init__(self,
|
||||||
|
id=None,
|
||||||
|
session_id=None,
|
||||||
|
data=None,
|
||||||
|
expiry=None):
|
||||||
|
self.id = id
|
||||||
|
self.session_id = session_id
|
||||||
|
self.data = data
|
||||||
|
self.expiry = expiry
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
return '<Sessions {0}>'.format(self.id)
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def clean_up_expired_sessions():
|
||||||
|
"""Clean up expired sessions in the database"""
|
||||||
|
from datetime import datetime
|
||||||
|
from sqlalchemy import or_
|
||||||
|
from sqlalchemy.exc import SQLAlchemyError
|
||||||
|
|
||||||
|
try:
|
||||||
|
db.session.query(Sessions).filter(or_(Sessions.expiry < datetime.now(), Sessions.expiry is None)).delete()
|
||||||
|
db.session.commit()
|
||||||
|
except SQLAlchemyError as e:
|
||||||
|
db.session.rollback()
|
||||||
|
current_app.logger.error(e)
|
||||||
|
return False
|
||||||
|
return True
|
@ -37,6 +37,11 @@ def before_request():
|
|||||||
minutes=int(Setting().get('session_timeout')))
|
minutes=int(Setting().get('session_timeout')))
|
||||||
session.modified = True
|
session.modified = True
|
||||||
|
|
||||||
|
# Clean up expired sessions in the database
|
||||||
|
if Setting().get('session_type') == 'sqlalchemy':
|
||||||
|
from ..models.sessions import Sessions
|
||||||
|
Sessions().clean_up_expired_sessions()
|
||||||
|
|
||||||
|
|
||||||
@user_bp.route('/profile', methods=['GET', 'POST'])
|
@user_bp.route('/profile', methods=['GET', 'POST'])
|
||||||
@login_required
|
@login_required
|
||||||
|
Loading…
Reference in New Issue
Block a user