mirror of
https://github.com/thegeneralist01/twitter-openapi
synced 2026-01-11 07:30:37 +01:00
92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
import os
|
|
import glob
|
|
import json
|
|
import yaml
|
|
import shutil
|
|
|
|
|
|
class placeholder_manager:
|
|
data: dict
|
|
config: str = "src/config/placeholder.json"
|
|
|
|
def __init__(self):
|
|
with open(self.config, mode="r", encoding="utf-8") as f:
|
|
self.data = json.load(f)
|
|
|
|
def __replace(self, file, old: str, new) -> str:
|
|
if type(new) is dict:
|
|
dump = f"'{json.dumps(new)}'"
|
|
return file.replace(f'"{old}"', dump)
|
|
else:
|
|
return file.replace(f"{old}", new)
|
|
|
|
def replace(self, file: str) -> str:
|
|
for prefix in self.data.keys():
|
|
for key in self.data[prefix]:
|
|
value = self.data[prefix][key]
|
|
file = self.__replace(file, f"{{{{{prefix}{key}}}}}", value)
|
|
return file
|
|
|
|
def replace_file(self, file: str, prefix: str) -> str:
|
|
if self.data.get(prefix, None) is None:
|
|
return file
|
|
for key in self.data[prefix].keys():
|
|
value = self.data[prefix][key]
|
|
file = self.__replace(file, f"{{{{{key}}}}}", value)
|
|
return file
|
|
|
|
|
|
OUTPUT_DIR = "dist"
|
|
INPUT_DIR = "src/openapi"
|
|
|
|
|
|
def read(file: str):
|
|
with open(file, mode="r", encoding="utf-8") as f:
|
|
return f.read()
|
|
|
|
|
|
def write(file: str, data: str) -> None:
|
|
with open(file.replace(INPUT_DIR, OUTPUT_DIR, 1), mode="w+", encoding="utf-8") as f:
|
|
f.write(data)
|
|
|
|
|
|
def get_yaml(data, key):
|
|
return yaml.safe_load(placeholder.replace_file(str(data), key))
|
|
|
|
|
|
shutil.rmtree("dist")
|
|
for dir in glob.glob(os.path.join(INPUT_DIR, "**/")):
|
|
os.makedirs(dir.replace(INPUT_DIR, OUTPUT_DIR, 1), exist_ok=True)
|
|
|
|
placeholder = placeholder_manager()
|
|
parameters = read("src/config/parameters.yaml")
|
|
header = read("src/config/header.yaml")
|
|
|
|
paths = {}
|
|
for file in glob.glob(os.path.join(INPUT_DIR, "**/*.yaml")):
|
|
file = file.replace(os.path.sep, "/")
|
|
relative = file.replace(INPUT_DIR, "", 1)
|
|
|
|
load = yaml.safe_load(placeholder.replace(read(file)))
|
|
|
|
for key in load["paths"].keys():
|
|
append = get_yaml(parameters, key.split("/")[-1])
|
|
req = load["paths"][key]["get"]
|
|
req["parameters"] = append["paths"]["/parameters"]["get"]["parameters"]
|
|
|
|
append = get_yaml(header, key.split("/")[-1])
|
|
req = load["paths"][key]["get"]
|
|
req["responses"]["200"]["headers"] = append["components"]["headers"]
|
|
|
|
escape = key.replace("/", "~1")
|
|
paths.update({key: {"$ref": f".{relative}#/paths/{escape}"}})
|
|
write(file, yaml.dump(load))
|
|
|
|
|
|
file = "src/openapi/openapi-3.0.yaml"
|
|
data = read(file)
|
|
for path in paths:
|
|
load = yaml.safe_load(placeholder.replace(data))
|
|
load["paths"] = paths
|
|
|
|
write(file, yaml.dump(load))
|