Merge pull request #122 from dod-ccpo/flask-session

Use Flask-Sessions for session management
This commit is contained in:
richard-dds 2018-08-03 10:19:39 -04:00 committed by GitHub
commit 394fb57353
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 14 additions and 1 deletions

View File

@ -1,7 +1,9 @@
import os
import re
from configparser import ConfigParser
from flask import Flask, request, g
from flask import Flask, request, g, session
from flask_session import Session
import redis
from unipath import Path
from atst.database import db
@ -24,11 +26,15 @@ def make_app(config):
template_folder=parent_dir.child("templates").absolute(),
static_folder=parent_dir.child("static").absolute(),
)
redis = make_redis(config)
app.config.update(config)
app.config.update({"SESSION_REDIS": redis})
make_flask_callbacks(app)
db.init_app(app)
Session(app)
assets_environment.init_app(app)
app.register_blueprint(bp)
@ -106,3 +112,6 @@ def make_config():
config.set("default", "DATABASE_URI", database_uri)
return map_config(config)
def make_redis(config):
return redis.Redis.from_url(config['REDIS_URI'])

View File

@ -16,3 +16,7 @@ PGPORT = 5432
PGUSER = postgres
PGPASSWORD = postgres
PGDATABASE = atat
SESSION_TYPE = "redis"
SESSION_COOKIE_DOMAIN= "atat.codes"
SESSION_COOKIE_SECURE = True
SESSION_USE_SIGNER = True