# -*- coding: utf-8 -*-
import json
import re
import time
import requests
from bs4 import BeautifulSoup
from kafka import KafkaProducer

import urllib3
urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

from base.BaseCore import BaseCore
baseCore = BaseCore()
cnx_ = baseCore.cnx
cursor_ = baseCore.cursor
log = baseCore.getLogger()

from classtool import Token, File, Tag
token = Token()
file = File()
tag = Tag()

# 发送数据
def sendkafka(post_data):
    try:
        producer = KafkaProducer(bootstrap_servers=['114.115.159.144:9092'], api_version=(2, 0, 2))
        kafka_result = producer.send("enterpriseInfo", json.dumps(post_data, ensure_ascii=False).encode('utf8'))
        print(kafka_result.get(timeout=10))
    except:
        exception = 'kafka传输失败'
        state = 0
        takeTime = baseCore.getTimeCost(start_time, time.time())
        baseCore.recordLog(social_code, taskType, state, takeTime, '', exception)
        log.info(f"{com_name}--{social_code}--kafka传输失败")

# 合并基本信息和工商信息字段
def getinfo(dict1,dict2):
    # 取出两个字典的key值集合
    keys1 = set(dict1.keys())
    keys2 = set(dict2.keys())

    # 取出并集
    union_keys = keys1 | keys2
    # 根据并集的key值，从两个字典中取出value值，组成新的字典
    result_dict = {key: dict1.get(key, None) or dict2.get(key, None) for key in union_keys}
    return result_dict

# 获取基本信息
def baseinfo(com_soup):
    baseinfo = com_soup.find('div', class_='contact-info')
    cominfo_list = baseinfo.find_all('span', class_='f')
    data = {}
    for cominfo in cominfo_list:
        # print(cominfo)
        value = cominfo.find('span', class_='val').text.replace('复制', '').strip(' ')
        pattern = r'\（\d{4}\s*年\）'
        match = re.search(pattern, value)
        if match:
            # print(match.group(0))
            value = value.split(match.group(0))[0]
        # print(value)

        tag.deletep(cominfo, 'span', 'class', 'val')
        tag.deletep(cominfo, 'a', '', '')
        tag.deletek(cominfo)
        # print(cominfo)

        name = cominfo.text.replace('\n', '').replace('复制', '').strip(' ').replace('：', '')
        # print(name,value)
        data[name] = value
    return data

# 检查登陆状态
def checklogin(key):

    # url = f'https://www.qcc.com/web/search?key=91110108558521630L'
    url = f'https://www.qcc.com/web/search?key={key}'
    req = requests.get(headers=headers, url=url)
    soup = BeautifulSoup(req.content, 'html.parser')
    if soup.find('title').text == '会员登录 - 企查查':
        log.info('状态---未登录')
        soup = ''
        return soup
    return soup

