#50: Create a parameter class

ral-facilities · Sep 17, 2019 · d34cef8 · d34cef8
1 parent abe4252
commit d34cef8
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/src/swagger/swagger_generator.py b/src/swagger/swagger_generator.py
@@ -6,6 +6,19 @@
 from common.config import config
 
 
+class Parameter(object):
+    def __init__(self, description, name, param_type, example, location, is_required):
+        self.parameter_as_dict = {
+            "description": description,
+            "in": location,
+            "required": is_required,
+            "name": name,
+            "schema": {
+                "type": param_type,
+                "example": example
+            }
+        }
+
 class SwaggerGenerator(object):
     FILE_PATH = Path.cwd() / "swagger" / "openapi.yaml"