sim1222/fedi-block-api
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

added better way of getting the software type

Browse Source
This commit is contained in:
Enju Aihara 2022-04-08 22:07:05 +02:00
parent f1c3126c0a
commit 433bcd3f00
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
fetch_blocks.py
View File

@ -30,17 +30,17 @@ def get_mastodon_blocks(domain: str) -> dict:
def get_type(domain: str) -> str:
try:
res = get("https://"+domain, headers=headers, timeout=5)
if "pleroma" in res.text.lower():
print("pleroma")
return "pleroma"
elif "mastodon" in res.text.lower():
print("mastodon")
res = get(f"https://{domain}/nodeinfo/2.1.json", headers=headers, timeout=5)
if res.status_code == 404:
res = get(f"https://{domain}/nodeinfo/2.0.json", headers=headers, timeout=5)
if res.ok:
return res.json()["software"]["name"]
elif res.status_code == 404:
res = get(f"https://{domain}/api/v1/instance", headers=headers, timeout=5)
if res.ok:
return "mastodon"
return ""
except Exception as e:
print("error:", e, domain)
return ""
except:
return None
conn = sqlite3.connect("blocks.db")
c = conn.cursor()