"""
@author: luojunhui
"""
import json
import requests


def get_cookie():
    """
    获取 cookie
    :return:
    """
    url = "https://admin.piaoquantv.com/manager/login?account=luojunhui&passWd=e10adc3949ba59abbe56e057f20f883e&muid=7"
    payload = {}
    headers = {
        'accept': 'application/json, text/plain, */*',
        'accept-language': 'en',
        'priority': 'u=1, i',
        'sec-ch-ua': '"Chromium";v="124", "Google Chrome";v="124", "Not-A.Brand";v="99"',
        'sec-ch-ua-mobile': '?0',
        'sec-ch-ua-platform': '"macOS"',
        'sec-fetch-dest': 'empty',
        'sec-fetch-mode': 'cors',
        'sec-fetch-site': 'same-origin',
        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36'
    }
    response = requests.request("GET", url, headers=headers, data=payload)
    return response.cookies.values()[0]


def auto_white(root_share_id):
    """
    自动加入白名单， 保证公众号百分百出广告
    :param root_share_id:
    :return:
    """
    url = "https://admin.piaoquantv.com/manager/ad/own/white/rootShare/save"
    dd = {
        "rootShareId": root_share_id,
        "commit": "算法自动加入白名单--"
    }
    payload = json.dumps(dd)
    cookie = get_cookie()
    headers = {
        'accept': 'application/json',
        'accept-language': 'en',
        'content-type': 'application/json;',
        'cookie': "SESSION=" + cookie,
        'origin': 'https://admin.piaoquantv.com',
        'priority': 'u=1, i',
        'user-agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36'
    }
    response = requests.request("POST", url, headers=headers, data=payload)
    return response.json()['content']