INSTALLED_APPS = [
    ...
    "python_utils.django",
    ...
]

MIDDLEWARE = [
    "python_utils.django.middleware.TenantAwareHttpMiddleware",
    ...
]

AUTH_USER_MODEL = "idp_user.User"

# Include the database configuration for each tenant in the DATABASES setting.
# You can use the get_database_configs() function from python_utils.django.db.utils as a helper.
from python_utils.django.db.utils import get_database_configs

for tenant, tenant_db_config in get_database_configs().items():
    DATABASES[tenant] = {
        "ENGINE": "django.db.backends.postgresql",
        "NAME": tenant_db_config["name"],
        "USER": tenant_db_config["user"],
        "PASSWORD": tenant_db_config["password"],
        "HOST": tenant_db_config["host"],
        "PORT": tenant_db_config.get("port", 5432),
    }

# If you want to override the database alias to use for local development (when DEBUG is True).
# By default, the first database defined in DATABASES is used.
DEVELOPMENT_TENANT = "development"

# This is required to use the tenant context when routing database queries
DATABASE_ROUTERS = ["python_utils.django.db.routers.TenantAwareRouter"]

# If using celery, set the task class to TenantAwareTask:
CELERY_TASK_CLS = "python_utils.django.celery.TenantAwareTask"

# If using Redis caching, configure the cache backend as follows:
CACHES = {
    "default": {
        "BACKEND": "django_redis.cache.RedisCache",
        "LOCATION": REDIS_LOCATION,
        "KEY_FUNCTION": "python_utils.django.redis.make_tenant_aware_key",
        **OPTIONS,
    }
}

# If using Django Storages with S3, and separate tenant buckets are needed, 
# configure the storage backends as follows:
STORAGES = {
    "default": {
        "BACKEND": "python_utils.django.storage.TenantAwarePrivateS3Storage",
    },
}

# If you want to exclude certain paths from tenant processing, use TENANT_AWARE_EXCLUDED_PATHS:
# They are considered as prefixes, so all paths starting with the given strings will be excluded.
TENANT_AWARE_EXCLUDED_PATHS = ("/some/path",)

from python_utils.django.admin.templates import TEMPLATE_PATH

INSTALLED_APPS.append("mozilla_django_oidc")
TEMPLATES[0]["DIRS"].append(TEMPLATE_PATH)

JWT_AUDIENCE = "myapp"
JWT_SCOPE_PREFIX = "myapp"

# If using DRF
REST_FRAMEWORK.update(
    {
        "DEFAULT_SCHEMA_CLASS": "drf_spectacular.openapi.AutoSchema",
        "DEFAULT_AUTHENTICATION_CLASSES": ("python_utils.django.api.drf.AuthenticationBackend",),
        "DEFAULT_PERMISSION_CLASSES": (
            "rest_framework.permissions.IsAuthenticated",
            "python_utils.django.api.drf.HasScope",
        ),
    }
)

# Admin auth backends
AUTHENTICATION_BACKENDS = [
    "django.contrib.auth.backends.ModelBackend",
    "python_utils.django.admin.auth.AdminAuthenticationBackend",
]

# If user groups are used for Row Level Security (RLS)
KEYCLOAK_USER_GROUP_MODEL = "myapp.UserGroup"

KEYCLOAK_CONFIDENTIAL_CLIENT_ID = os.getenv("KEYCLOAK_CONFIDENTIAL_CLIENT_ID", f"{JWT_AUDIENCE}_confidential")

OIDC_RP_CLIENT_ID = KEYCLOAK_CONFIDENTIAL_CLIENT_ID
OIDC_RP_SIGN_ALGO = "RS256"
OIDC_CREATE_USER = True
OIDC_AUTHENTICATE_CLASS = "python_utils.django.admin.views.TenantAwareOIDCAuthenticationRequestView"

LOGIN_REDIRECT_URL = "/admin"
SESSION_COOKIE_AGE = 60 * 30  # 30 minutes
SESSION_SAVE_EVERY_REQUEST = True  # Extend session on each request

# If using django-easy-audit

from python_utils.django.db.alias import DynamicDatabaseAlias
DJANGO_EASY_AUDIT_DATABASE_ALIAS = DynamicDatabaseAlias()

urlpatterns.append(path("oidc/", include("mozilla_django_oidc.urls")))

from python_utils.django.admin.auth import has_admin_site_permission
from python_utils.django.admin.views import TenantAwareOIDCAuthenticationRequestView

admin.site.login = TenantAwareOIDCAuthenticationRequestView.as_view()
admin.site.has_permission = has_admin_site_permission

from python_utils.django.db.transaction import tenant_atomic

@tenant_atomic
def my_function():
    # Some logic

    transaction.on_commit(do_smth, using=TenantContext.get())

from python_utils.django.db.utils import get_connection
from python_utils.django.tenant_context import TenantContext

connection = get_connection(TenantContext.get())

import pytest

from python_utils.django.tenant_context import TenantContext

# Add this, if the test utils of the package are needed
pytest_plugins = ["python_utils.django.tests"]


@pytest.fixture(scope="session", autouse=True)
def test_database() -> str:
    return "default"


@pytest.fixture(scope="session", autouse=True)
def set_tenant_context_session(test_database):
    """Set tenant context for the entire test session."""
    TenantContext.set(test_database)
    yield
    TenantContext.clear()


@pytest.fixture(autouse=True)
def set_tenant_context(set_tenant_context_session, test_database):
    """Ensure tenant context is set for each test (depends on session fixture)."""
    # Re-set in case it was cleared between tests
    if not TenantContext.is_set():
        TenantContext.set(test_database)
    yield