79 lines
2.3 KiB
Python
79 lines
2.3 KiB
Python
import os
|
|
from typing import Dict, List, Tuple
|
|
|
|
from flask import Flask
|
|
|
|
from fittrackee import db
|
|
from fittrackee.users.models import User
|
|
|
|
from .models import AppConfig
|
|
|
|
MAX_FILE_SIZE = 1 * 1024 * 1024 # 1MB
|
|
|
|
|
|
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
|
|
|
|
|
|
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
|
|
|
|
|
|
def verify_app_config(config_data: Dict) -> List:
|
|
"""
|
|
Verify if application config is valid.
|
|
|
|
If not, it returns not empty string
|
|
"""
|
|
ret = []
|
|
if (
|
|
'gpx_limit_import' in config_data
|
|
and config_data['gpx_limit_import'] <= 0
|
|
):
|
|
ret.append('Max. files in a zip archive must be greater than 0')
|
|
|
|
if (
|
|
'max_single_file_size' in config_data
|
|
and config_data['max_single_file_size'] <= 0
|
|
):
|
|
ret.append('Max. size of uploaded files must be greater than 0')
|
|
|
|
if (
|
|
'max_zip_file_size' in config_data
|
|
and config_data['max_zip_file_size'] <= 0
|
|
):
|
|
ret.append('Max. size of zip archive must be greater than 0')
|
|
return ret
|