1
1
Prechádzať zdrojové kódy

Implemented factory method for warehouse_model

Vsevolod Levitan 1 rok pred
rodič
commit
8d14d6e513

+ 2 - 2
src/logic/start_factory.py

@@ -141,13 +141,13 @@ class start_factory:
         nm = nomenclature_model(
             "Test", "Test Test", units[0], nomenclature_group_model.create_group()
         )
-        wh = warehouse_model("Test", "Testtest")
+        wh = warehouse_model.create("Test", "Testtest")
         res = []
 
         for i in range(0, 20):
             unit = units[i % 5]
             res.append(
-                transaction_model(
+                transaction_model.create(
                     "test", wh, nm, i, i % 2, unit, datetime.datetime.now()
                 )
             )

+ 6 - 2
src/models/warehouse_model.py

@@ -6,11 +6,15 @@ class warehouse_model(abstract_reference):
     # Адрес
     __address = ""
 
-    def __init__(self, name, address):
+    def __init__(self, name=None):
         self.__vtor = validator()
-        self.address = address
         super().__init__(name)
 
+    def create(name, address):
+        obj = warehouse_model(name)
+        obj.address = address
+        return obj
+
     @property
     def address(self):
         """Адрес склада"""