Files

121 lines
4.8 KiB
Python

"""
Database configuration for thrillwiki project.
This module configures database connections, connection pooling, and
GeoDjango settings using python-decouple for consistent environment
variable management.
Why python-decouple?
- Already used in base.py for consistency
- Simpler API than django-environ
- Sufficient for our configuration needs
- Better separation of config from code
Database URL Format:
- PostgreSQL: postgres://user:password@host:port/dbname
- PostGIS: postgis://user:password@host:port/dbname
- SQLite: sqlite:///path/to/db.sqlite3
- SpatiaLite: spatialite:///path/to/db.sqlite3
"""
import dj_database_url
from decouple import config
# =============================================================================
# Database Configuration
# =============================================================================
# Parse DATABASE_URL environment variable into Django database settings
DATABASE_URL = config("DATABASE_URL", default="postgis://thrillwiki_user:thrillwiki@localhost:5432/thrillwiki_test_db")
# Parse the database URL
db_config = dj_database_url.parse(DATABASE_URL)
# Force PostGIS backend for spatial data support
# This ensures GeoDjango features work correctly
if "postgis" in DATABASE_URL or "postgresql" in DATABASE_URL:
db_config["ENGINE"] = "django.contrib.gis.db.backends.postgis"
DATABASES = {
"default": db_config,
}
# =============================================================================
# Database Connection Pooling Configuration
# =============================================================================
# Connection pooling improves performance by reusing database connections
# CONN_MAX_AGE: How long to keep connections open (in seconds)
# 0 = Close after each request (default Django behavior)
# None = Unlimited reuse (not recommended)
# 60 = 1 minute (good for development to prevent connection accumulation)
# 600 = 10 minutes (good for production)
# Check if we're in debug mode (imported from base settings)
DEBUG = config("DEBUG", default=False, cast=bool)
# Use shorter connection lifetime in development to prevent accumulation
CONN_MAX_AGE = config(
"DATABASE_CONN_MAX_AGE",
default=60 if DEBUG else 600,
cast=int
)
# Apply CONN_MAX_AGE to the default database
DATABASES["default"]["CONN_MAX_AGE"] = CONN_MAX_AGE
# =============================================================================
# Database Connection Options (PostgreSQL-specific)
# =============================================================================
# These settings are passed to psycopg2 when creating new connections
# Shorter timeouts in development to fail fast
connect_timeout = config("DATABASE_CONNECT_TIMEOUT", default=5 if DEBUG else 10, cast=int)
statement_timeout = config("DATABASE_STATEMENT_TIMEOUT", default=30000, cast=int)
# Idle in transaction timeout: close connections that sit idle in a transaction
# This prevents connection leaks from unclosed transactions
idle_in_transaction_timeout = config("DATABASE_IDLE_IN_TRANSACTION_TIMEOUT", default=60000, cast=int)
DATABASE_OPTIONS = {
# Connection timeout in seconds
"connect_timeout": connect_timeout,
# PostgreSQL server-side options
"options": (
f"-c statement_timeout={statement_timeout} "
f"-c idle_in_transaction_session_timeout={idle_in_transaction_timeout}"
),
}
# Apply options to PostgreSQL databases
if "postgis" in DATABASE_URL or "postgresql" in DATABASE_URL:
DATABASES["default"].setdefault("OPTIONS", {})
DATABASES["default"]["OPTIONS"].update(DATABASE_OPTIONS)
# =============================================================================
# GeoDjango Settings
# =============================================================================
# Library paths for GDAL and GEOS (required for GeoDjango)
# These vary by operating system and installation method
# macOS with Homebrew (default)
# Linux: /usr/lib/x86_64-linux-gnu/libgdal.so
# Docker: Usually handled by the image
GDAL_LIBRARY_PATH = config("GDAL_LIBRARY_PATH", default="/opt/homebrew/lib/libgdal.dylib")
GEOS_LIBRARY_PATH = config("GEOS_LIBRARY_PATH", default="/opt/homebrew/lib/libgeos_c.dylib")
# =============================================================================
# Read Replica Configuration (Optional)
# =============================================================================
# Configure read replicas for read-heavy workloads
# Set DATABASE_READ_REPLICA_URL to enable
DATABASE_READ_REPLICA_URL = config("DATABASE_READ_REPLICA_URL", default="")
if DATABASE_READ_REPLICA_URL:
replica_config = dj_database_url.parse(DATABASE_READ_REPLICA_URL)
if "postgis" in DATABASE_READ_REPLICA_URL or "postgresql" in DATABASE_READ_REPLICA_URL:
replica_config["ENGINE"] = "django.contrib.gis.db.backends.postgis"
replica_config["CONN_MAX_AGE"] = CONN_MAX_AGE
DATABASES["replica"] = replica_config