from typing import Any, AnyStr, Dict
from sqlalchemy import Column, ForeignKey, Integer, String
from sqlalchemy.orm import relationship

from db.models.base import Base, BaseMixin


class Article(Base, BaseMixin):
    __tablename__ = 'article'
    title = Column(String, unique=True)
    article_authors = relationship("ArticleAuthor")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.title = import_data['title']


class Author(Base, BaseMixin):
    __tablename__ = 'author'
    first_name = Column(String)
    last_name = Column(String)
    article_authors = relationship("ArticleAuthor")
    book_authors = relationship("BookAuthor")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.first_name = import_data['first_name']
        self.last_name = import_data['last_name']


class BookshelfPublisher(Base, BaseMixin):
    __tablename__ = 'bookshelf_publisher'
    name = Column(String, unique=True)
    books = relationship("Book")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.name = import_data['name']


class Book(Base, BaseMixin):
    __tablename__ = 'book'
    isbn = Column(String, unique=True)
    title = Column(String)
    year = Column(Integer, nullable=False)
    publisher_id = Column(String, ForeignKey('bookshelf_publisher.id'), nullable=False)
    publisher = relationship('BookshelfPublisher', back_populates="books")
    book_authors = relationship("BookAuthor")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.isbn = import_data['isbn']
        self.title = import_data['title']
        self.year = import_data['year']
        self.publisher_id = import_data['publisher_id']


class ArticleAuthor(Base, BaseMixin):
    __tablename__ = 'article_author'
    article_id = Column(String, ForeignKey('article.id'), nullable=False)
    article = relationship('Article', back_populates="article_authors")
    author_id = Column(String, ForeignKey('author.id'), nullable=False)
    author = relationship('Author', back_populates="article_authors")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.article_id = import_data['article_id']
        self.author_id = import_data['author_id']


class BookAuthor(Base, BaseMixin):
    __tablename__ = 'book_author'
    author_id = Column(String, ForeignKey('author.id'), nullable=False)
    author = relationship('Author', back_populates="book_authors")
    book_id = Column(String, ForeignKey('book.id'), nullable=False)
    book = relationship('Book', back_populates="book_authors")

    def import_dict(self, import_data: Dict[AnyStr, Any]):
        self.id = import_data['id']
        self.created_date = import_data['created_date']
        self.last_modified_date = import_data['last_modified_date']
        self.version = import_data['version']
        self.author_id = import_data['author_id']
        self.book_id = import_data['book_id']