One more fork rewrite

This commit is contained in:
Mint 2022-09-03 15:09:31 +03:00
parent 5613ee63a9
commit f0032a7146
2 changed files with 6 additions and 14 deletions

View file

@ -162,15 +162,11 @@ def get_type(domain: str) -> str:
if res.ok and "text/html" in res.headers["content-type"]: if res.ok and "text/html" in res.headers["content-type"]:
res = get(f"https://{domain}/nodeinfo/2.1", headers=headers, timeout=5) res = get(f"https://{domain}/nodeinfo/2.1", headers=headers, timeout=5)
if res.ok: if res.ok:
if res.json()["software"]["name"] == "akkoma": if res.json()["software"]["name"] in ["akkoma", "rebased"]:
return "pleroma" return "pleroma"
elif res.json()["software"]["name"] == "rebased": elif res.json()["software"]["name"] in ["hometown", "ecko"]:
return "pleroma"
elif res.json()["software"]["name"] == "hometown":
return "mastodon" return "mastodon"
elif res.json()["software"]["name"] == "ecko": elif res.json()["software"]["name"] in ["calckey", "groundpolis"]:
return "mastodon"
elif res.json()["software"]["name"] == "calckey":
return "misskey" return "misskey"
else: else:
return res.json()["software"]["name"] return res.json()["software"]["name"]

View file

@ -34,15 +34,11 @@ def get_type(instdomain: str) -> str:
if res.ok and "text/html" in res.headers["content-type"]: if res.ok and "text/html" in res.headers["content-type"]:
res = get(f"https://{instdomain}/nodeinfo/2.1", headers=headers, timeout=5) res = get(f"https://{instdomain}/nodeinfo/2.1", headers=headers, timeout=5)
if res.ok: if res.ok:
if res.json()["software"]["name"] == "akkoma": if res.json()["software"]["name"] in ["akkoma", "rebased"]:
return "pleroma" return "pleroma"
elif res.json()["software"]["name"] == "rebased": elif res.json()["software"]["name"] in ["hometown", "ecko"]:
return "pleroma"
elif res.json()["software"]["name"] == "hometown":
return "mastodon" return "mastodon"
elif res.json()["software"]["name"] == "ecko": elif res.json()["software"]["name"] in ["calckey", "groundpolis"]:
return "mastodon"
elif res.json()["software"]["name"] == "calckey":
return "misskey" return "misskey"
else: else:
return res.json()["software"]["name"] return res.json()["software"]["name"]