# =============================================================================
# NetBox Branching Configuration
# =============================================================================
# This file configures DynamicSchemaDict required by netbox-branching plugin.
# Mount this file to /etc/netbox/config/local_settings.py
# =============================================================================

import os
from netbox_branching.utilities import DynamicSchemaDict

DATABASES = DynamicSchemaDict({
    'default': {
        'ENGINE': 'django.db.backends.postgresql',
        'NAME': os.environ.get('DB_NAME', 'netbox'),
        'USER': os.environ.get('DB_USER', 'netbox'),
        'PASSWORD': os.environ.get('DB_PASSWORD', ''),
        'HOST': os.environ.get('DB_HOST', 'localhost'),
        'PORT': os.environ.get('DB_PORT', ''),
        'CONN_MAX_AGE': int(os.environ.get('DB_CONN_MAX_AGE', '300')),
    }
})

DATABASE_ROUTERS = [
    'netbox_branching.database.BranchAwareRouter',
]