# -*- coding: utf-8 -*-

from pyjeeves.models.raw import Article as ArticleModel, ProductClass, ArticleClass, CommodityGroup
from pyjeeves.models import db
from sqlalchemy.sql.expression import and_
from sqlalchemy.orm.exc import NoResultFound

from pyjeeves import logging
logger = logging.getLogger("PyJeeves." + __name__)


# Relocate Jeeves modules to separate folder and let a "master" module handle imports, and setup.
class Article():
    """Handles articles in Jeeves, currently filters out all articles with class = 2"""

    @staticmethod
    def get(art_no):
        """ Query an article by number """
        try:
            return db.raw.query(ArticleModel).filter_by(
                ArtNr=str(art_no)
            ).one()
        except NoResultFound:
            raise KeyError

    @staticmethod
    def get_all(filter_=and_(ArticleModel.ItemStatusCode == 0, ArticleModel.ArtKod != 2)):
        # .filter_by(ItemStatusCode=0, ArtKod=2)
        return db.raw.query(ArticleModel).filter(filter_).all()

    @staticmethod
    def is_salable(art_no_list=[]):
        """ Returns true if all articles are salable,
            else false with error information """
        articles = db.raw.query(ArticleModel).filter(
            and_(ArticleModel.ArtNr.in_(art_no_list))).all()

        blocked_articles = [article.ArtNr for article in articles
                            if article.ArtKod == 2 or article.ItemStatusCode != 0]
        unknown_articles = [x for x in art_no_list
                            if x not in set([article.ArtNr for article in articles])]

        if blocked_articles or unknown_articles:
            errors = {}
            if blocked_articles:
                errors['blocked_articles'] = blocked_articles
            if unknown_articles:
                errors['unknown_articles'] = unknown_articles
            return False, errors

        return True, {}


class ArticleCategory():
    """Handles article categories, such as classes and groups in Jeeves"""

    @staticmethod
    def get_all():
        # .filter_by(ItemStatusCode=0, ArtKod=2)
        prod_classes = db.raw.query(ProductClass).all()
        art_classes = db.raw.query(ArticleClass).all()
        com_groups = db.raw.query(CommodityGroup).all()

        return {'ProductClasses': prod_classes,
                'ArticleClasses': art_classes, 'CommodityGroups': com_groups}


if __name__ == '__main__':
    # print([column.key for column in Company.__table__.columns])

    logger.info("Starting TEST")
    # session = RawSession()

    logger.info("Testing gettings an article")
    # c1 = session.query(Company).filter_by(FtgNr="179580").first()
    # print(ArticleModel)
    c1 = ArticleModel.query.filter_by(ArtNr="2103").first()
    print(c1)
    logger.info(c1.json)

    print(
        len(ArticleModel.get_all())
    )