import sys
import os

sys.path.append(os.path.dirname(sys.path[0]))  # 上级目录
import curl_cffi
from lxml import etree
from threading_spider.db_connectivity import connect_db
import json
import time
import pymysql
import random


def get_cid():
    # 获取所有站点的bsr 大类名称 和 分类id，存储到us站点
    url = 'https://www.sellersprite.com/v2/tools/sales-estimator'
    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, zstd",
        "Accept-Language": "zh-CN,zh-TW;q=0.9,zh;q=0.8",
        "Cache-Control": "no-cache",
        "User-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/118.0.0.0 Safari/537.36",
    }
    resp = curl_cffi.get(url, headers=headers, impersonate="chrome")
    html = etree.HTML(resp.text)
    data_category = html.xpath("//script[@id='data-category']/text()")[0]
    print(data_category)
    save_site_category(json.loads(data_category))


def junglescout_spider(db_base):
    month = time.strftime("%m")
    year = time.strftime("%Y")
    year_month = f'{year}_{int(month)}'

    db_class = connect_db(db_base)
    cursor_mysql_db, connect_mysql_db = db_class.mysql_db()  # mysql
    cursor_us, connect_us = db_class.us_mysql_db()  # us站点
    cursor_us = connect_us.cursor(cursor=pymysql.cursors.DictCursor)  # 以字典的格式输出
    category_name_sql_select = f"select `name`,c_id  from all_site_category where site='{db_base}' and state =1"
    print(category_name_sql_select)
    cursor_us.execute(category_name_sql_select)
    category_name_list = cursor_us.fetchall()
    print('category_name_list:',category_name_list)
    for category_name in category_name_list:
        db_class = connect_db(db_base)
        print(db_base)
        cursor_mysql_db, db = db_class.mysql_db()  # mysql
        db_class_us = connect_db('us')
        cursor_us_mysql_db, db_us = db_class_us.us_mysql_db()  # us 站点 mysql
        # cursor = db.cursor(cursor=pymysql.cursors.DictCursor)  # 以字典的格式输出
        print(category_name['name'], category_name['c_id'])
        name_rnak_list = []
        up_sql = f"UPDATE all_site_category set state=2 WHERE site='{db_base}' and state=1 and c_id='{category_name['c_id']}'"
        print('更新状态：', up_sql)
        cursor_us_mysql_db.execute(up_sql)
        db_us.commit()
        rank_list = [1, 10, 30, 50, 100, 200, 300, 400, 500, 600, 700, 800, 900, 1000, 1100, 1200, 1300, 1400, 1500,
                     1600,
                     1700,
                     1800,
                     1900,
                     2000, 2100, 2200, 2300, 2400, 2500, 2600, 2700, 2800, 2900, 3000, 3100, 3200, 3300, 3400, 3500,
                     3600,
                     3700,
                     3800, 3900, 4000, 4100, 4200, 4300, 4400, 4500, 4600, 4700, 4800, 4900, 5000, 5100, 5200, 5300,
                     5400,
                     5500,
                     5600, 5700, 5800, 5900, 6000, 6100, 6200, 6300, 6400, 6500, 6600, 6700, 6800, 6900, 7000, 7100,
                     7200,
                     7300,
                     7400, 7500, 7600, 7700, 7800, 7900, 8000, 8100, 8200, 8300, 8400, 8500, 8600, 8700, 8800, 8900,
                     9000,
                     9100,
                     9200, 9300, 9400, 9500, 9600, 9700, 9800, 9900, 10000, 11000, 12000, 13000, 14000, 15000, 16000,
                     17000,
                     18000, 19000, 20000, 25000, 30000, 35000, 40000, 45000, 50000
                     ]
        for i in rank_list:
            headers = {
                "Referer": "https://www.sellersprite.com/v2/tools/sales-estimator",
                "Origin":"https://www.sellersprite.com",
                "Accept": "application/json, text/javascript, */*; q=0.01",
                "Accept-Encoding": "gzip, deflate, br, zstd",
                "Accept-Language": "zh-CN,zh-TW;q=0.9,zh;q=0.8",
                "Cache-Control": "no-cache",
                'Cookie': '_ga=GA1.1.19240078.1751854600; Hm_lvt_e0dfc78949a2d7c553713cb5c573a486=1751854601; HMACCOUNT=28ABEEABEFA97E4A; _gcl_au=1.1.536675967.1751854601; MEIQIA_TRACK_ID=2zWlEnsYAqnZRdhJqJ5txX7tpXm; MEIQIA_VISIT_ID=2zWlEmUkBQV745rliAtXEdAk0CJ; ecookie=ZyZ05gxOxlDTPkM1_CN; 8f00639f9c446a2d0213=54fb71d3f2c9e8acb7878e0f73abbf33; _fp=65dbbe41a37f8f9fbe702eba96328267; _gaf_fp=e03eac62da4f8988dc796341e1bd822c; current_guest=jsxcNvsgBJO1_250707-100340; rank-login-user=502219157192wVgAJpdturGN5Im+nPDQqTtoVYwVNo1oWP9MD0mtMHFwS3LrhtAUhuCnvMHsCl; rank-login-user-info="eyJuaWNrbmFtZSI6IuWViuWTiOWTiOWTiCIsImlzQWRtaW4iOmZhbHNlLCJhY2NvdW50IjoiMTUzKioqKjEyNzAiLCJ0b2tlbiI6IjUwMjIxOTE1NzE5MndWZ0FKcGR0dXJHTjVJbStuUERRcVR0b1ZZd1ZObzFvV1A5TUQwbXRNSEZ3UzNMcmh0QVVodUNudk1Ic0NsIn0="; Sprite-X-Token=eyJhbGciOiJSUzI1NiIsImtpZCI6IjE2Nzk5NjI2YmZlMDQzZTBiYzI5NTEwMTE4ODA3YWExIn0.eyJqdGkiOiIwZ01FdlJuNWJ1dlZhVW5IZ1lKSDFRIiwiaWF0IjoxNzUxODU0NjA1LCJleHAiOjE3NTE5NDEwMDUsIm5iZiI6MTc1MTg1NDU0NSwic3ViIjoieXVueWEiLCJpc3MiOiJyYW5rIiwiYXVkIjoic2VsbGVyU3BhY2UiLCJpZCI6MTQ2NjIwMSwicGkiOm51bGwsIm5uIjoi5ZWK5ZOI5ZOI5ZOIIiwic3lzIjoiU1NfQ04iLCJlZCI6Ik4iLCJwaG4iOiIxNTM2ODA1MTI3MCIsImVtIjoibWVpeW91bGFAbWVpeW91bGEuY29tIiwibWwiOiJHIn0.Ujr6_K3vHIQRw3x52QAQdTftMy6GbZ_TunmFMgW76onCy3EkBzx7uxEv-42zRRXgKLMUfJz2t0ierqXV6Evh9i-o5F0ZUBREzm48LHpGSw6Iupjx4Udc3VQwVqgiUOmYBvnTAQqmaj6iA5l06zAZcVNHQASZ5xe5QFUCllIOL0m8tf3Xad6T8u5oLHRHTTuyy5nDAqLu6ZxVOqUYYXsIzq9H2qAsPhqIgRy_5Av1zyoAcQErddadCe25H_ILmKO0Az9ANIFg4o1r_is_VFVZpGvbz8nCN0JLuY3uajAjf2JXoEzhHT9YbMP0o2TrZDRPdORV3HVK1N5uvghRaRyJvw; ao_lo_to_n="502219157192wVgAJpdturGN5Im+nPDfbd9htCMUGF/tdMS8/gmBNzv9/utYT5ucwmHHPC71S6i4RnT3fLUZW/nDI61eZx1uqLqr+hBy0X/aeJ6c/sSSc="; rank-guest-user=502219157192wVgAJpdturGN5Im+nPDYsyQgRxjbXtKYdDjju8ax0OkcsNUNGWP3xY6uiwKVVO; JSESSIONID=96FF611DCBDF20B9C6C921EAD2A55205; _ga_38NCVF2XST=GS2.1.s1751854600$o1$g1$t1751854612$j48$l0$h1855838417; Hm_lpvt_e0dfc78949a2d7c553713cb5c573a486=1751854612; _ga_CN0F80S6GL=GS2.1.s1751854600$o1$g1$t1751854613$j47$l0$h0',
                "User-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
            }
            url = "https://www.sellersprite.com/v2/tools/sales-estimator/bsr.json"
            data = {
                "station": "US",
                "cid": category_name['c_id'],  # 分类id
                "bsr": f"{i}"  # 排名
            }
            print(category_name['name'], '请求参数 data::', data)
            for i1 in range(3):
                try:
                    response = curl_cffi.post(url, headers=headers, data=data, impersonate="chrome",timeout=300)
                    print(response.url)
                    # print(response.text)
                    response = json.loads(response.text)
                    break
                except:
                    time.sleep(random.uniform(15, 35.75))

            response_data = response['data']
            print('code::', response['code'])
            print('message::', response['message'])
            print('estMonSales::', response_data['estMonSales'])
            est = response_data.get('estMonSales')
            if est is None:
                # 没拿到数据，跳出循环
                break
            if est == 0.0:
                print(f"{category_name['name']} 排名{i}：销量 0，跳出循环。")
                break

            # 2) 既然不会有 0.3、0.99 这种情况，剩下的 est 都是 ≥ 1
            print(type(est))
            print('获取数据：', category_name['name'], i, est, year_month)
            sales = int(est)
            name_rnak_list.append((category_name['name'], i, sales, year_month))
            time.sleep(random.uniform(30, 95.75))
            # break
        for i in range(4):
            try:
                inset_sql = f"INSERT INTO {db_base}_one_category (name, rank,orders,`year_month`) values (%s, %s, %s, %s)"
                print(inset_sql)
                cursor_mysql_db.executemany(inset_sql, name_rnak_list)
                db.commit()
                up_sql = f"UPDATE all_site_category set state=3 WHERE site='{db_base}' and state=2 and c_id='{category_name['c_id']}'"
                print('更新状态：', up_sql)
                cursor_us_mysql_db.execute(up_sql)
                db_us.commit()
                break
            except Exception as e:
                print('存储失败：',e)
                db_class = connect_db(db_base)
                print(db_base)
                cursor_mysql_db, db = db_class.mysql_db()  # mysql
                cursor_us_mysql_db, db_us = db_class_us.us_mysql_db()  # us 站点 mysql
                time.sleep(20)

        print('当前完成。获取下一个分类销量')
        time.sleep(random.uniform(120, 240.5))


def save_site_category(site_bsr_dict=None):
    db_class = connect_db('us')
    cursor_us_mysql_db, db = db_class.us_mysql_db()  # mysql
    for i in site_bsr_dict.keys():
        print(i)
        delete_sql = f'DELETE from all_site_category where `name` ="{i}"'  # 删除旧分类
        print(delete_sql)
        cursor_us_mysql_db.execute(delete_sql)
        db.commit()
        site_category_list = site_bsr_dict[i]
        for site_category in site_category_list:
            insert_sql = f'insert into all_site_category (site,`name`, c_id)values (%s,%s,%s)'
            cursor_us_mysql_db.execute(insert_sql, (i, site_category['categoryLabel'], site_category['cid']))
            db.commit()
        db_class = connect_db(i)
        cursor_site_mysql_db, db = db_class.mysql_db()  # mysql


def run():
    # get_cid()
    junglescout_spider('us')


if __name__ == '__main__':
    run()