# 处理要发送的字段
def dic_handle(result_dic):
    zxss = ['北京市', '天津市', '上海市', '重庆市']
    try:
        company_name = result_dic['企业名称']
    except:
        company_name = ''

    try:
        CreditCode = result_dic['统一社会信用代码']
    except:
        CreditCode = ''

    try:
        OperName = result_dic['法定代表人']
    except:
        OperName = ''

    try:
        PhoneNumber = result_dic['电话']
    except:
        PhoneNumber = ''

    try:
        WebSite = result_dic['官网']
    except:
        WebSite = ''

    try:
        Email = result_dic['邮箱']
    except:
        Email = ''

    try:
        Desc = result_dic['简介']
    except:
        Desc = ''

    try:
        Status = result_dic['登记状态']
    except:
        Status = ''

    try:
        StartDate = result_dic['成立日期']
    except:
        StartDate = ''

    try:
        RecCap = result_dic['实缴资本']
    except:
        RecCap = ''

    try:
        RegistCapi = result_dic['注册资本']
    except:
        RegistCapi = ''

    try:
        CheckDate = result_dic['核准日期']
    except:
        CheckDate = ''

    try:
        OrgNo = result_dic['组织机构代码']
    except:
        OrgNo = ''

    try:
        No = result_dic['工商注册号']
    except:
        No = ''

    try:
        taxpayerNo = result_dic['纳税人识别号']
    except:
        taxpayerNo = ''

    try:
        EconKind = result_dic['企业类型']
    except:
        EconKind = ''

    try:
        TermStart = result_dic['营业期限'].split('至')[0]
    except:
        TermStart = ''

    try:
        TeamEnd = result_dic['营业期限'].split('至')[1]
    except:
        TeamEnd = ''

    try:
        TaxpayerType = result_dic['纳税人资质']
    except:
        TaxpayerType = ''

    try:
        SubIndustry = result_dic['国标行业']
    except:
        SubIndustry = ''

    try:
        region = result_dic['所属地区']
    except:
        region = ''
    try:
        pattern = r'^(.*?省|.*?自治区)?(.*?市|.*?自治州)?(.*?区|.*?县|.*?自治县|.*?市辖区)?(.*?区|.*?县|.*?自治县|.*?市辖区)?$'
        matches = re.match(pattern, region)
        Province = matches.group(1)
        City = matches.group(2)
        County = matches.group(3)
        if Province is None:
            for zxs in zxss:
                if zxs in region:
                    Province = zxs
                    break

    except:
        Province = ''
        City = ''
        County = ''

    try:
        BelongOrg = result_dic['登记机关']
    except:
        BelongOrg = ''

    try:
        Info = result_dic['人员规模']
    except:
        Info = ''

    try:
        can_bao = result_dic['参保人数']
    except:
        can_bao = ''

    try:
        OriginalName = result_dic['曾用名']
    except:
        OriginalName = ''

    try:
        EnglishName = result_dic['英文名']
    except:
        EnglishName = ''

    try:
        IxCode = result_dic['进出口企业代码']
    except:
        IxCode = ''

    try:
        Address = result_dic['地址']
    except:
        Address = ''

    try:
        Scope = result_dic['经营范围']
    except:
        Scope = ''

    aa_dict = {
        'name': company_name,  # 企业名称
        'shortName': '',  # 企业简称
        'socialCreditCode': CreditCode,  # 统一社会信用代码
        'legalPerson': OperName,  # 法定代表人
        'officialPhone': PhoneNumber,  # 电话
        'officialUrl': WebSite,  # 官网
        'officialEmail': Email,  # 邮箱
        'briefInfo': Desc,  # 简介
        'registerStatus': Status,  # 登记状态
        'incorporationDate': StartDate,  # 成立日期
        'capital': RegistCapi,  # 注册资本
        'paidCapital': RecCap,  # 实缴资本
        'approvalDate': CheckDate,  # 核准日期
        'organizationCode': OrgNo,  # 组织机构代码
        'registerNo': No,  # 工商注册号
        'taxpayerNo': taxpayerNo,  # 纳税人识别号
        'type': EconKind,  # 企业类型
        'businessStartDate': TermStart,  # 营业期限自
        'businessEndDate': TeamEnd,  # 营业期限至
        'taxpayerQualification': TaxpayerType,  # 纳税人资质
        'industry': SubIndustry,  # 所属行业
        'region': region,
        'province': Province,  # 所属省
        'city': City,  # 所属市
        'county': County,  # 所属县
        'registerDepartment': BelongOrg,  # 登记机关
        'scale': Info,  # 人员规模
        'insured': can_bao,  # 参保人数
        'beforeName': OriginalName,  # 曾用名
        'englishName': EnglishName,  # 英文名
        'importExportEnterpriseCode': IxCode,  # 进出口企业代码
        'address': Address,  # 地址
        'businessRange': Scope,  # 经营范围
        'status': 0,  # 状态
    }

    return aa_dict

