import random

import requests
import json
import csv

"""
随机生成手机号，按省份进行分类，存入csv文件
"""

# 调360的免费接口查询手机号归属地
def test_script(phone):
    url = 'http://cx.shouji.360.cn/phonearea.php?number='
    headers = {
        'Content-Type': 'application/json;charset=UTF-8',
        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/94.0.4606.61 Safari/537.36"

    }
    params = {'number': phone}
    res = requests.get(url=url, params=params,headers=headers)
    data = json.loads(res.text)['data']
    print(data)
    return data


# 随机生成手机号，并按省份进行分类
def generate_phone_number():
    pref_list = ['130', '131', '132', '133', '134', '135', '136', '137', '138', '139', '140', '145', '146', '147',
                 '149', '150', '151', '152', '153', '155', '156', '157', '158', '159', '162', '165', '166', '167',
                 '170', '171', '172', '173', '174', '175', '176', '177', '178', '180', '181', '182', '183', '184',
                 '185', '186', '187', '188', '189', '190', '191', '192', '193', '195', '196', '197', '198', '199']
    data_list_zhejiang = []
    data_list_jiangsu = []
    data_list_guangdong = []
    data_list_fujian = []
    data_list_hunan = []
    data_list_hubei = []
    data_list_liaoning = []
    data_list_jilin = []
    data_list_heilongjiang = []
    data_list_hebei = []
    data_list_henan = []
    data_list_shandong = []
    data_list_shuanxi = []
    data_list_shanxi = []
    data_list_gansu = []
    data_list_qinhai = []
    data_list_xinjiang = []
    data_list_sichuan = []
    data_list_guizhou = []
    data_list_anhui = []
    data_list_jiangxi = []
    data_list_yunnan = []
    data_list_neimenggu = []
    data_list_guangxi = []
    data_list_xizang = []
    data_list_ningxia = []
    data_list_hainan = []
    data_list_beijing = []
    data_list_shanghai = []
    data_list_tianjing = []
    data_list_chongqing = []
    for j in pref_list:
        for i in range(0, 200):
            a = str(random.randint(0, 100000000))
            b = a.zfill(8)
            phone = j+b
            print(j+b)
            data = test_script(phone)
            data['phone'] = str(phone)
            province = data['province']
            if province == '浙江':
                data_list_zhejiang.append(data)
            if province == '江苏':
                data_list_jiangsu.append(data)
            if province == '广东':
                data_list_guangdong.append(data)
            if province == '福建':
                data_list_fujian.append(data)
            if province == '湖南':
                data_list_hunan.append(data)
            if province == '湖北':
                data_list_hubei.append(data)
            if province == '辽宁':
                data_list_liaoning.append(data)
            if province == '吉林':
                data_list_jilin.append(data)
            if province == '黑龙江':
                data_list_heilongjiang.append(data)
            if province == '河北':
                data_list_hebei.append(data)
            if province == '河南':
                data_list_henan.append(data)
            if province == '山东':
                data_list_shandong.append(data)
            if province == '陕西':
                data_list_shuanxi.append(data)
            if province == '山西':
                data_list_shanxi.append(data)
            if province == '甘肃':
                data_list_gansu.append(data)
            if province == '青海':
                data_list_qinhai.append(data)
            if province == '新疆':
                data_list_xinjiang.append(data)
            if province == '四川':
                data_list_sichuan.append(data)
            if province == '贵州':
                data_list_guizhou.append(data)
            if province == '安徽':
                data_list_anhui.append(data)
            if province == '江西':
                data_list_jiangxi.append(data)
            if province == '云南':
                data_list_yunnan.append(data)
            if province == '广西':
                data_list_guangxi.append(data)
            if province == '内蒙古':
                data_list_neimenggu.append(data)
            if province == '西藏':
                data_list_xizang.append(data)
            if province == '宁夏':
                data_list_ningxia.append(data)
            if province == '海南省':
                data_list_hainan.append(data)
            if province == '北京':
                data_list_beijing.append(data)
            if province == '上海':
                data_list_shanghai.append(data)
            if province == '重庆':
                data_list_chongqing.append(data)
            if province == '天津':
                data_list_tianjing.append(data)
    save(data_list_zhejiang, 'province/浙江.csv')
    save(data_list_jiangsu, 'province/江苏.csv')
    save(data_list_guangdong, 'province/广东.csv')
    save(data_list_fujian, 'province/福建.csv')
    save(data_list_hunan, 'province/湖南.csv')
    save(data_list_hubei, 'province/湖北.csv')
    save(data_list_liaoning, 'province/辽宁.csv')
    save(data_list_jilin, 'province/吉林.csv')
    save(data_list_heilongjiang, 'province/黑龙江.csv')
    save(data_list_hebei, 'province/河北.csv')
    save(data_list_henan, 'province/河南.csv')
    save(data_list_shandong, 'province/山东.csv')
    save(data_list_shuanxi, 'province/陕西.csv')
    save(data_list_shanxi, 'province/山西.csv')
    save(data_list_gansu, 'province/甘肃.csv')
    save(data_list_qinhai, 'province/青海.csv')
    save(data_list_xinjiang, 'province/新疆.csv')
    save(data_list_sichuan, 'province/四川.csv')
    save(data_list_guizhou, 'province/贵州.csv')
    save(data_list_anhui, 'province/安徽.csv')
    save(data_list_jiangxi, 'province/江西.csv')
    save(data_list_yunnan, 'province/云南.csv')
    save(data_list_neimenggu, 'province/内蒙古.csv')
    save(data_list_guangxi, 'province/广西.csv')
    save(data_list_xizang, 'province/西藏.csv')
    save(data_list_ningxia, 'province/宁夏.csv')
    save(data_list_hainan, 'province/海南.csv')
    save(data_list_beijing, 'province/北京.csv')
    save(data_list_shanghai, 'province/上海.csv')
    save(data_list_tianjing, 'province/重庆.csv')
    save(data_list_chongqing, 'province/天津.csv')


def save(data, path):
    header = ['phone', 'province', 'city', 'sp']
    with open(path,'w',encoding='utf8', newline='') as f:
        # 写
        writer = csv.DictWriter(f, header)
        # 写入标题
        writer.writeheader()
        # 将数据写入
        # writer = csv.writer(f)
        writer.writerows(data)


if __name__ == '__main__':
    generate_phone_number()