Browse Source

Disable sessions

merge-requests/1/head
Drew Short 7 years ago
parent
commit
897b7154e3
  1. 3
      server/atheneum/__init__.py
  2. 16
      server/atheneum/utility/session_utility.py
  3. 10
      server/manage.py

3
server/atheneum/__init__.py

@ -6,7 +6,7 @@ from flask import Flask
from flask_migrate import Migrate from flask_migrate import Migrate
from atheneum.db import db from atheneum.db import db
from atheneum.utility import json_utility
from atheneum.utility import json_utility, session_utility
dictConfig({ dictConfig({
'version': 1, 'version': 1,
@ -61,6 +61,7 @@ def create_app(test_config: dict = None) -> Flask:
pass pass
app.json_encoder = json_utility.CustomJSONEncoder app.json_encoder = json_utility.CustomJSONEncoder
app.session_interface = session_utility.DisableSessionInterface()
app.logger.debug('Initializing Application') app.logger.debug('Initializing Application')
db.init_app(app) db.init_app(app)

16
server/atheneum/utility/session_utility.py

@ -0,0 +1,16 @@
"""Session specific utilities."""
from typing import Any
from flask.sessions import SecureCookieSessionInterface
class DisableSessionInterface(SecureCookieSessionInterface):
"""Make sure no session data is transmitted or stored."""
def should_set_cookie(self, app: Any, session: Any) -> bool:
"""Disable default cookie generation."""
return False
def save_session(self, app: Any, session: Any, response: Any) -> Any:
"""Prevent creating session from requests."""
return None

10
server/manage.py

@ -1,8 +1,9 @@
import base64
import logging import logging
import random import random
import string import string
from typing import Optional
from os import path from os import path
from typing import Optional
import click import click
from click import Context from click import Context
@ -111,7 +112,14 @@ def list_users():
[click.echo(user.name) for user in all_users] [click.echo(user.name) for user in all_users]
@click.command(name='base64')
@click.argument('text')
def convert_to_base64(text: str):
print(base64.b64encode(text.encode('utf8')).decode('utf8'))
main.add_command(user_command_group) main.add_command(user_command_group)
main.add_command(convert_to_base64)
user_command_group.add_command(register_user) user_command_group.add_command(register_user)
user_command_group.add_command(register_admin_user) user_command_group.add_command(register_admin_user)
user_command_group.add_command(delete_user) user_command_group.add_command(delete_user)

Loading…
Cancel
Save