# Support both django-tenant-schemas and django-tenants packages
try:
    from django_tenants.test.cases import TenantTestCase
    from django_tenants.utils import (get_public_schema_name, get_tenant_model,
                                      get_tenant_domain_model, schema_context)
    from django_tenants.models import TenantMixin
    TENANT_SCHEMAS = False

except ImportError as e:
    from tenant_schemas.test.cases import TenantTestCase
    from tenant_schemas.utils import (get_public_schema_name, get_tenant_model,
                                      schema_context)
    from tenant_schemas.models import TenantMixin
    get_tenant_domain_model = None

    TENANT_SCHEMAS = True