# 采集准备
def redaytowork(com_name,social_code,securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name):

    # if social_code:
    #     dic_info = baseCore.getInfomation(social_code)
    # elif not social_code:
    #     dic_info = baseCore.getBYnameInfomation(com_name)
    # else:
    #     dic_info = ''
    # if dic_info:
    #     pass
    log.info(f'----当前企业{social_code}-{com_name}--开始处理---')
    # count = dic_info[14]
    count = 0
    # 企查查id
    # company_id = dic_info[12]
    # 如果没有信用代码 就通过名字搜索 如果有信用代码 就通过信用代码
    if social_code:
        soup = checklogin(social_code)
    else:
        soup = checklogin(com_name)
    if not soup:
        log.info("登录失效===重新放入redis")
        baseCore.r.lpush('BaseInfoEnterprise:gnqy_socialCode', company_field)
        token.delete_token(cookie_)
        log.info('=====已重新放入redis,失效cookies已删除======')
        time.sleep(20)
        return count
    else:
        searchinfo = soup.find_all('div', class_='npanel-heading')[1].find('span', class_='text-danger').text
        if searchinfo == '0':
            log.info('=====搜索不到该企业====')
            data = [com_name, social_code]
            # todo:搜不到的企业需要返回到一个表格中
            file.appenddata(file_name, '需处理企业', data)
            return count
        else:
            # 开始采集
            try:
                if spiderwork(soup, com_name, securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name):
                    count += 1
                    log.info(f'采集{com_name}成功=======耗时{baseCore.getTimeCost(start_time, time.time())}')
                    return count
                else:
                    return count
            except Exception as e:
                log.info(f'====={social_code}=====获取基本信息失败，重新放入redis=====')
                baseCore.r.lpush('BaseInfoEnterprise:gnqy_social_code', company_field)
                token.delete_token(cookie_)
                log.info('=====已重新放入redis,失效cookies已删除======')
                return count


def ifbeforename(company_url):

    req_ = requests.get(headers=headers, url=company_url)
    com_soup = BeautifulSoup(req_.content, 'html.parser')
    try:
        businessinfo = com_soup.find('div', class_='cominfo-normal')
    except:
        businessinfo = ''
    if businessinfo:
        try:
            name = businessinfo.find('div', class_='ntag text-gray original-tag').text
            value = businessinfo.find('div', class_='original-name-list').text.replace('展开', '').replace(' ', '').replace('…','').replace('\n', '').replace('复制', '').split('（')[0]
        except:
            name = '曾用名'
            value = ''
        return value
    else:
        return ''

# 采集基本信息和工商信息
def spiderinfo(company_url, securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name):
    qccid = company_url.split('firm/')[1].split('.html')[0]
    # 将采集到的企查查id更新
    updateSql = f"update EnterpriseInfo set QCCID = '{qccid}' where SocialCode = '{social_code}'"
    cursor_.execute(updateSql)
    cnx_.commit()
    req_ = requests.get(headers=headers, url=company_url)
    com_soup = BeautifulSoup(req_.content, 'html.parser')
    try:
        businessinfo = com_soup.find('div', class_='cominfo-normal')
    except:
        businessinfo = ''
    if businessinfo:
        data_businfo = {}
        data_baseinfo = baseinfo(com_soup)
        # print(data_baseinfo)
        try:
            name = businessinfo.find('div', class_='ntag text-gray original-tag').text
            value = \
                businessinfo.find('div', class_='original-name-list').text.replace('展开', '').replace(' ',
                                                                                                     '').replace(
                    '…',
                    '').replace(
                    '\n', '').replace('复制', '').split('（')[0]
        except:
            name = '曾用名'
            value = ''
        data_businfo[name] = value
        td_tags = businessinfo.find_all('td')
        # print(td_tags)

        for td in td_tags:

            if 'class' in td.attrs and 'tb' in td['class']:
                div_tags = td.find_all('div')
                texts = [div.text for div in div_tags]
                if len(texts) > 0:
                    for text in texts[::-1]:
                        data_businfo[text.replace('复制', '').replace('\n', '').strip(' ')] = None
                else:
                    data_businfo[td.text.replace('复制', '').replace('\n', '').strip(' ')] = None
            else:
                # 没有class='tb'属性的标签
                att_list = ['inline-block', 'ntag-v2', 'm-l-r-10', 'm-l-sm']
                for att in att_list:
                    tag.deletep(td, 'a', 'class', att)
                tag.deletek(td)
                tag.deletep(td, 'div', 'class', 'text-gray clearfix original-name-part')
                tag.deletespan(td)
                # if len(result_dict) <= len(td_tags) // 2:
                div_tags = td.find_all('div')
                texts = [div.text for div in div_tags if len(div.attrs) == 0]
                if len(texts) > 0:
                    i = 1
                    for text in texts:
                        if text == ' ':
                            continue
                        data_businfo[list(data_businfo.keys())[-i]] = text.replace('复制', '').replace('\n',
                                                                                                     '').replace(
                            ' ', '')
                        i += 1
                else:
                    if '实缴资本' in td.text:
                        # pattern = r"\d+万美元"
                        # match = re.search(pattern, td.text.replace('复制', '').replace('\n', '').replace(' ', ''))
                        # if match:
                        #     value = match.group()
                        value = td.text.replace('复制', '').replace('\n', '').replace(' ', '').split('实缴资本')[0]
                        data_businfo[list(data_businfo.keys())[-1]] = value

                    else:
                        data_businfo[list(data_businfo.keys())[-1]] = td.text.replace('复制', '').replace('\n','').replace(' ','')

        result_dict = getinfo(data_businfo, data_baseinfo)

        # print(result_dict)
        # 采集成功的企业
        data = [com_name, social_code, result_dict['统一社会信用代码']]
        file.appenddata(file_name, '获取基本信息成功企业', data)
        # 将字段转化成英文驼峰
        aa_dic = dic_handle(result_dict)
        aa_dic['qccId'] = qccid
        aa_dic['ynDomestic'] = ynDomestic
        aa_dic['countryName'] = countryName
        aa_dic['securitiesCode'] = securitiesCode
        aa_dic['securitiesShortName'] = securitiesShortName
        aa_dic['listingDate'] = listingDate
        aa_dic['category'] = category
        aa_dic['exchange'] = exchange

        print(aa_dic)
        sendkafka(aa_dic)

    else:
        data_baseinfo = baseinfo(com_soup)
        # 采集成功的企业
        data = [com_name, social_code, data_baseinfo['统一社会信用代码']]
        file.appenddata(file_name, '获取基本信息成功企业', data)
        # 将字段转化成英文驼峰
        aa_dic = dic_handle(data_baseinfo)
        aa_dic['qccId'] = qccid
        aa_dic['ynDomestic'] = ynDomestic
        aa_dic['countryName'] = countryName
        aa_dic['securitiesCode'] = securitiesCode
        aa_dic['securitiesShortName'] = securitiesShortName
        aa_dic['listingDate'] = listingDate
        aa_dic['category'] = category
        aa_dic['exchange'] = exchange

        sendkafka(aa_dic)

