代码优化&新增ip

ipdb.txt

@@ -47,5 +47,6 @@
 221.6.171.0/24
 221.7.251.0/24
 221.205.168.0/23
+222.94.227.0/24
 222.125.57.0/24
 222.189.163.0/24

main.py

@@ -3,14 +3,16 @@
 # 将ipdb.txt备份为ipdb_日期_时间.txt 格式
 import datetime
 
-date = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')
-with open('ipdb.txt', 'r') as f:
+def main(filename='ipdb.txt'):
+    date = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')
+    with open(filename, 'r') as f:
         lines = f.readlines()
-    with open(f'ipdb_{date}.txt', 'w') as f:
+        print(f"{filename.split('.')[0]}_{date}.txt")
+        with open(f"{filename.split('.')[0]}_{date}.txt", 'w') as f:
             f.write(''.join(lines))
             print('备份完成！')
 
-with open('ipdb.txt', 'r') as f:
+    with open(filename, 'r') as f:
         # 读取内容，并将每行的数据进行排序
         lines = f.readlines()
         lines_without_newlines = [line.strip() for line in lines]
@@ -26,6 +28,11 @@ with open('ipdb.txt', 'r') as f:
         # 打印lines_without_newlines的长度
         print(len(lines_without_newlines))
         # 将排序后的内容写入文件，要求每个元素一行
-    with open('ipdb.txt', 'w') as f:
+        with open(filename, 'w') as f:
             f.write('\n'.join(lines_without_newlines))
         print('排序完成，更新完成！')
+
+
+
+if __name__ == '__main__':
+    main()