config.py

import datetime
from dotenv import load_dotenv
import os
from typing import List, Set

load_dotenv()


def getenv_bool(key, default=False) -> bool:
    """Retrieve boolean value from environment variable."""
    value = os.getenv(key, None)
    if value:
        if value.lower() in ("false", "0", "off"):
            return False
        else:
            return True
    return default


def getenv_list(key) -> List[str]:
    """Retrieve list of strings from comma-separated environment variable."""
    value = os.getenv(key, None)
    if value:
        return [part.strip().lower() for part in value.split(",")]

    return list()


def load_lines_from_file(filename: str) -> List[str]:
    """Loads lines from file into a list of strings.

    :param filename: Filename without path or extension (assumes app root
     location and cfg extension)
    :return: List of strings as they appear in file
    """
    with open(f"{os.path.join(BASE_PATH, filename)}.cfg", "r") as f:
        return [line.strip() for line in f.read().strip().splitlines()]


# Check if running on production server or local development
is_debug_server = getenv_bool("IS_DEBUG_SERVER", True)

SECRET_KEY = os.getenv("SECRET_KEY", "crabs are far superior to lobsters")
BASE_PATH = os.path.dirname(os.path.abspath(__file__))
MOLT_CHAR_LIMIT: int = 280
MOLTS_PER_PAGE: int = 20
NOTIFS_PER_PAGE: int = 20
MINUTES_EDITABLE: int = 5
MUTED_WORDS_CHAR_LIMIT: int = 2048
UPLOAD_FOLDER: str = os.path.join(BASE_PATH, "static/img/user_uploads")
ALLOWED_EXTENSIONS: Set[str] = {"png", "jpg", "jpeg"}
# Users suggested on post-signup page
RECOMMENDED_USERS: List[str] = load_lines_from_file("recommended_users")
BASE_URL = "http://localhost" if is_debug_server else "https://crabber.net"
# Timestamp of when the server went up
SERVER_START = round(datetime.datetime.utcnow().timestamp())
FEATURED_MOLT_ID = int(os.getenv("FEATURED_MOLT_ID") or "1")
FEATURED_CRAB_USERNAME = os.getenv("FEATURED_CRAB_USERNAME", "jake")
BLACKLIST_IP = getenv_list("BLACKLIST_IP")
BLACKLIST_POST_CODE = getenv_list("BLACKLIST_POST_CODE")
BLACKLIST_CITY_ID = getenv_list("BLACKLIST_CITY_ID")

ADMINS: List[str] = getenv_list("ADMINS")
MODERATORS: List[str] = getenv_list("MODERATORS")

SPRITE_URL = os.getenv("SPRITE_URL", "/static/img/sprites.svg")

DEBUG_QUERIES = getenv_bool("DEBUG_QUERIES")
DATABASE_PATH = os.getenv("CRABBER_DATABASE") or "sqlite:///CRABBER_DATABASE.db"

GEO_PATH = os.path.join(BASE_PATH, "GeoLite2-City.mmdb")
GEO_ENABLED = os.path.exists(GEO_PATH) and getenv_bool("GEO_ENABLED", True)

MAIL_ADDRESS = os.getenv("MAIL_ADDRESS")
MAIL_PASSWORD = os.getenv("MAIL_PASSWORD")
MAIL_ENABLED = MAIL_ADDRESS and MAIL_PASSWORD and getenv_bool("MAIL_ENABLED", False)

CDN_ENABLED = getenv_bool("CDN_ENABLED", False)
CDN_ACCESS_KEY = os.getenv("CDN_ACCESS_KEY")
CDN_SECRET_KEY = os.getenv("CDN_SECRET_KEY")
CDN_ENDPOINT = os.getenv("CDN_ENDPOINT")
CDN_SPACE_NAME = os.getenv("CDN_SPACE_NAME")

SITE_RATE_LIMIT_MINUTE = 200
SITE_RATE_LIMIT_SECOND = 10

API_RATE_LIMIT_SECOND = 2
API_RATE_LIMIT_MINUTE = 20
API_RATE_LIMIT_HOUR = 1000

API_DEFAULT_CRAB_LIMIT = 10
API_MAX_CRAB_LIMIT = 50
API_DEFAULT_MOLT_LIMIT = 10
API_MAX_MOLT_LIMIT = 50
API_MAX_DEVELOPER_KEYS = 5
API_MAX_ACCESS_TOKENS = 5

RSS_MOLT_LIMIT = 50

HCAPTCHA_ENABLED = getenv_bool("HCAPTCHA_ENABLED", False)
REGISTRATION_ENABLED = getenv_bool("REGISTRATION_ENABLED", True)

LIMITS = {
    "age": 32,
    "pronouns": 64,
    "quote": 256,
    "jam": 256,
    "obsession": 256,
    "remember": 256,
    "emoji": 32,
    "description": 512,
    "website": 512,
    "location": 128,
    "display_name": 64,
    "social": 128,
}