Forráskód Böngészése

Implemented factory method for warehouse_model

Vsevolod Levitan 1 éve
szülő
commit
8d14d6e513
2 módosított fájl, 8 hozzáadás és 4 törlés
  1. 2 2
      src/logic/start_factory.py
  2. 6 2
      src/models/warehouse_model.py

+ 2 - 2
src/logic/start_factory.py

@@ -141,13 +141,13 @@ class start_factory:
         nm = nomenclature_model(
         nm = nomenclature_model(
             "Test", "Test Test", units[0], nomenclature_group_model.create_group()
             "Test", "Test Test", units[0], nomenclature_group_model.create_group()
         )
         )
-        wh = warehouse_model("Test", "Testtest")
+        wh = warehouse_model.create("Test", "Testtest")
         res = []
         res = []
 
 
         for i in range(0, 20):
         for i in range(0, 20):
             unit = units[i % 5]
             unit = units[i % 5]
             res.append(
             res.append(
-                transaction_model(
+                transaction_model.create(
                     "test", wh, nm, i, i % 2, unit, datetime.datetime.now()
                     "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 = ""
     __address = ""
 
 
-    def __init__(self, name, address):
+    def __init__(self, name=None):
         self.__vtor = validator()
         self.__vtor = validator()
-        self.address = address
         super().__init__(name)
         super().__init__(name)
 
 
+    def create(name, address):
+        obj = warehouse_model(name)
+        obj.address = address
+        return obj
+
     @property
     @property
     def address(self):
     def address(self):
         """Адрес склада"""
         """Адрес склада"""