# 判断名称是否统一
def spiderwork(soup, receptname, securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name):
    company_url = ''
    try:
        company_list = soup.find('table', class_='app-ltable ntable ntable-list ntable ntable-list')
        tr_list = company_list.find_all('tr', class_='tsd0')
    except:
        log.info(f'====={social_code}=====获取基本信息失败，重新放入redis=====')
        baseCore.r.lpush('BaseInfoEnterprise:gnqy_social_code', company_field)
        token.delete_token(cookie_)
        log.info('=====已重新放入redis,失效cookie已删除======')
        return False

    # receptname = '小米通讯技术有限公司'
    for tr in tr_list:
        info_t = tr.find('span',class_='copy-title')
        getname = info_t.find('span').text
        log.info(f'接收到的企业名称--{receptname}---采到的企业名称--{getname}')

        if receptname and getname == receptname:
            company_url = info_t.find('a')['href']
            break
        elif not receptname:
            company_url = info_t.find('a')['href']
            break
        else:
            continue
    if company_url:
        # company_url = 'https://www.qcc.com/firm/80af5085726bb6b9c7770f1e4d0580f4.html'
        # company_url = 'https://www.qcc.com/firm/50f75e8a8859e609ec37976f8abe827d.html'
        # 采集基本信息和工商信息
        spiderinfo(company_url, securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name)
    else:
        # 判断是否是曾用名
        tr = tr_list[:1][0]
        info_t = tr.find('span', class_='copy-title')
        getname = info_t.find('span').text
        log.info(f'------可能是曾用名------接收到的企业名称--{receptname}---采到的企业名称--{getname}')
        company_url = info_t.find('a')['href']
        beforename = ifbeforename(company_url)
        if beforename == receptname:
            spiderinfo(company_url, securitiesCode, securitiesShortName, listingDate, category, exchange, ynDomestic, countryName, file_name)
        else:
            #没有搜到相同的企业名称
            data = [com_name, social_code]
            file.appenddata(file_name, '需处理企业',data)
            time.sleep(2)
            return False
    return True

