remove factory classes and refactor product creation to use a single …

…factory

main.py

@@ -1,19 +1,39 @@
 from db import Connection
 
-from factories.tv_factory import TVFactory
-from factories.laptop_factory import LaptopFactory
+from pick import pick
+
+from product_factory import ProductFactory
 
 def main():
     db = Connection('localhost', 'root', 'password')
 
-    tv_factory = TVFactory()
-    laptop_factory = LaptopFactory()
+    inventory = ProductFactory()
+
+    title = 'Main Menu'
+    options = ['Create Product', 'View Products', 'Close']
+    option, index = pick(options, title, indicator='>')
+
+    if index == 0:
+        title = 'Select product to create'
+        options = list(inventory.product_types.keys())
+        option, index = pick(options, title, indicator='>')
+
+        if option in inventory.product_types.keys():
+            title = f'Create Product: {option}'
+            print(title)
+            name = input('Name: ')
+            price = input('Price: ')
+            product = inventory.create_product(option, name, price)
+            product.display_info()
 
-    product1 = tv_factory.create_product('Samsung 65 4k QLED', 7000)
-    product2 = laptop_factory.create_product('Lenovo Thinkpad', 3000)
+    elif index == 1:
+        title = option
+        options = ['All', 'Search', *inventory.product_types.keys()]
+        option, index = pick(options, title, indicator='>')
+
+    if index == 2:
+        return
 
-    product1.display_info()
-    product2.display_info()
 
 if __name__ == '__main__':
     main()

product_factory.py

@@ -0,0 +1,20 @@
+from inspect import getmembers, isclass, isabstract
+import products
+
+class ProductFactory:
+    product_types = {}
+
+    def __init__(self):
+        self.load_products()
+
+    def load_products(self) -> None:
+        members = getmembers(products, lambda m: isclass(m) and not isabstract(m))
+        for name, _type in members:
+            if isclass(_type) and issubclass(_type, products.Product):
+                self.product_types[name] = _type
+
+    def create_product(self, product_type, name, price, *args, **kwargs) -> products.Product:
+        if product_type in self.product_types:
+            return self.product_types[product_type](name, price, *args, **kwargs)
+        else:
+            raise ValueError(f"Invalid product type: {product_type}")

products/__init__.py

@@ -1,2 +1,3 @@
+from .product import Product
 from .tv import TV
 from .laptop import Laptop

requirements.txt

@@ -1,2 +1,4 @@
 mysql-connector-python==8.3.0
+pick==2.2.0
 tabulate==0.9.0
+windows-curses==2.3.2