Feature 'list warehouses' implemented

app_customer.py

@@ -51,6 +51,12 @@ while running:
                 display.error('No item found for order', bookId)
             else:
                 display.orderedItemList(itemList)
+    elif action == 'list warehouses':
+        houseList = consumer.getWarehousesOfCompany(login)
+        if len(houseList) == 0:
+            display.error('No warehouse for', login, 'yet.')
+        else:
+            display.warehouseList(houseList)
     else:
         print('Action', action, 'not implemented yet.')
 

basic_display.py

@@ -1,6 +1,7 @@
 from model.product import Product
 from model.ord import Ord
 from model.ori import Ori
+from model.house import Warehouse
 
 class BasicDisplay():
     def title(self, title):
@@ -36,3 +37,11 @@ class BasicDisplay():
         for iRaw in iList:
             i = Ori(iRaw)
             self.orderedItem(i)
+
+    def warehouse(self, w):
+        print('Warehouse', w.id, ':', w.number, w.name, ',', w.city)
+
+    def warehouseList(self, wList):
+        for wRaw in wList:
+            w = Warehouse(wRaw)
+            self.warehouse(w)

consumer/connect.py

@@ -61,6 +61,17 @@ class DataConsumer():
 
         return rows
         
+    def getWarehousesOfCompany(self, company):
+        cursor = self.conn.cursor()
+
+        cursor.execute("SELECT war_id, add_number, add_name, add_city FROM warehouse AS w JOIN company AS c ON w.com_id = c.com_id JOIN postaladdress AS p ON w.add_id = p.add_id WHERE com_name = %s", (company,))
+
+        rows = cursor.fetchall()
+
+        cursor.close()
+
+        return rows
+
     def commit(self):
         """Make the changes to the database persistent"""
         self.conn.commit()

model/house.py

@@ -0,0 +1,4 @@
+class Warehouse():
+    """Bean for a warehouse with its address"""
+    def __init__(self, raw):
+        (self.id, self.number, self.name, self.city) = raw