テストサイトベタ書きを修正

cfedge.py

@@ -5,11 +5,6 @@ import requests
 import os
 import argparse
 
-class CFSite:
-    def __init__(self, url, plan):
-        self.url = url
-        self.plan = plan
-
 parser = argparse.ArgumentParser(description='Check Cloudflare Edge Locations and post to Misskey.')
 parser.add_argument('--dry-run', action='store_true', help='Run without posting to Misskey')
 args = parser.parse_args()
@@ -20,18 +15,13 @@ else:
     print("config.py not found")
     exit(1)
 
-test_sites = [
-    CFSite('misskey.io', 'Enterprise'),
-    CFSite('misskey.systems', 'Pro'),
-    CFSite('ohtrmi.cbrx.io', 'Free'),
-]
-
 result_txt = []
 result_txt.append('Cloudflare Edge Location Check')
 result_txt.append('')
 
 # read from /cdn-cgi/trace
-for site in test_sites:
+for site in config.test_sites:
+    planstr = f' ({site.plan})' if site.plan else ''
     try:
         response = requests.get(f"https://{site.url}/cdn-cgi/trace", timeout=5)
         response.raise_for_status()
@@ -40,9 +30,9 @@ for site in test_sites:
         if colo_line:
             colo = colo_line.split('=')[1]
             if colo in config.expected_edge_locations:
-                result_txt.append(f'- {site.url} ({site.plan}) : {colo}✅️')
+                result_txt.append(f'- {site.url}{planstr} : {colo}✅️')
             else:
-                result_txt.append(f'- {site.url} ({site.plan}) : {colo}⚠')
+                result_txt.append(f'- {site.url}{planstr} : {colo}⚠')
         else:
             print(f"Could not find 'colo' in the trace info for {site.url}")
     except requests.RequestException as e: