瀏覽代碼

Added key validation for exporter.export_by_key

Vsevolod Levitan 1 年之前
父節點
當前提交
81ad8b8636
共有 1 個文件被更改,包括 4 次插入0 次删除
  1. 4 0
      src/export/exporter.py

+ 4 - 0
src/export/exporter.py

@@ -26,6 +26,10 @@ class exporter:
     def export_by_key(self, key):
         """Экспортировать модели из storage по ключу"""
 
+        # Проверяем чтобы ключ был строкой с длиной >0
+        self.__vtor.check_type(key, str)
+        self.__vtor.check_length_greater(key, 0)
+
         models = storage().data[key]
 
         return self.export_models(models)