config.py

# coding: utf-8
# Code By DaTi_Co
# Config.py
# App Configuration File
from os import environ, path
import secrets
import ssl
from dotenv import load_dotenv
from generate_service_account_file import generate_file


def _get_int_env(name, default):
    """
    Safely parse an integer environment variable.

    Returns the provided default if the variable is unset, empty, or invalid.
    """
    raw_value = environ.get(name)
    if raw_value is None or raw_value == "":
        return default
    try:
        return int(raw_value)
    except (TypeError, ValueError):
        return default


basedir = path.abspath(path.dirname(__file__))
load_dotenv(path.join(basedir, '.env'))

data = generate_file()


class Config:
    # General Config
    DEBUG = False
    TESTING = False
    SECRET_KEY = environ.get('SECRET_KEY')
    UPLOAD_FOLDER = './static/upload'
    SQLALCHEMY_DATABASE_URI = environ.get('SQLALCHEMY_DATABASE_URI')
    SQLALCHEMY_TRACK_MODIFICATIONS = False
    MQTT_BROKER_URL = environ.get('MQTT_BROKER_URL')
    MQTT_BROKER_PORT = _get_int_env('MQTT_BROKER_PORT', 11868)
    MQTT_USERNAME = environ.get('MQTT_USERNAME')
    MQTT_PASSWORD = environ.get('MQTT_PASSWORD')
    MQTT_KEEPALIVE = 90
    MQTT_TLS_ENABLED = environ.get('MQTT_TLS_ENABLED', 'False').lower() in ('1', 'true', 'yes', 'on')
    MQTT_TLS_VERSION = ssl.PROTOCOL_TLS_CLIENT if MQTT_TLS_ENABLED else None
    API_KEY = environ.get('API_KEY')
    DATABASEURL = environ.get('DATABASEURL')  # your Project database URL
    SERVICE_ACCOUNT_DATA = data


class ProductionConfig(Config):
    DEBUG = False


class DevelopmentConfig(Config):
    DEBUG = True
    SECRET_KEY = secrets.token_urlsafe(16)


class TestingConfig(Config):
    DB_SERVER = 'localhost'
    DEBUG = True
    TESTING = True
    SQLALCHEMY_DATABASE_URI = 'sqlite:///:memory:'
    SQLALCHEMY_TRACK_MODIFICATIONS = False