# -*- coding: utf-8 -*-
# Author  : Charley
# Python  : 3.8.10
# Date: 2024-10-14 11:01
import json
import time
import requests
import schedule
import user_agent
import concurrent.futures
from loguru import logger
from retrying import retry
from parsel import Selector
from datetime import datetime

from mysq_pool import MySQLConnectionPool

logger.remove()
logger.add("logs/{time:YYYYMMDD}.log", encoding='utf-8', rotation="00:00",
           format="[{time:YYYY-MM-DD HH:mm:ss.SSS}] {level} {message}",
           level="DEBUG", retention="7 day")


@retry(stop_max_attempt_number=3, wait_fixed=2000)
def get_proxys_(log):
    """
    获取代理
    :return: 代理
    """
    tunnel = "x371.kdltps.com:15818"
    kdl_username = "t13753103189895"
    kdl_password = "o0yefv6z"
    try:
        proxies = {
            "http": "http://%(user)s:%(pwd)s@%(proxy)s/" % {"user": kdl_username, "pwd": kdl_password, "proxy": tunnel},
            "https": "http://%(user)s:%(pwd)s@%(proxy)s/" % {"user": kdl_username, "pwd": kdl_password, "proxy": tunnel}
        }
        return proxies
    except Exception as e:
        log.error(f"Error getting proxy: {e}")
        raise e


@retry(stop_max_attempt_number=5, wait_fixed=1000)
def get_proxys():
    # 已购买账户  北美
    http_proxy = "http://u1952150085001297:sJMHl4qc4bM0@proxy.123proxy.cn:36928"
    https_proxy = "http://u1952150085001297:sJMHl4qc4bM0@proxy.123proxy.cn:36928"

    try:
        proxySettings = {
            "http": http_proxy,
            "https": https_proxy,
        }
        return proxySettings
    except Exception as e:
        logger.error(f"Error getting proxy: {e}")
        raise e


@retry(stop_max_attempt_number=3, wait_fixed=2000)
def get_price(ucid):
    # ucid = '204224'
    headers = {
        "accept": "application/json, text/plain, */*",
        "referer": "https://robograding.com/feed/00205223/view",
        "user-agent": user_agent.generate_user_agent()
        # "x-xsrf-token": "eyJpdiI6ImVBTksreFRBejJXS0tRWmQvUERjRVE9PSIsInZhbHVlIjoiWXA0VktNMGFpQ0RydkR0SmZrVDNhU0ZCQTU3T0pMcExIUFI0VkFkSm85NURGeEdNUXJKU1hxdlIwOHJkbXJGdVcrMXlpVlJPUDRUR3dBV0ZaQ0d2bFhHeTQzaG40dDFMb1lBSVpmcTF6cDh6UHVwTEZEZFViYXo1RnVjQ0dCQ2siLCJtYWMiOiJjYTdkNDJhMDkyNmE3ZmVjZmRiMWY2ZTBjMDBhNzJjMjhkZWYwY2M4NjAwNDAxMTU0ZmI5YjE4NWZhNTVhNWM3IiwidGFnIjoiIn0="
    }
    url = f"https://robograding.com/api/v3/card-price/{ucid}"
    response = requests.get(url, headers=headers, proxies=get_proxys())
    # response = requests.get(url, headers=headers)

    if response.status_code != 200:
        logger.debug('请求失败，重试......................')
        raise Exception('请求失败，重试......................')
    price = response.json().get('price')
    return price


def transform_date(date_str):
    """
    November 13, 2020 类型的日期字符串 转换成年月日格式
    :param date_str: 日期字符串
    :return: formatted_date
    """
    # 解析日期字符串
    date_obj = datetime.strptime(date_str, "%B %d, %Y")

    # 格式化日期
    formatted_date = date_obj.strftime("%Y-%m-%d")
    return formatted_date


def save_data(sql_pool, info):
    sql = "INSERT INTO ags_record (cert_id, name, title, score, card_type, release_date, series, card, ags_set, owner, centering_overall, surface_overall, edges_overall, corners_overall, price, front_img, back_img) VALUES (%s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s, %s)"
    sql_pool.insert_one(sql, info)


def parse_data(cert_id, resp_text, sql_pool):
    selector = Selector(text=resp_text)
    name = selector.xpath('//div[@class="feed-view__header__content"]//h1/text()').get()
    title = selector.xpath('//div[@class="feed-view__header__content"]//h2/text()').get()
    score = selector.xpath(
        '//div[@class="feed-view__header__content"]//p[@class="feed-view__header__grade-score"]/text()').get()

    tr_list = selector.xpath('//div[@class="feed-view__right-side"]//tbody/tr[not(contains(@class, "feed-view"))]')
    result_dict = {'card_type': '',
                   'release_date': '',
                   'series': '',
                   'card': '',
                   'set': '',
                   'owner': ''}
    for tr in tr_list:
        item_key = tr.xpath('./td[1]/h3/text()').get()
        item_val = tr.xpath('./td[2]/text()').get()
        try:
            item_key = item_key.strip().replace(':', '')
            item_val = item_val.strip().replace(':', '')
            # print('item_key,item_val', item_key, item_val)
            key_snake_case = item_key.replace(' ', '_').replace('/', '_').lower()
            if key_snake_case in list(result_dict.keys()):
                # print(key_snake_case, item_val)
                result_dict[key_snake_case] = item_val
        except Exception as e:
            logger.debug(e)

    # overall
    centering_overall = selector.xpath(
        '//div[@class="feed-view__breakdown__scores feed-view__breakdown__scores--contained"]/div[1]/p[2]/text()').get()
    surface_overall = selector.xpath(
        '//div[@class="feed-view__breakdown__scores feed-view__breakdown__scores--contained"]/div[2]/p[2]/text()').get()
    edges_overall = selector.xpath(
        '//div[@class="feed-view__breakdown__scores feed-view__breakdown__scores--contained"]/div[3]/p[2]/text()').get()
    corners_overall = selector.xpath(
        '//div[@class="feed-view__breakdown__scores feed-view__breakdown__scores--contained"]/div[4]/p[2]/text()').get()

    # 获取用户id
    user_card_id = selector.xpath(
        '//div[@class="feed-view__breakdown__scores-holder feed-view__breakdown__scores-card-price"]/div/@data-user-card-id').get()
    if user_card_id:
        price = get_price(user_card_id)
    else:
        price = None

    img_json = selector.xpath(
        '//section[@class="feed-view__content"]//div[@class="feed-view__card"]/div/@data-images').get()
    img_dict = json.loads(img_json)
    front_img = img_dict.get('front_slab_image')
    back_img = img_dict.get('back_slab_image')
    if not front_img:
        front_img = img_dict.get('image_path')
        back_img = None

    info = (cert_id, name, title, score, result_dict.get('card_type'), transform_date(result_dict.get('release_date')),
            result_dict.get('series'), result_dict.get('card'), result_dict.get('set'), result_dict.get('owner'),
            centering_overall, surface_overall, edges_overall, corners_overall, price, front_img, back_img)
    # print(info)
    save_data(sql_pool, info)


