改为类

common/ipgeo.py

@@ -1,11 +1,13 @@
 import zipfile
-import IP2Location
+from IP2Location import IP2Location
 from config.setting import data_storage_dir
 
 
+class IpGeoInfo(IP2Location):
+    def __init__(self):
         zip_path = data_storage_dir.joinpath("IP2LOCATION-LITE-DB3.BIN.ZIP")
         bin_path = data_storage_dir.joinpath("IP2LOCATION-LITE-DB3.BIN")
         if not bin_path.exists():
             zf = zipfile.ZipFile(zip_path)
             zf.extract('IP2LOCATION-LITE-DB3.BIN', data_storage_dir)
-IpGeoInfo = IP2Location.IP2Location(bin_path)
+        IP2Location.__init__(self, bin_path)

common/resolve.py

@@ -70,7 +70,7 @@ def deal_output(output_path):
     logger.log('INFOR', f'Processing resolved results')
     records = dict()  # 用来记录所有域名解析数据
     ip_asn = IPAsnInfo()
-    ip_geo = IpGeoInfo
+    ip_geo = IpGeoInfo()
     db_path = settings.data_storage_dir.joinpath('ip2region.db')
     ip_reg = IpRegInfo(db_path)
     with open(output_path) as fd: