From 735eda40118a1d9b49af085775c520f2d8705703 Mon Sep 17 00:00:00 2001 From: Keiran Price Date: Mon, 5 Aug 2019 17:28:09 +0100 Subject: [PATCH] #15: Use config to control generation --- src/swagger/swagger_generator.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/swagger/swagger_generator.py b/src/swagger/swagger_generator.py index f09da43a..dd83befc 100644 --- a/src/swagger/swagger_generator.py +++ b/src/swagger/swagger_generator.py @@ -2,10 +2,11 @@ import re from pathlib import Path +from common.config import config + class SwaggerGenerator(object): FILE_PATH = Path.cwd() / "swagger" / "openapi.yaml" - is_generating = False def __init__(self): self.endpoints = [] @@ -26,7 +27,7 @@ def resource_wrapper(self): Wrapper for Resource classes that appends the class name to the endpoints list """ def decorate(cls): - if SwaggerGenerator.is_generating: + if config.is_generate_swagger(): self.endpoints.append(cls.__name__) return cls return decorate @@ -36,7 +37,7 @@ def write_swagger_spec(self): Writes the openapi.yaml file """ - if SwaggerGenerator.is_generating: + if config.is_generate_swagger(): with open(SwaggerGenerator.FILE_PATH, "w+") as target: target.write(self.get_yaml_top()) target.write(self.get_yaml_paths())