#109: Create dictionary to store examples

ral-facilities · Oct 29, 2019 · e9d37c7 · e9d37c7
1 parent 0df043e
commit e9d37c7
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/swagger/swagger_generator.py b/src/swagger/swagger_generator.py
@@ -58,6 +58,12 @@ def _map_db_type_to_json_type(self, type_str):
         }
         return type_map[type_str]
 
+
+
+    def __init__(self, entity_name, model):
+        self.properties_dict = {}
+        for column in model.__table__.columns:
+            self.properties_dict[column.name] = {"type": self._map_db_type_to_json_type(str(column.type))}
         self.entity_no_id_endpoint = {
             f"/{entity_name.lower()}": {
                 "get": {