perf: 优化使用两个 ip 库

apps/common/utils/ip/ipip/utils.py

@@ -0,0 +1,23 @@
+# -*- coding: utf-8 -*-
+#
+import os
+from django.utils.translation import ugettext as _
+
+import ipdb
+
+__all__ = ['get_ip_city_by_ipip']
+ipip_db = None
+
+
+def get_ip_city_by_ipip(ip):
+    global ipip_db
+    if not ip or not isinstance(ip, str):
+        return _("Invalid ip")
+    if ':' in ip:
+        return 'IPv6'
+    if ipip_db is None:
+        ipip_db_path = os.path.join(os.path.dirname(__file__), 'ipipfree.ipdb')
+        ipip_db = ipdb.City(ipip_db_path)
+
+    info = ipip_db.find_info(ip, 'CN')
+    return {'city': info.city_name, 'country': info.country_name}