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

update sereialize

Signed-off-by: ふぁ <yuki@yuki0311.com>
This commit is contained in:
ふぁ 2023-11-21 11:56:40 +09:00
parent c0ad260ab6
commit a506fd6506
No known key found for this signature in database
GPG key ID: 83A8A5E74872A8AA

View file

@ -28,14 +28,14 @@ def get_key(snake_str):
def get_cursor(obj, fn): def get_cursor(obj, fn):
res = [] res = []
if type(obj) == dict: if isinstance(obj, dict):
callback = fn(obj) callback = fn(obj)
if callback is not None: if callback is not None:
res.extend(callback) res.extend(callback)
else: else:
for v in obj.values(): for v in obj.values():
res.extend(get_cursor(v, fn)) res.extend(get_cursor(v, fn))
elif type(obj) == list: elif isinstance(obj, list):
for v in obj: for v in obj:
res.extend(get_cursor(v, fn)) res.extend(get_cursor(v, fn))
return res return res
@ -77,9 +77,8 @@ def match_rate(a, b, key=""):
return 1 return 1
if a is False and b is None: if a is False and b is None:
return 1 return 1
if type(a) != type(b): if isinstance(a, list):
return match_rate_zero(key) data = [match_rate(a[i], b[i], key=f"{key}[{i}]") for i in range(len(a))]
if type(a) == dict and type(b) == dict:
if len(a) == 0 and len(b) == 0: if len(a) == 0 and len(b) == 0:
return 1 return 1
if len(a) == 0 or len(b) == 0: if len(a) == 0 or len(b) == 0:
@ -88,11 +87,11 @@ def match_rate(a, b, key=""):
data = [match_rate(a.get(k), b.get(k), key=f"{key}.{k}") for k in a.keys()] data = [match_rate(a.get(k), b.get(k), key=f"{key}.{k}") for k in a.keys()]
return sum(data) / len(a) return sum(data) / len(a)
if type(a) == list and type(b) == list: if isinstance(a, list) and isinstance(b, list):
if len(a) == 0 and len(b) == 0: if len(a) == 0 and len(b) == 0:
return 1 return 1
if len(a) != len(b): if len(a) != len(b):
return match_rate_zero(a, b, key=key) return match_rate_zero(key)
data = [match_rate(a[i], b[i], key=f"{key}[{i}]") for i in range(len(a))] data = [match_rate(a[i], b[i], key=f"{key}[{i}]") for i in range(len(a))]
return sum(data) / len(a) return sum(data) / len(a)
if a == b: if a == b:
@ -114,7 +113,7 @@ def task_callback(file, thread=True):
data = pt.__dict__[cache["type"]].from_json(cache["raw"]) data = pt.__dict__[cache["type"]].from_json(cache["raw"])
rate = match_rate(data.to_dict(), json.loads(cache["raw"])) rate = match_rate(data.to_dict(), json.loads(cache["raw"]))
return rate, file return rate, file
except Exception as e: except Exception:
if thread: if thread:
return 0, file return 0, file
else: else:
@ -229,7 +228,7 @@ if __name__ == "__main__":
error_count += 1 error_count += 1
try: try:
logger.info(f"Try: Self UserByScreenName Test") logger.info("Try: Self UserByScreenName Test")
kwargs = get_kwargs("UserByScreenName", {"screen_name": "a810810931931"}) kwargs = get_kwargs("UserByScreenName", {"screen_name": "a810810931931"})
res = pt.UserApi(api_client).get_user_by_screen_name_with_http_info(**kwargs) res = pt.UserApi(api_client).get_user_by_screen_name_with_http_info(**kwargs)
data = res.data.to_dict() data = res.data.to_dict()
@ -244,7 +243,7 @@ if __name__ == "__main__":
error_count += 1 error_count += 1
try: try:
logger.info(f"Try: Self UserTweets Test") logger.info("Try: Self UserTweets Test")
kwargs = get_kwargs("UserTweets", {"userId": "1180389371481976833"}) kwargs = get_kwargs("UserTweets", {"userId": "1180389371481976833"})
res = pt.TweetApi(api_client).get_user_tweets_with_http_info(**kwargs) res = pt.TweetApi(api_client).get_user_tweets_with_http_info(**kwargs)
data = res.data.to_dict() data = res.data.to_dict()