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

from pyjeeves.models.raw import PriceList as PriceListModel
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 PriceList():
    """Handles price lists in Jeeves"""

    @staticmethod
    def get(price_list_no):
        """ Query a price list by number """
        try:
            return PriceListModel.query.filter_by(
                PrisLista=price_list_no
            ).one()
        except NoResultFound:
            raise KeyError