One more fork rewrite

[fba.git] / fetch_instances.py

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