FitTrackee/fittrackee/application/utils.py

51 lines
1.5 KiB
Python
Raw Normal View History

import os
2021-01-02 19:28:03 +01:00
from typing import Tuple
from fittrackee import db
from fittrackee.users.models import User
2021-01-02 19:28:03 +01:00
from flask import Flask
from .models import AppConfig
MAX_FILE_SIZE = 1 * 1024 * 1024 # 1MB
2021-01-02 19:28:03 +01:00
def init_config() -> Tuple[bool, AppConfig]:
"""
init application configuration if not existing in database
Note: get some configuration values from env variables
(for FitTrackee versions prior to v0.3.0)
"""
existing_config = AppConfig.query.one_or_none()
nb_users = User.query.count()
if not existing_config:
config = AppConfig()
config.max_users = (
nb_users
if os.getenv('REACT_APP_ALLOW_REGISTRATION') == "false"
else 0
)
config.max_single_file_size = os.environ.get(
'REACT_APP_MAX_SINGLE_FILE_SIZE', MAX_FILE_SIZE
)
config.max_zip_file_size = os.environ.get(
'REACT_APP_MAX_ZIP_FILE_SIZE', MAX_FILE_SIZE * 10
)
db.session.add(config)
db.session.commit()
return True, config
return False, existing_config
2021-01-02 19:28:03 +01:00
def update_app_config_from_database(
current_app: Flask, db_config: AppConfig
) -> None:
current_app.config['gpx_limit_import'] = db_config.gpx_limit_import
current_app.config['max_single_file_size'] = db_config.max_single_file_size
current_app.config['MAX_CONTENT_LENGTH'] = db_config.max_zip_file_size
current_app.config['max_users'] = db_config.max_users
current_app.config[
'is_registration_enabled'
] = db_config.is_registration_enabled