Started using 'sqlservice'. Added support for many new tables.

pyjeeves/repositories/company.py

@@ -0,0 +1,55 @@
+# -*- coding: utf-8 -*-
+
+from pyjeeves.models.raw import Company as CompanyModel, Customer as CustomerModel
+from sqlalchemy.sql.expression import and_
+
+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 Company():
+    """Handles companies in Jeeves"""
+
+    @staticmethod
+    def get(ftg_nr):
+        """ Query an article by number """
+        return CompanyModel.query.filter_by(
+            FtgNr=ftg_nr
+        ).one()
+
+    @staticmethod
+    def get_all_active_customers():
+        cust = CustomerModel.query.filter(and_(CustomerModel.Makulerad == 0)).all()
+        return [c.CompanyModel for c in cust]
+
+    @staticmethod
+    def get_list(ftg_nr=[]):
+        return CompanyModel.query.filter(
+            CompanyModel.FtgNr.in_(ftg_nr)
+        ).all()
+
+
+if __name__ == '__main__':
+    # print([column.key for column in CompanyModel.__table__.columns])
+
+    logger.info("Starting TEST")
+    # session = RawSession()
+
+    logger.info("Testing gettings a company")
+    # c1 = session.query(CompanyModel).filter_by(FtgNr="179580").first()
+    # print(CompanyModel)
+
+    # print(CompanyModel.get_list(['406569', '179580', '2440070', '179584']))
+    from pprint import pprint
+    pprint(CompanyModel.get('179584').to_dict())
+
+    # c1 = CompanyModel.query.filter_by(FtgNr="406569").first()
+    # print(c1)
+    # logger.info(c1.json)
+
+    # print(
+    #     len(CompanyModel.get_all_active_customers())
+    # )
+
+    # print(CompanyModel.get_all_active_customers()[0].CompanyModel)