import_images.py

import csv

from sqlalchemy import pool
from app.orm_decl import (EditionImage, Work, Edition, Part, Person,
                          Contributor)
from app import app
import sys
import os
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.pool import NullPool
from typing import List, Any

db_url = app.config['SQLALCHEMY_DATABASE_URI']

just_check: bool = False

img_dir = '/static/images/books/'


def add_image(s: Any, image: List[str]) -> None:
    image_src = img_dir + image[0]
    author_str = image[1]
    title_str = image[2]
    pubyear_str = image[3]

    editions = s.query(Edition)\
                .filter(Edition.title == title_str)\
                .filter(Edition.pubyear == pubyear_str)\
                .join(Part)\
                .filter(Part.edition_id == Edition.id)\
                .join(Contributor)\
                .filter(Contributor.part_id == Part.id, Contributor.role_id == 1)\
                .join(Person, Person.id == Contributor.person_id)\
                .filter(Person.id == Contributor.person_id)\
                .filter(Person.name.like(author_str + '%'))\
                .all()

    if not editions:
        print(f'Not found: {author_str}: {title_str}.')
        return
    else:
        print(f'Found: {author_str}: {title_str}')
    if just_check == False:
        for edition in editions:
            ei = EditionImage(edition_id=edition.id, image_src=image_src)
            s.add(ei)
        s.commit()


def import_images(params) -> None:
    engine = create_engine(db_url, poolclass=NullPool)
    session = sessionmaker()
    session.configure(bind=engine)
    s = session()

    if len(params) > 0:
        just_check = True

    with open('bibfiles/covers.csv') as csvfile:
        imagescsv = csv.reader(csvfile, delimiter=';', quotechar='"')
        for image in imagescsv:
            add_image(s, image)


if __name__ == '__main__':
    import_images(sys.argv[1:])