"""Defines the ``DatabaseWrapper`` class of the ``mtp_db_backend``"""


from django.db.backends.postgresql_psycopg2.base import DatabaseWrapper as BaseDatabaseWrapper

from .schema import DatabaseSchemaEditor


class DatabaseWrapper(BaseDatabaseWrapper):
    """Simple override to use our own schema editor."""

    SchemaEditorClass = DatabaseSchemaEditor
