Merge branch 'development'

service/Sql_cheat.py

@@ -2,6 +2,7 @@ import pandas as pd
 from config.concat_sql import *
 import re
 import requests
+from utils.phone import *
 from utils import tools
 import datetime
 from functools import reduce
@@ -136,9 +137,6 @@ def requisition_quantity_business_flow_new(uuid,apply_time):
 
 
 
-
-from phone import Phone
-
 def get_phone_info(phone_num):
     """根据手机号查询省份"""
     phone_info = Phone().find(phone_num)
@@ -153,8 +151,7 @@ def get_phone_info(phone_num):
         print('----- the phone number is empty, please check your phone number! -----')
     return phone, province, city, zip_code, area_code, phone_type
 
-
-
+# print(get_phone_info('18631179951'))
 
 def city_dict_df():
     """标准的归属地码表"""

utils/phone/__init__.py

+from .phone import Phone
+
+__all__ = ['Phone']

utils/phone/phone.py

+# -*- coding: utf-8 -*-
+
+import os
+import struct
+import sys
+
+__author__ = 'lovedboy'
+
+if sys.version_info > (3, 0):
+    def get_record_content(buf, start_offset):
+        end_offset = start_offset + buf[start_offset:-1].find(b'\x00')
+        return buf[start_offset:end_offset].decode()
+else:
+    def get_record_content(buf, start_offset):
+        end_offset = start_offset + buf[start_offset:-1].find('\x00')
+        return buf[start_offset:end_offset]
+
+
+class Phone(object):
+    def __init__(self, dat_file=None):
+
+        if dat_file is None:
+            dat_file = os.path.join(os.path.dirname(__file__), "phone.dat")
+
+        with open(dat_file, 'rb') as f:
+            self.buf = f.read()
+
+        self.head_fmt = "<4si"
+        self.phone_fmt = "<iiB"
+        self.head_fmt_length = struct.calcsize(self.head_fmt)
+        self.phone_fmt_length = struct.calcsize(self.phone_fmt)
+        self.version, self.first_phone_record_offset = struct.unpack(
+            self.head_fmt, self.buf[:self.head_fmt_length])
+        self.phone_record_count = (len(
+            self.buf) - self.first_phone_record_offset) / self.phone_fmt_length
+
+    def get_phone_dat_msg(self):
+        print("版本号:{}".format(self.version))
+        print("总记录条数:{}".format(self.phone_record_count))
+
+    @staticmethod
+    def get_phone_no_type(no):
+        if no == 4:
+            return "电信虚拟运营商"
+        if no == 5:
+            return "联通虚拟运营商"
+        if no == 6:
+            return "移动虚拟运营商"
+        if no == 3:
+            return "电信"
+        if no == 2:
+            return "联通"
+        if no == 1:
+            return "移动"
+
+    @staticmethod
+    def _format_phone_content(phone_num, record_content, phone_type):
+
+        province, city, zip_code, area_code = record_content.split('|')
+        return {
+            "phone": phone_num,
+            "province": province,
+            "city": city,
+            "zip_code": zip_code,
+            "area_code": area_code,
+            "phone_type": Phone.get_phone_no_type(phone_type)
+        }
+
+    def _lookup_phone(self, phone_num):
+
+        phone_num = str(phone_num)
+        assert 7 <= len(phone_num) <= 11
+        int_phone = int(str(phone_num)[0:7])
+
+        left = 0
+        right = self.phone_record_count
+        while left <= right:
+            middle = int((left + right) / 2)
+            current_offset = int(self.first_phone_record_offset + middle * self.phone_fmt_length)
+            if current_offset >= len(self.buf):
+                return
+
+            buffer = self.buf[current_offset: current_offset + self.phone_fmt_length]
+            cur_phone, record_offset, phone_type = struct.unpack(self.phone_fmt,
+                                                                 buffer)
+
+            if cur_phone > int_phone:
+                right = middle - 1
+            elif cur_phone < int_phone:
+                left = middle + 1
+            else:
+                record_content = get_record_content(self.buf, record_offset)
+                return Phone._format_phone_content(phone_num, record_content,
+                                                   phone_type)
+
+    def find(self, phone_num):
+        return self._lookup_phone(phone_num)
+
+    @staticmethod
+    def human_phone_info(phone_info):
+        if not phone_info:
+            return ''
+
+        return "{}|{}|{}|{}|{}|{}".format(phone_info['phone'],
+                                          phone_info['province'],
+                                          phone_info['city'],
+                                          phone_info['zip_code'],
+                                          phone_info['area_code'],
+                                          phone_info['phone_type'])
+
+    def test(self):
+        self.get_phone_dat_msg()
+        for i in range(1529900, 1529999):
+            print(self.human_phone_info(self.find(i)))
+
+
+if __name__ == "__main__":
+    phone = Phone()
+    phone.test()