@retry(stop_max_attempt_number=5, wait_fixed=2000)
def get_data(ags_id_list, sql_pool):
    sql_id = ags_id_list[0]
    cert_id = ags_id_list[1]
    headers = {
        "user-agent": user_agent.generate_user_agent()
    }
    url = f"https://robograding.com/feed/{cert_id}/view"

    try:
        response = requests.get(url, headers=headers, proxies=get_proxys())
        # response = requests.get(url, headers=headers)
        response.raise_for_status()

        if "Grades are not available yet" in response.text:
            logger.debug("Grades are not available yet in response.text......................")
            # 更新数据库状态为未完成
            sql_pool.update_one("UPDATE ags_task SET state=2 WHERE id=%s", (sql_id,))
        else:
            parse_data(cert_id, response.text, sql_pool)
            # 更新数据库状态为已完成
            sql_pool.update_one("UPDATE ags_task SET state=1 WHERE id=%s", (sql_id,))
    except requests.RequestException as e:
        logger.error(f"Request error: {e}")
        raise  # 可以选择重新抛出异常以便外部处理


def process_urls(ids, mysql_pool, batch_size=1000, max_workers=5):
    with concurrent.futures.ThreadPoolExecutor(max_workers=max_workers) as executor:
        for i in range(0, len(ids), batch_size):
            batch = ids[i:i + batch_size]
            try:
                futures_to_urls = {executor.submit(get_data, url, mysql_pool): url for url in batch}
                for future in concurrent.futures.as_completed(futures_to_urls):
                    url = futures_to_urls[future]
                    try:
                        future.result()
                        logger.debug(f"处理 {url} 成功")
                    except Exception as exc:
                        logger.debug(f"处理 {url} 出错: {exc}")
            except Exception as e:
                logger.error(f"提交任务失败: {e}")


# def get_add_cert(start_, end_):
#     num_list = []
#     # 循环生成8位数的字符串
#     for num in range(start_, end_ + 1):
#         # 使用zfill将数字转换为字符串，并确保长度为8位，不足部分用0填充
#         formatted_num = str(num).zfill(8)
#         # print(formatted_num)
#         num_list.append(formatted_num)
#     return num_list


def get_new_task(sql_pool):
    # 查询最后一条数据  并且 +3000  将新任务插入表中
    # max_cert = sql_pool.select_one("SELECT cert_id FROM ags_record ORDER BY id DESC LIMIT 1")
    # start_max_cert = int(max_cert[0]) + 1
    # end_max_cert = start_max_cert + 3000
    # new_id_list = get_add_cert(start_max_cert, end_max_cert)
    # logger.debug(f'查询到最新的 id 为:{max_cert[0]}, 开始生成新数据, 并添加到任务表中.........')
    #
    # # 防止重复添加任务
    # state0_id_list = sql_pool.select_all("SELECT cert_id FROM ags_task WHERE state = 0")
    # state0_id_list = [i[0] for i in state0_id_list]
    # for i in state0_id_list:
    #     if i in new_id_list:
    #         new_id_list.remove(i)
    # sql_pool.insert_all("INSERT INTO ags_task(cert_id) VALUES (%s)", new_id_list)

    # 查询新任务列表
    # ags_id_list = sql_pool.select_all("SELECT id,cert_id FROM ags_task WHERE state != 1")
    ags_id_list = sql_pool.select_all("SELECT id,cert_id FROM ags_task WHERE state = 0 LIMIT 5000")
    ags_id_list = [i for i in ags_id_list]
    return ags_id_list


@retry(stop_max_attempt_number=5, wait_fixed=2000)
def main():
    """
    爬虫主函数
    """
    try:
        logger.info("开始运行 tag_spider 爬虫任务............................................................")
        sql_pool = MySQLConnectionPool(log=logger)
        if not sql_pool:
            logger.error("数据库连接失败")
            raise Exception("数据库连接失败")
        new_task = get_new_task(sql_pool)
        try:
            process_urls(new_task, sql_pool, batch_size=1000, max_workers=5)
        except Exception as e:
            logger.error('process urls: ', e)

    except Exception as e:
        logger.error(f'error:{e}')
    finally:
        logger.info("爬虫程序运行结束,等待下一轮的采集任务.............")


def schedule_task():
    # main()
    schedule.every().day.at("00:01").do(main)
    while True:
        schedule.run_pending()
        time.sleep(1)


if __name__ == '__main__':
    schedule_task()