chore:fafa region

Guovin · Sep 5, 2024 · 64777fc · 64777fc
1 parent fdb54ed
commit 64777fc
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/updates/fofa/request.py b/updates/fofa/request.py
@@ -20,15 +20,15 @@ def get_fofa_urls_from_region_list():
     """
     Get the FOFA url from region
     """
-    # region_list = config.get("Settings", "hotel_region_list").split(",")
+    region_list = config.get("Settings", "hotel_region_list").split(",")
     urls = []
     region_url = getattr(fofa_map, "region_url")
-    # if "all" in region_list or "ALL" in region_list or "全部" in region_list:
-    urls = [url for url_list in region_url.values() for url in url_list if url]
-    # else:
-    #     for region in region_list:
-    #         if region in region_url:
-    #             urls.append(region_url[region])
+    if "all" in region_list or "ALL" in region_list or "全部" in region_list:
+        urls = [url for url_list in region_url.values() for url in url_list if url]
+    else:
+        for region in region_list:
+            if region in region_url:
+                urls.append(region_url[region])
     return urls