Skip to content
Snippets Groups Projects
Verified Commit 3f3a41d4 authored by nd's avatar nd
Browse files

use secrets module instead of random

parent 34e97658
No related branches found
No related tags found
No related merge requests found
import os import os
import secrets
from flask import Flask, redirect, url_for from flask import Flask, redirect, url_for
from werkzeug.routing import IntegerConverter from werkzeug.routing import IntegerConverter
...@@ -16,7 +17,7 @@ def create_app(test_config=None): ...@@ -16,7 +17,7 @@ def create_app(test_config=None):
# set development default config values # set development default config values
app.config.from_mapping( app.config.from_mapping(
TEMPLATES_AUTO_RELOAD=True, TEMPLATES_AUTO_RELOAD=True,
SECRET_KEY=os.urandom(128), SECRET_KEY=secrets.token_hex(128),
SQLALCHEMY_DATABASE_URI="sqlite:///{}".format(os.path.join(app.instance_path, 'uffd.sqlit3')), SQLALCHEMY_DATABASE_URI="sqlite:///{}".format(os.path.join(app.instance_path, 'uffd.sqlit3')),
SQLALCHEMY_ECHO=True, SQLALCHEMY_ECHO=True,
) )
......
import datetime import datetime
import random import secrets
import string import string
import functools import functools
...@@ -38,7 +38,7 @@ def login(): ...@@ -38,7 +38,7 @@ def login():
return redirect(url_for('.login')) return redirect(url_for('.login'))
session['user_uid'] = user.uid session['user_uid'] = user.uid
session['logintime'] = datetime.datetime.now().timestamp() session['logintime'] = datetime.datetime.now().timestamp()
session['_csrf_token'] = ''.join(random.SystemRandom().choice(string.ascii_letters + string.digits) for _ in range(64)) session['_csrf_token'] = secrets.token_hex(128)
return redirect(request.values.get('ref', url_for('index'))) return redirect(request.values.get('ref', url_for('index')))
def get_current_user(): def get_current_user():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment