generate name randomly if empty

shipeng101 · Jul 17, 2024 · 4a89e97 · 4a89e97
1 parent 4610038
commit 4a89e97
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/subscribe/location.py b/subscribe/location.py
@@ -6,8 +6,10 @@
 import json
 import math
 import os
+import random
 import re
 import socket
+import string
 import urllib
 from collections import defaultdict
 
@@ -162,6 +164,9 @@ def regularize(
     records = defaultdict(list)
     for proxy in proxies:
         name = re.sub(r"(\d+|(\d+)?(-\d+)?[A-Z])$", "", proxy.get("name", "")).strip()
+        if not name:
+            name = "".join(random.sample(string.ascii_uppercase, 6))
+
         proxy["name"] = name
         records[name].append(proxy)
 

diff --git a/tools/clean.py b/tools/clean.py
@@ -7,7 +7,9 @@
 import argparse
 import math
 import os
+import random
 import re
+import string
 from collections import defaultdict
 
 import yaml
@@ -61,6 +63,9 @@ def main(args: argparse.Namespace) -> None:
                 records.add(key)
 
                 name = re.sub(r"(\d+|(\d+)?(-\d+)?[A-Z])$", "", item.get("name", "")).strip()
+                if not name:
+                    name = "".join(random.sample(string.ascii_uppercase, 6))
+
                 item["name"] = name
 
                 if args.secure: