thegeneralist01/twitter-openapi
1
Fork 0
mirror of https://github.com/thegeneralist01/twitter-openapi synced 2026-01-11 15:40:26 +01:00
Code Issues Activity

update hook

Browse source 
Signed-off-by: ふぁ <yuki@yuki0311.com>
This commit is contained in:
ふぁ 2023-11-21 10:25:16 +09:00
parent 5c36043f3a
commit 37276039e9
No known key found for this signature in database
GPG key ID: 83A8A5E74872A8AA
1 changed files with 29 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

50
tools/hooks.py
View file

@ -10,19 +10,25 @@ class HookBase:
self.PLACEHOLDER = json.load(f) self.PLACEHOLDER = json.load(f)
def placeholder_to_yaml(self, obj, default=False, example=False) -> dict: def placeholder_to_yaml(self, obj, default=False, example=False) -> dict:
fn = lambda x: self.placeholder_to_yaml(x, default=default, example=example) def mine(x):
if type(obj) is dict: return self.placeholder_to_yaml(x, default=default, example=example)
properties = {i: fn(obj[i]) for i in obj}
def fn(x: str):
return x[:-1] if x.endswith("?") else x
if isinstance(obj, dict):
req = {k: v for k, v in obj.items() if not k.endswith("?")}
properties = {fn(k): mine(v) for k, v in obj.items()}
value = { value = {
"type": "object", "type": "object",
"properties": properties, "properties": properties,
} }
value.update({"required": [i for i in obj]} if len(obj) > 0 else {}) value.update({"required": [i for i in req]} if len(req) > 0 else {})
value.update({"default": properties} if default else {}) value.update({"default": properties} if default else {})
value.update({"example": properties} if example else {}) value.update({"example": properties} if example else {})
return value return value
elif type(obj) is list and len(obj) > 0: elif isinstance(obj, list) and len(obj) > 0:
properties = fn(obj[0]) properties = mine(obj[0])
value = { value = {
"type": "array", "type": "array",
"items": properties, "items": properties,
@ -30,7 +36,7 @@ class HookBase:
value.update({"default": [properties]} if default else {}) value.update({"default": [properties]} if default else {})
value.update({"example": [properties]} if example else {}) value.update({"example": [properties]} if example else {})
return value return value
elif type(obj) is list and len(obj) == 0: elif isinstance(obj, list) and len(obj) == 0:
value = { value = {
"type": "array", "type": "array",
"items": {"type": "object"}, "items": {"type": "object"},
@ -38,11 +44,11 @@ class HookBase:
value.update({"default": []} if default else {}) value.update({"default": []} if default else {})
value.update({"example": []} if example else {}) value.update({"example": []} if example else {})
return value return value
elif type(obj) is str: elif isinstance(obj, str):
return {"type": "string", "example": obj, "default": obj} return {"type": "string", "example": obj, "default": obj}
elif type(obj) is bool: elif isinstance(obj, bool):
return {"type": "boolean", "example": obj, "default": obj} return {"type": "boolean", "example": obj, "default": obj}
elif type(obj) is int: elif isinstance(obj, int):
return {"type": "integer", "example": obj, "default": obj} return {"type": "integer", "example": obj, "default": obj}
def load_component(self, name: str) -> dict: def load_component(self, name: str) -> dict:
@ -112,7 +118,7 @@ class RemoveDiscriminator(SchemasHookBase):
class SchemasCheck(SchemasHookBase): class SchemasCheck(SchemasHookBase):
def hook(self, value: dict): def hook(self, value: dict):
if value.get("allOf") is not None: if value.get("allOf") is not None:
print(f"allOf is used") print("allOf is used")
if value.get("type") is None: if value.get("type") is None:
print("Type is None") print("Type is None")
return value return value
@ -211,14 +217,18 @@ class AddParametersOnParameters(RequestHookBase):
for key in data.keys(): for key in data.keys():
if self.schemaType == "string": if self.schemaType == "string":
example = data[key] if type(data[key]) is str else json.dumps(data[key]) example = (
data[key] if isinstance(data[key], str) else json.dumps(data[key])
)
schema = { schema = {
"type": "string", "type": "string",
"default": example, "default": example,
"example": example, "example": example,
} }
elif self.schemaType == "object": elif self.schemaType == "object":
example = data[key] if type(data[key]) is str else json.dumps(data[key]) example = (
data[key] if isinstance(data[key], str) else json.dumps(data[key])
)
schema = { schema = {
"type": "object", "type": "object",
"default": example, "default": example,
@ -259,18 +269,18 @@ class AddParametersOnBody(RequestHookBase):
def hook(self, path: str, value: dict): def hook(self, path: str, value: dict):
path, value = super().hook(path, value) path, value = super().hook(path, value)
data = self.PLACEHOLDER[self.path_name] data: dict[str, dict] = self.PLACEHOLDER[self.path_name]
data = {key: data[key] for key in data.keys() if key not in self.ignoreKeys} data = {k: v for k, v in data.items() if k not in self.ignoreKeys}
if self.schemaType == "string": if self.schemaType == "string":
example = data if type(data) is str else json.dumps(data) example = data if isinstance(data, str) else json.dumps(data)
schema = { schema = {
"type": "string", "type": "string",
"default": example, "default": example,
"example": example, "example": example,
} }
elif self.schemaType == "object": elif self.schemaType == "object":
example = data if type(data) is str else json.dumps(data) example = data if isinstance(data, str) else json.dumps(data)
schema = { schema = {
"type": "object", "type": "object",
"default": example, "default": example,
@ -278,10 +288,8 @@ class AddParametersOnBody(RequestHookBase):
} }
else: else:
schema = { schema = {
"properties": { "properties": {k: self.placeholder_to_yaml(v) for k, v in data.items()},
i: self.placeholder_to_yaml(data[i]) for i in data.keys() "required": [k for k in data.keys()],
},
"required": [i for i in data.keys()],
} }
value["requestBody"] = { value["requestBody"] = {
"description": "body", "description": "body",