GreenMine/config.py at master · AndreySolod/GreenMine · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
import os
import secrets


basedir = os.path.abspath(os.path.dirname(__file__))


class DevelopmentConfig:
    SECRET_KEY = os.environ.get('SECRET_KEY') or 'very long secret'
    if os.environ.get('SESSION_COOKIE_SECURE') == None:
        SESSION_COOKIE_SECURE = False
    else:
        SESSION_COOKIE_SECURE = bool(os.environ.get('SESSION_COOKIE_SECURE'))
    SESSION_COOKIE_SAMESITE = os.environ.get('SESSION_COOKIE_SAMESITE') or 'Lax'
    try:
        WTF_CSRF_TIME_LIMIT = int(os.environ.get('WTF_CSRF_TIME_LIMIT')) or 3600
    except (ValueError, TypeError):
        WTF_CSRF_TIME_LIMIT = 3600
    APPLICATION_ROOT = os.environ.get("APPLICATION_ROOT") or "/"
    SQLALCHEMY_DATABASE_URI = os.environ.get('DATABASE_URL') or 'sqlite:///' + os.path.join(basedir, 'app.db')
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    REST_FORBIDDEN_ATTRIBUTES = os.environ.get("REST_FORBIDDEN_ATTRIBUTES").split(",") if os.environ.get("REST_FORBIDDEN_ATTRIBUTES") else ["User.password_hash", "User.token", "User.token_expiration"]
    TOKEN_EXPIRATION = os.environ.get('TOKEN_EXPIRATION') or 365 * 24 * 60 * 60
    CELERY_BROKER_URL = os.environ.get('CELERY_BROKER_URL') or 'redis://localhost:6379/0'
    CELERY_RESULT_BACKEND = os.environ.get('CELERY_RESULT_BACKEND') or 'redis://localhost:6379/0'
    CELERY_TASK_IGNORE_RESULT = False if os.environ.get("CELERY_TASK_IGNORE_RESULT") == 'False' else True
    CSP_ENABLED = False if os.environ.get("CSP_ENABLED") == 'False' else True
    ACTIVATE_PASSWORD_POLICY = False if os.environ.get('ACTIVATE_PASSWORD_POLICY') == 'False' else True
    DEFAULT_LANGUAGE = os.environ.get('DEFAULT_LANGUAGE') or 'ru'
    try:
        USER_ACTION_LOGGING_ON_STDOUT = bool(os.environ.get("USER_ACTION_LOGGING_ON_STDOUT")) or True
    except:
        USER_ACTION_LOGGING_ON_STDOUT = True
    USER_ACTION_LOGGING_FILE = os.environ.get("USER_ACTION_LOGGING_FILE") or "logs/user_action.log"
    try:
        FLASK_LOGGING_ON_STDOUT = bool(os.environ.get("FLASK_LOGGING_ON_STDOUT")) or True
    except:
        FLASK_LOGGING_ON_STDOUT = True
    FLASK_LOGGING_FILE = os.environ.get("FLASK_LOGGING_FILE") or "logs/flask.log"
    ERROR_LOGGING_FILE = os.environ.get("ERROR_LOGGING_FILE") or "logs/error.log"
    METASPLOIT_HOST = os.environ.get("METASPLOIT_HOST") or "127.0.0.1"
    METASPLOIT_PORT = os.environ.get("METASPLOIT_PORT") or "55553"
    METASPLOIT_PASSWORD = os.environ.get("METASPLOIT_PASSWORD") or "secret"
    METASPLOIT_VERIFY_SSL = os.environ.get("METASPLOIT_VERIFY_SSL") == "True"


class ProductionConfig(DevelopmentConfig):
    SECRET_KEY = os.environ.get('SECRET_KEY') or secrets.token_hex()
    if os.environ.get('SESSION_COOKIE_SECURE') == None:
        SESSION_COOKIE_SECURE = False
    else:
        SESSION_COOKIE_SECURE = bool(os.environ.get('SESSION_COOKIE_SECURE'))
    CSP_ENABLED = True
    try:
        CELERY_WORKERS_COUNT = int(os.environ.get('CELERY_WORKERS_COUNT') or 0)
    except (ValueError, TypeError):
        CELERY_WORKERS_COUNT = 0
    USER_ACTION_LOGGING_FILE = os.environ.get("USER_ACTION_LOGGING_FILE")
    FLASK_LOGGING_FILE = os.environ.get("FLASK_LOGGING_FILE")
    ERROR_LOGGING_FILE = os.environ.get("ERROR_LOGGING_FILE")


class TestConfig(DevelopmentConfig):
    SECRET_KEY = 'secret'
    SQLALCHEMY_DATABASE_URI = 'sqlite:///' + os.path.join(basedir, 'test_database.db')
    WTF_CSRF_ENABLED = False