from pathlib import Path
import sys
from logging.config import fileConfig

import sqlalchemy
from alembic.runtime.environment import EnvironmentContext
from alembic import context
from sqlalchemy import engine_from_config, pool

sys.path.append(str(Path(".").resolve()))

from cmf.manage import *


def include_object(_obj, name: str, type_: str, _reflected: bool, _compare_to):
    """
    include_object: A callable function which is given
    the chance to return ``True`` or ``False`` for any object,
    indicating if the given object should be considered in the
    autogenerate sweep.

    The function accepts the following positional arguments:

    * ``object``: a :class:`~sqlalchemy.schema.SchemaItem` object such
    as a :class:`~sqlalchemy.schema.Table`,
    :class:`~sqlalchemy.schema.Column`,
    :class:`~sqlalchemy.schema.Index`
    :class:`~sqlalchemy.schema.UniqueConstraint`,
    or :class:`~sqlalchemy.schema.ForeignKeyConstraint` object
    * ``name``: the name of the object. This is typically available
    via ``object.name``.
    * ``type``: a string describing the type of object; currently
    ``"table"``, ``"column"``, ``"index"``, ``"unique_constraint"``,
    or ``"foreign_key_constraint"``
    * ``reflected``: ``True`` if the given object was produced based on
    table reflection, ``False`` if it's from a local :class:`.MetaData`
    object.
    * ``compare_to``: the object being compared against, if available,
    else ``None``.
    """
    if type_ == "column" and name.startswith('cf_'):
        # Кастом поля пользователя не должны попадать в миграции и рулиться алембиком.
        return False
    elif type_ == "table" and '_cf_' in name:
        return False
    elif type_ == "index" and '_cf_' in name:
        return False
    else:
        return True


# hack for type hinting
context = context  # type: EnvironmentContext

# this is the Alembic Config object, which provides
# access to the values within the .ini file in use.
config = context.config

# Interpret the config file for Python logging.
# This line sets up loggers basically.
fileConfig(config.config_file_name)

# init_ds()
# Эта команда создает таблицы в обход алембика
# init_db()


# add your model's MetaData object here
# for 'autogenerate' support
# from myapp import mymodel
# target_metadata = mymodel.Base.metadata

# other values from the config, defined by the needs of env.py,
# can be acquired:
# my_important_option = config.get_main_option("my_important_option")
# ... etc.
# Для генерации моделей надо иметь текущую мету
if not cmf.models.BaseModel.dp.data_driver:
    init_ds()
connectable = cmf.models.BaseModel.dp.data_driver.engine
target_metadata = cmf.models.BaseModel.dp.data_driver.db_meta

with connectable.connect() as connection:
    context.configure(
        connection=connection,
        target_metadata=target_metadata,
        compare_type=True,
        include_object=include_object,
    )

    with context.begin_transaction():
        context.run_migrations()
connectable.dispose()

# def run_migrations_offline():
#     """Run migrations in 'offline' mode.

#     This configures the context with just a URL
#     and not an Engine, though an Engine is acceptable
#     here as well.  By skipping the Engine creation
#     we don't even need a DBAPI to be available.

#     Calls to context.execute() here emit the given string to the
#     script output.

#     """
#     url = config.get_main_option("sqlalchemy.url")
#     context.configure(
#         url=url,
#         target_metadata=target_metadata,
#         literal_binds=True,
#         dialect_opts={"paramstyle": "named"},
#     )

#     with context.begin_transaction():
#         context.run_migrations()


# def run_migrations_online():
#     """Run migrations in 'online' mode.

#     In this scenario we need to create an Engine
#     and associate a connection with the context.

#     """
#     print(config)
#     connectable = engine_from_config(
#         config.get_section(config.config_ini_section),
#         prefix="sqlalchemy.",
#         poolclass=pool.NullPool,
#     )

#     with connectable.connect() as connection:
#         context.configure(connection=connection, target_metadata=target_metadata)

#         with context.begin_transaction():
#             context.run_migrations()


# if context.is_offline_mode():
#     run_migrations_offline()
# else:
#     run_migrations_online()