Updated. Added "count" method in "BaseModel"

gil9red · gil9red · commit 032de7dc029a · 2022-02-22T18:48:17.000+05:00
diff --git a/cbr_ru/xml_metall/db.py b/cbr_ru/xml_metall/db.py
@@ -112,12 +112,16 @@ def paginating(
     def get_inherited_models(cls) -> List[Type['BaseModel']]:
         return sorted(cls.__subclasses__(), key=lambda x: x.__name__)
 
+    @classmethod
+    def count(cls) -> int:
+        return cls.select().count()
+
     @classmethod
     def print_count_of_tables(cls):
         items = []
         for sub_cls in cls.get_inherited_models():
             name = sub_cls.__name__
-            count = sub_cls.select().count()
+            count = sub_cls.count()
             items.append(f'{name}: {count}')
 
         print(', '.join(items))
