cleanup (#744)

Author: nora-codecov

codecov/settings_base.py

@@ -1,21 +1,18 @@
 import os
-from urllib.parse import urlparse
 
-import django_prometheus
 import sentry_sdk
 from corsheaders.defaults import default_headers
 from sentry_sdk.integrations.celery import CeleryIntegration
 from sentry_sdk.integrations.django import DjangoIntegration
 from sentry_sdk.integrations.httpx import HttpxIntegration
 from sentry_sdk.integrations.redis import RedisIntegration
 from sentry_sdk.scrubber import DEFAULT_DENYLIST, EventScrubber
+from shared.django_apps.db_settings import *
 
 from utils.config import SettingsModule, get_config, get_settings_module
 
 SECRET_KEY = get_config("django", "secret_key", default="*")
 
-AUTH_USER_MODEL = "codecov_auth.User"
-
 # Application definition
 
 INSTALLED_APPS = [
@@ -97,171 +94,10 @@
 # Database
 # https://docs.djangoproject.com/en/2.1/ref/settings/#databases
 
-db_url = get_config("services", "database_url")
-if db_url:
-    db_conf = urlparse(db_url)
-    DATABASE_USER = db_conf.username
-    DATABASE_NAME = db_conf.path.replace("/", "")
-    DATABASE_PASSWORD = db_conf.password
-    DATABASE_HOST = db_conf.hostname
-    DATABASE_PORT = db_conf.port
-else:
-    DATABASE_USER = get_config("services", "database", "username", default="postgres")
-    DATABASE_NAME = get_config("services", "database", "name", default="postgres")
-    DATABASE_PASSWORD = get_config(
-        "services", "database", "password", default="postgres"
-    )
-    DATABASE_HOST = get_config("services", "database", "host", default="postgres")
-    DATABASE_PORT = get_config("services", "database", "port", default=5432)
-
-DATABASE_READ_REPLICA_ENABLED = get_config(
-    "setup", "database", "read_replica_enabled", default=False
-)
-
-db_read_url = get_config("services", "database_read_url")
-if db_read_url:
-    db_conf = urlparse(db_read_url)
-    DATABASE_READ_USER = db_conf.username
-    DATABASE_READ_NAME = db_conf.path.replace("/", "")
-    DATABASE_READ_PASSWORD = db_conf.password
-    DATABASE_READ_HOST = db_conf.hostname
-    DATABASE_READ_PORT = db_conf.port
-else:
-    DATABASE_READ_USER = get_config(
-        "services", "database_read", "username", default="postgres"
-    )
-    DATABASE_READ_NAME = get_config(
-        "services", "database_read", "name", default="postgres"
-    )
-    DATABASE_READ_PASSWORD = get_config(
-        "services", "database_read", "password", default="postgres"
-    )
-    DATABASE_READ_HOST = get_config(
-        "services", "database_read", "host", default="postgres"
-    )
-    DATABASE_READ_PORT = get_config("services", "database_read", "port", default=5432)
-
 GRAPHQL_QUERY_COST_THRESHOLD = get_config(
     "setup", "graphql", "query_cost_threshold", default=10000
 )
 
-TIMESERIES_ENABLED = get_config("setup", "timeseries", "enabled", default=False)
-TIMESERIES_REAL_TIME_AGGREGATES = get_config(
-    "setup", "timeseries", "real_time_aggregates", default=False
-)
-
-timeseries_database_url = get_config("services", "timeseries_database_url")
-if timeseries_database_url:
-    timeseries_database_conf = urlparse(timeseries_database_url)
-    TIMESERIES_DATABASE_USER = timeseries_database_conf.username
-    TIMESERIES_DATABASE_NAME = timeseries_database_conf.path.replace("/", "")
-    TIMESERIES_DATABASE_PASSWORD = timeseries_database_conf.password
-    TIMESERIES_DATABASE_HOST = timeseries_database_conf.hostname
-    TIMESERIES_DATABASE_PORT = timeseries_database_conf.port
-else:
-    TIMESERIES_DATABASE_USER = get_config(
-        "services", "timeseries_database", "username", default="postgres"
-    )
-    TIMESERIES_DATABASE_NAME = get_config(
-        "services", "timeseries_database", "name", default="postgres"
-    )
-    TIMESERIES_DATABASE_PASSWORD = get_config(
-        "services", "timeseries_database", "password", default="postgres"
-    )
-    TIMESERIES_DATABASE_HOST = get_config(
-        "services", "timeseries_database", "host", default="timescale"
-    )
-    TIMESERIES_DATABASE_PORT = get_config(
-        "services", "timeseries_database", "port", default=5432
-    )
-
-TIMESERIES_DATABASE_READ_REPLICA_ENABLED = get_config(
-    "setup", "timeseries", "read_replica_enabled", default=False
-)
-
-timeseries_database_read_url = get_config("services", "timeseries_database_read_url")
-if timeseries_database_read_url:
-    timeseries_database_conf = urlparse(timeseries_database_read_url)
-    TIMESERIES_DATABASE_READ_USER = timeseries_database_conf.username
-    TIMESERIES_DATABASE_READ_NAME = timeseries_database_conf.path.replace("/", "")
-    TIMESERIES_DATABASE_READ_PASSWORD = timeseries_database_conf.password
-    TIMESERIES_DATABASE_READ_HOST = timeseries_database_conf.hostname
-    TIMESERIES_DATABASE_READ_PORT = timeseries_database_conf.port
-else:
-    TIMESERIES_DATABASE_READ_USER = get_config(
-        "services", "timeseries_database_read", "username", default="postgres"
-    )
-    TIMESERIES_DATABASE_READ_NAME = get_config(
-        "services", "timeseries_database_read", "name", default="postgres"
-    )
-    TIMESERIES_DATABASE_READ_PASSWORD = get_config(
-        "services", "timeseries_database_read", "password", default="postgres"
-    )
-    TIMESERIES_DATABASE_READ_HOST = get_config(
-        "services", "timeseries_database_read", "host", default="timescale"
-    )
-    TIMESERIES_DATABASE_READ_PORT = get_config(
-        "services", "timeseries_database_read", "port", default=5432
-    )
-
-# this is the time in seconds django decides to keep the connection open after the request
-# the default is 0 seconds, meaning django closes the connection after every request
-# https://docs.djangoproject.com/en/3.1/ref/settings/#conn-max-age
-CONN_MAX_AGE = int(get_config("services", "database", "conn_max_age", default=0))
-
-DATABASES = {
-    "default": {
-        "ENGINE": "psqlextra.backend",
-        "NAME": DATABASE_NAME,
-        "USER": DATABASE_USER,
-        "PASSWORD": DATABASE_PASSWORD,
-        "HOST": DATABASE_HOST,
-        "PORT": DATABASE_PORT,
-        "CONN_MAX_AGE": CONN_MAX_AGE,
-    }
-}
-
-if DATABASE_READ_REPLICA_ENABLED:
-    DATABASES["default_read"] = {
-        "ENGINE": "psqlextra.backend",
-        "NAME": DATABASE_READ_NAME,
-        "USER": DATABASE_READ_USER,
-        "PASSWORD": DATABASE_READ_PASSWORD,
-        "HOST": DATABASE_READ_HOST,
-        "PORT": DATABASE_READ_PORT,
-        "CONN_MAX_AGE": CONN_MAX_AGE,
-    }
-
-if TIMESERIES_ENABLED:
-    DATABASES["timeseries"] = {
-        "ENGINE": "django_prometheus.db.backends.postgresql",
-        "NAME": TIMESERIES_DATABASE_NAME,
-        "USER": TIMESERIES_DATABASE_USER,
-        "PASSWORD": TIMESERIES_DATABASE_PASSWORD,
-        "HOST": TIMESERIES_DATABASE_HOST,
-        "PORT": TIMESERIES_DATABASE_PORT,
-        "CONN_MAX_AGE": CONN_MAX_AGE,
-    }
-
-    if TIMESERIES_DATABASE_READ_REPLICA_ENABLED:
-        DATABASES["timeseries_read"] = {
-            "ENGINE": "django_prometheus.db.backends.postgresql",
-            "NAME": TIMESERIES_DATABASE_READ_NAME,
-            "USER": TIMESERIES_DATABASE_READ_USER,
-            "PASSWORD": TIMESERIES_DATABASE_READ_PASSWORD,
-            "HOST": TIMESERIES_DATABASE_READ_HOST,
-            "PORT": TIMESERIES_DATABASE_READ_PORT,
-            "CONN_MAX_AGE": CONN_MAX_AGE,
-        }
-
-# See https://django-postgres-extra.readthedocs.io/en/master/settings.html
-POSTGRES_EXTRA_DB_BACKEND_BASE: "django_prometheus.db.backends.postgresql"  # type: ignore
-
-# Allows to use the pgpartition command
-PSQLEXTRA_PARTITIONING_MANAGER = (
-    "shared.django_apps.user_measurements.partitioning.manager"
-)
-
 DATABASE_ROUTERS = ["codecov.db.DatabaseRouter"]
 
 # Password validation
@@ -502,7 +338,7 @@
 CORS_ALLOWED_ORIGIN_REGEXES = get_config(
     "setup", "api_cors_allowed_origin_regexes", default=[]
 )
-CORS_ALLOWED_ORIGINS = []
+CORS_ALLOWED_ORIGINS: list[str] = []
 
 GRAPHQL_PLAYGROUND = True