mirror of
https://github.com/LukeHagar/pypistats.dev.git
synced 2025-12-06 12:47:48 +00:00
72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
"""Application configuration."""
|
|
import os
|
|
|
|
from celery.schedules import crontab
|
|
from flask import json
|
|
|
|
|
|
def get_db_uri():
|
|
"""Get the database URI."""
|
|
return "postgresql://{username}:{password}@{host}:{port}/{dbname}".format(
|
|
username=os.environ.get("POSTGRESQL_USERNAME"),
|
|
password=os.environ.get("POSTGRESQL_PASSWORD"),
|
|
host=os.environ.get("POSTGRESQL_HOST"),
|
|
port=os.environ.get("POSTGRESQL_PORT"),
|
|
dbname=os.environ.get("POSTGRESQL_DBNAME"),
|
|
)
|
|
|
|
|
|
class Config:
|
|
"""Base configuration."""
|
|
|
|
APP_DIR = os.path.abspath(os.path.dirname(__file__))
|
|
CELERY_BROKER_URL = (os.environ.get("CELERY_BROKER_URL"),)
|
|
BROKER_TRANSPORT_OPTIONS = {"visibility_timeout": 86400}
|
|
CELERY_IMPORTS = "pypistats.tasks.pypi"
|
|
CELERYBEAT_SCHEDULE = {
|
|
"update_db": {"task": "pypistats.tasks.pypi.etl", "schedule": crontab(minute=0, hour=1)} # 1am UTC
|
|
}
|
|
GITHUB_CLIENT_ID = os.environ.get("GITHUB_CLIENT_ID")
|
|
GITHUB_CLIENT_SECRET = os.environ.get("GITHUB_CLIENT_SECRET")
|
|
PROJECT_ROOT = os.path.abspath(os.path.join(APP_DIR, os.pardir))
|
|
SECRET_KEY = os.environ.get("PYPISTATS_SECRET", "secret-key")
|
|
SQLALCHEMY_TRACK_MODIFICATIONS = False
|
|
SQLALCHEMY_DATABASE_URI = get_db_uri()
|
|
|
|
# Plotly chart definitions
|
|
PLOT_BASE = json.load(open(os.path.join(os.path.dirname(__file__), "plots", "plot_base.json")))
|
|
DATA_BASE = json.load(open(os.path.join(os.path.dirname(__file__), "plots", "data_base.json")))
|
|
|
|
|
|
class LocalConfig(Config):
|
|
"""Local configuration."""
|
|
|
|
DEBUG = True
|
|
ENV = "local"
|
|
|
|
|
|
class ProdConfig(Config):
|
|
"""Production configuration."""
|
|
|
|
DEBUG = False
|
|
ENV = "prod"
|
|
|
|
|
|
class DevConfig(Config):
|
|
"""Development configuration."""
|
|
|
|
DEBUG = True
|
|
ENV = "dev"
|
|
|
|
|
|
class TestConfig(Config):
|
|
"""Test configuration."""
|
|
|
|
DEBUG = True
|
|
ENV = "dev"
|
|
TESTING = True
|
|
WTF_CSRF_ENABLED = False # Allows form testing
|
|
|
|
|
|
configs = {"development": DevConfig, "local": LocalConfig, "production": ProdConfig, "test": TestConfig}
|