if __name__ == '__main__':
    taskType = '基本信息/企查查'

    while True:
        nowtime = baseCore.getNowTime(1).replace('-', '')[:8]
        file_name = f'./data/国内企业基本信息采集情况_{nowtime}.xlsx'
        file.createFile(file_name)
        # TODO:需要隔两个小时左右抓包修改,token从数据库中获得
        cookies = token.getToken()
        print(type(cookies))
        if cookies:
            pass
        else:
            log.info('==========已无cookies==========')
            time.sleep(30)
            continue
        cookie_ = json.loads(cookies[0])
        print(type(cookie_))
        log.info(f"获取cookie到----{cookie_}")

        headers = {
            'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/avif,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
            'Accept-Encoding': 'gzip, deflate, br',
            'Accept-Language': 'zh-CN,zh;q=0.9',
            'Connection': 'keep-alive',
            # 'Cookie': 'qcc_did=046d99c9-566e-4046-9094-689901b79748; UM_distinctid=18aac5b8c21810-046f8431aecf58-26031f51-1fa400-18aac5b8c22efd; CNZZDATA1254842228=109635008-1695108795-https%253A%252F%252Fwww.qcc.com%252F%7C1695113473; _uab_collina=169935323766710839405007; acw_tc=db9062a717000200596487102e63dac7bed6aad2a049361c973816fabf; QCCSESSID=3c95642bd6445b7681c8fc6411',
            'Cookie': f'qcc_did={cookie_["qcc_did"]}; acw_tc={cookie_["acw_tc"]}; QCCSESSID={cookie_["QCCSESSID"]}',
            'Host': 'www.qcc.com',
            'Referer': 'https://www.qcc.com/',
            'Sec-Ch-Ua': '"Google Chrome";v="119", "Chromium";v="119", "Not?A_Brand";v="24"',
            'Sec-Ch-Ua-Mobile': '?0',
            'Sec-Ch-Ua-Platform': '"Windows"',
            'Sec-Fetch-Dest': 'document',
            'Sec-Fetch-Mode': 'navigate',
            'Sec-Fetch-Site': 'same-origin',
            'Sec-Fetch-User': '?1',
            'Upgrade-Insecure-Requests': '1',
            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36'
        }
        start_time = time.time()
        # 获取企业信息
        # company_field = baseCore.redicPullData('BaseInfoEnterprise:gnqy_socialCode')
        company_field = '91220101606092819L||'
        if company_field == 'end':
            # 本轮处理完毕，需要发送邮件，并且进入下一轮
            baseCore.sendEmail(file_name)
            time.sleep(20)
            file.deleteFile(file_name)
            continue

        if company_field == '' or company_field is None:
            # 本轮结束后没有新增的企业要采集
            file.deleteFile(file_name)
            flag = True
            while flag:
                time.sleep(5400)
                company_field = baseCore.redicPullData('BaseInfoEnterprise:gnqy_socialCode')
                if company_field:
                    flag = False
                    baseCore.r.lpush('BaseInfoEnterprise:gnqy_socialCode', company_field)
                    continue
            continue

        social_code = company_field.split('|')[0]
        com_name = company_field.split('|')[2]

        ynDomestic = company_field.split('|')[15]
        countryName = company_field.split('|')[16]
        securitiesCode = company_field.split('|')[17]
        securitiesShortName = company_field.split('|')[18]
        listingDate = company_field.split('|')[21]
        category = company_field.split('|')[19]
        exchange = company_field.split('|')[20]
        # ynDomestic = ''
        # countryName = ''
        # securitiesCode = ''
        # securitiesShortName = ''
        # listingDate = ''
        # category = ''
        # exchange = ''


        count = redaytowork(com_name, social_code, securitiesCode, securitiesShortName, listingDate, category, exchange,ynDomestic, countryName, file_name)
        # baseCore.r.close()
        # break
        # baseCore.sendEmail(file_name)
        # 信息采集完成后将该企业的采集次数更新
        # runType = 'BaseInfoRunCount'
        # baseCore.updateRun(social_code, runType, count)

    baseCore.close()