import json

from kafka import KafkaProducer

import requests, time,  fitz
import urllib3
from base import BaseCore

urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)

baseCore = BaseCore.BaseCore()

cnx_ = baseCore.cnx_
cursor_ = cnx_.cursor()

cnx = baseCore.cnx
cursor = baseCore.cursor

def sendKafka(dic_news):
    try:  # 114.116.116.241
        producer = KafkaProducer(bootstrap_servers=['114.115.159.144:9092'],max_request_size=1024*1024*20)
        kafka_result = producer.send("researchReportYearTopic",
                                     json.dumps(dic_news, ensure_ascii=False).encode('utf8'))

        print(kafka_result.get(timeout=10))

        dic_result = {
            'success': 'ture',
            'message': '操作成功',
            'code': '200',
        }
        log.info(dic_result)
        return True
    except Exception as e:
        dic_result = {
            'success': 'false',
            'message': '操作失败',
            'code': '204',
            'e': e
        }
        log.info(dic_result)
        return False
#state1
if __name__ == '__main__':
    header = {
        'Connection': 'keep-alive',
        'Cache-Control': 'max-age=0',
        'sec-ch-ua': '"Chromium";v="112", "Microsoft Edge";v="112", "Not:A-Brand";v="99"',
        'sec-ch-ua-mobile': '?0',
        'sec-ch-ua-platform': '"Windows"',
        'Upgrade-Insecure-Requests': '1',
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/112.0.0.0 Safari/537.36 Edg/112.0.1722.64',
        'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7',
        'Sec-Fetch-Site': 'none',
        'Sec-Fetch-Mode': 'navigate',
        'Sec-Fetch-User': '?1',
        'Sec-Fetch-Dest': 'document',
        'Accept-Encoding': 'gzip, deflate, br',
        'Accept-Language': 'zh-CN,zh;q=0.9,en;q=0.8,en-GB;q=0.7,en-US;q=0.6',
    }
    query = "SELECT * FROM clb_sys_attachment WHERE type_id=1 AND create_time >= '2023-10-27' AND create_time <= '2023-10-28' AND source like '%官网' "
    cursor_.execute(query)
    results = cursor_.fetchall()
    for result in results:
        att_id = result[0]
        name = result[1]
        source = result[18]
        social_code = result[3]
        # selectcom = f"select * from EnterpriseInfo where SocialCode = '{social_code}'"
        # cursor.execute(selectcom)
        # selects = cursor.fetchone()
        # com_name = selects[4]
        # if com_name:
        #     pass
        # else:
        #     com_name = selects[1]
        full_path = 'http://zzsn.luyuen.com/' + result[19]
        year = result[9]
        create_time = result[13]
        publish = str(result[21])
        content = ''
        for i in range(0, 3):
            try:
                response = requests.get(url=full_path, headers=header, timeout=30)
                break
            except Exception as e:
                time.sleep(3)
                continue
        with fitz.open(stream=response.content, filetype='pdf') as doc:
            page_size = doc.page_count
            log = baseCore.getLogger()
            log.info(f'当前页码----{page_size}')
            for page in doc.pages():
                content += page.get_text()
        detect_language = baseCore.detect_language(content)
        dic_news = {
            'attachmentIds': att_id,
            'author': '',
            'content': content,
            'contentWithTag': '',
            'createDate': str(create_time),
            'deleteFlag': '0',
            'id': '',
            'keyWords': '',
            'lang': detect_language,
            'origin': source,
            # 'origin': '雪球网',
            'publishDate': publish,
            'sid': '1684032033495392257',
            'sourceAddress': '',  # 原文链接
            'summary': '',
            'title': name.replace('.pdf',''),
            'type': 1,
            'socialCreditCode': social_code,
            'year': year
        }
        if sendKafka(dic_news):
            # 100表示成功
            log.info(f'==========={social_code}成功============')

    cnx.close()
    cursor_.close()
    baseCore.close()



