From 33ef610f9a310582165da54afcda78fd9df60aa2 Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Mon, 29 Jul 2019 07:34:43 +0100 Subject: [PATCH] #15: Allow generator to be disabled --- src/swagger/swagger_generator.py | 20 +++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/src/swagger/swagger_generator.py b/src/swagger/swagger_generator.py index 40735f88..c940b2f2 100644 --- a/src/swagger/swagger_generator.py +++ b/src/swagger/swagger_generator.py @@ -5,6 +5,7 @@ class SwaggerGenerator(object): FILE_PATH = Path.cwd() / "swagger" / "openapi.yaml" + is_generating = False def __init__(self): self.endpoints = [] @@ -24,22 +25,23 @@ def resource_wrapper(self): """ Wrapper for Resource classes that appends the class name to the endpoints list """ + if SwaggerGenerator.is_generating: + def decorate(cls): + self.endpoints.append(cls.__name__) + return cls - def decorate(cls): - self.endpoints.append(cls.__name__) - return cls - - return decorate + return decorate def write_swagger_spec(self): """ Writes the openapi.yaml file """ - with open(SwaggerGenerator.FILE_PATH, "w+") as target: - target.write(self.get_yaml_top()) - target.write(self.get_yaml_paths()) - target.close() + if SwaggerGenerator.is_generating: + with open(SwaggerGenerator.FILE_PATH, "w+") as target: + target.write(self.get_yaml_top()) + target.write(self.get_yaml_paths()) + target.close() @staticmethod def get_yaml_top():