• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-postgres-extra-master
  • .circleci
    • config.yml
  • .prospector.yaml
  • .coveragerc
  • psqlextra
    • apps.py
    • sql.py
    • util.py
    • expressions.py
    • type_assertions.py
    • backend
      • introspection.py
      • schema.py
      • base_impl.py
      • __init__.py
      • migrations
        • operations
          • create_view_model.py
          • apply_state.py
          • delete_view_model.py
          • add_default_partition.py
          • add_range_partition.py
          • delete_materialized_view_model.py
          • add_list_partition.py
          • create_materialized_view_model.py
          • delete_partitioned_model.py
          • delete_partition.py
          • create_partitioned_model.py
          • partition.py
          • __init__.py
          • delete_range_partition.py
          • delete_list_partition.py
          • delete_default_partition.py
        • state
          • materialized_view.py
          • model.py
          • __init__.py
          • view.py
          • partitioning.py
        • patched_project_state.py
        • patched_autodetector.py
        • __init__.py
        • README.md
        • patched_migrations.py
      • side_effects
        • hstore_required.py
        • __init__.py
        • hstore_unique.py
      • base.py
      • operations.py
    • models
      • options.py
      • __init__.py
      • partitioned.py
      • view.py
      • base.py
    • fields
      • hstore_field.py
      • __init__.py
    • query.py
    • __init__.py
    • compiler.py
    • partitioning
      • current_time_strategy.py
      • time_partition_size.py
      • config.py
      • time_strategy.py
      • range_partition.py
      • constants.py
      • manager.py
      • range_strategy.py
      • partition.py
      • __init__.py
      • shorthands.py
      • error.py
      • plan.py
      • strategy.py
      • time_partition.py
    • manager
      • manager.py
      • __init__.py
    • types.py
    • indexes
      • case_insensitive_unique_index.py
      • conditional_unique_index.py
      • __init__.py
    • management
      • commands
        • pgpartition.py
        • pgmakemigrations.py
        • pgrefreshmv.py
        • __init__.py
      • __init__.py
  • LICENSE
  • .readthedocs.yml
  • manage.py
  • CONTRIBUTING.md
  • setup.py
  • setup.cfg
  • README.md
  • tests
    • fake_model.py
    • test_schema_editor_view.py
    • test_on_conflict.py
    • test_partitioned_model.py
    • test_db_backend.py
    • test_manager_context.py
    • test_upsert.py
    • test_insert.py
    • db_introspection.py
    • test_view_models.py
    • snapshots
      • __init__.py
      • snap_test_management_command_partition.py
    • test_partitioning_manager.py
    • test_management_command_partition.py
    • test_query.py
    • test_migration_operations.py
    • test_conditional_unique_index.py
    • test_schema_editor_partitioning.py
    • benchmarks
      • test_upsert.py
      • test_insert_nothing.py
      • __init__.py
      • test_upsert_bulk.py
    • test_partitioning_time.py
    • test_materialized_view_model.py
    • test_on_conflict_update.py
    • test_hstore_required.py
    • __init__.py
    • test_hstore_autodetect.py
    • test_query_values.py
    • test_hstore_unique.py
    • test_partitioned_model_state.py
    • test_hstore_field.py
    • test_make_migrations.py
    • test_case_insensitive_unique_index.py
    • migrations.py
    • test_on_conflict_nothing.py
    • conftest.py
    • test_manager.py
  • pytest.ini
  • settings.py
  • pyproject.toml
  • .gitignore
  • docs
    • Makefile
    • source
      • conflict_handling.rst
      • expressions.rst
      • table_partitioning.rst
      • deletion.rst
      • snippets
        • manager_model_warning.rst
        • postgres_doc_links.rst
        • django_doc_links.rst
      • annotations.rst
      • hstore.rst
      • indexes.rst
      • major_releases.rst
      • api_reference.rst
      • settings.rst
      • index.rst
      • conf.py
      • managers_models.rst
      • installation.rst
    • make.bat
    • .gitignore
  • pytest-benchmark.ini
  • tox.ini

import os
import sys
import uuid

from contextlib import contextmanager

from django.apps import AppConfig, apps
from django.db import connection

from psqlextra.models import (
    PostgresMaterializedViewModel,
    PostgresModel,
    PostgresPartitionedModel,
    PostgresViewModel,
)


def define_fake_model(
    fields=None, model_base=PostgresModel, meta_options={}, **attributes
):
    """Defines a fake model (but does not create it in the database)."""

    name = str(uuid.uuid4()).replace("-", "")[:8].title()

    attributes = {
        "app_label": meta_options.get("app_label") or "tests",
        "__module__": __name__,
        "__name__": name,
        "Meta": type("Meta", (object,), meta_options),
        **attributes,
    }

    if fields:
        attributes.update(fields)

    model = type(name, (model_base,), attributes)

    apps.app_configs[attributes["app_label"]].models[name] = model
    return model


def define_fake_view_model(
    fields=None, view_options={}, meta_options={}, model_base=PostgresViewModel
):
    """Defines a fake view model."""

    model = define_fake_model(
        fields=fields,
        model_base=model_base,
        meta_options=meta_options,
        ViewMeta=type("ViewMeta", (object,), view_options),
    )

    return model


def define_fake_materialized_view_model(
    fields=None,
    view_options={},
    meta_options={},
    model_base=PostgresMaterializedViewModel,
):
    """Defines a fake materialized view model."""

    model = define_fake_model(
        fields=fields,
        model_base=model_base,
        meta_options=meta_options,
        ViewMeta=type("ViewMeta", (object,), view_options),
    )

    return model


def define_fake_partitioned_model(
    fields=None, partitioning_options={}, meta_options={}
):
    """Defines a fake partitioned model."""

    model = define_fake_model(
        fields=fields,
        model_base=PostgresPartitionedModel,
        meta_options=meta_options,
        PartitioningMeta=type(
            "PartitioningMeta", (object,), partitioning_options
        ),
    )

    return model


def get_fake_model(fields=None, model_base=PostgresModel, meta_options={}):
    """Defines a fake model and creates it in the database."""

    model = define_fake_model(fields, model_base, meta_options)

    with connection.schema_editor() as schema_editor:
        schema_editor.create_model(model)

    return model


@contextmanager
def define_fake_app():
    """Creates and registers a fake Django app."""

    name = str(uuid.uuid4()).replace("-", "")[:8] + "-app"

    app_config_cls = type(
        name + "Config",
        (AppConfig,),
        {"name": name, "path": os.path.dirname(__file__)},
    )

    app_config = app_config_cls(name, "")
    app_config.apps = apps
    app_config.models = {}

    apps.app_configs[name] = app_config
    sys.modules[name] = {}

    try:
        yield app_config
    finally:
        del apps.app_configs[name]
        del sys.modules[name]