import itertools
import contextlib
import logging

from typing import List, Any, Optional

from sqlalchemy.ext.declarative import declarative_base
from sqlalchemy.orm import sessionmaker
from sqlalchemy import create_engine

from website import settings


class Row(dict):
    """A dict that allows for object-like property access syntax."""

    def __getattr__(self, name):
        try:
            return self[name]
        except KeyError:
            raise AttributeError(name)


def to_json_list(cursor: Any) -> Optional[List[Row]]:
    column_names = list(cursor.keys())
    result = cursor.fetchall()
    if not result:
        return None

    return [Row(itertools.zip_longest(column_names, row)) for row in result]


def to_json(cursor):
    column_names = list(cursor.keys())
    result = cursor.fetchone()

    if not result:
        return None

    return Row(itertools.zip_longest(column_names, result))


app_engine = create_engine(
    'mysql+pymysql://{}:{}@{}/{}?charset=utf8mb4'.format(
        settings.mysql_app['user'],
        settings.mysql_app['password'],
        settings.mysql_app['host'],
        settings.mysql_app['database']
    ),  # SQLAlchemy 数据库连接串，格式见下面
    echo=bool(settings.SQLALCHEMY_ECHO),  # 是不是要把所执行的SQL打印出来，一般用于调试
    # pool_pre_ping=True,
    # pool_size=int(settings.SQLALCHEMY_POOL_SIZE),  # 连接池大小
    # max_overflow=int(settings.SQLALCHEMY_POOL_MAX_SIZE),  # 连接池最大的大小
    # pool_recycle=int(settings.SQLALCHEMY_POOL_RECYCLE),  # 多久时间回收连接
)

# Session = sessionmaker(bind=engine)
# Base = declarative_base(engine)
#
#
# @contextlib.contextmanager
# def get_session():
#     s = Session()
#     try:
#         yield s
#         s.commit()
#     except Exception as e:
#         s.rollback()
#         raise e
#     finally:
#         s.close()