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

from pyjeeves.models.raw import Company, DelivLoc

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 Location():
    """Handles dispatch locations in Jeeves"""

    def __init__(self):
        super(Location, self).__init__()
        self.associated_company = ''  # Company with new/existing locations
        self._deliv_locs = []  # List of locations to be connected

    def _connect_deliv_loc(self, ftgnr, description, code):
        if self.associated_company == '':
            raise
        if len(description) > 36:
            logger.warn("Truncated description %s", (description))
            description = description[:36]
        _deliv_loc = DelivLoc(
            FtgNr=self.associated_company, OrdLevPlats1=ftgnr,
            OrdLevPlBeskr=description, ForetagKod=1)
        self._deliv_locs.append(_deliv_loc)
        # self.session.merge(_deliv_loc)
        return _deliv_loc

    def create_lev_location(self, ftgnr='', name='', address='',
                            postal_code='', city='', gln='', invoice_ref='', phone=''):

        _loc = Company(
            FtgNr=str(ftgnr), FtgNamn=name, FtgPostadr5=address,
            FtgLevPostNr=postal_code, FtgPostLevAdr3=city,
            EAN_Loc_Code=gln, FtgPostAdr1=invoice_ref, ComNr=phone,
            ForetagKod=1)

        # logger.debug("Adding company to location session")
        # with self.session.no_autoflush:
        #    # self.session.merge(_loc)  # "merge" updates if existing location exists.
        _deliv_loc = self._connect_deliv_loc(ftgnr, name, gln)

        return _loc, _deliv_loc

    def save_locations(self):
        logger.debug("Committing all location changes")
        # self.session.commit()  # Location company needs to be created in order to connect them.
        for deliv_loc in self._deliv_locs:
            deliv_loc.merge()
            # self.session.merge(deliv_loc)  # Create "connnections" between Customer and Location.
        Company.commit()
        # self.session.commit()


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

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

    logger.info("Testing gettings a company")
    # c1 = session.query(Company).filter_by(FtgNr="179580").first()
    print(Company)
    c1 = Company.query.filter_by(FtgNr="179580").first()
    logger.info(c1.json)
    # RawSession.remove()
#    from sqlalchemy.inspection import inspect
#    print (inspect(Company).columns.items())