# coding=utf-8

import os
from selenium import webdriver
import time
import requests
import sys
import warnings

warnings.filterwarnings("ignore")

'''
记录日志
'''
def print_log(content):
    content = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime()) + ' ' + content
    print(content)
    # logfile = './kwaiLiveCookie.log'
    # file = open(logfile, 'a', encoding='utf8')
    #
    # file.write(content + '\n')
    # # 关闭文件
    # file.close()


class KwaiLiveWebDriver:

    # 初始化方法：设置浏览器驱动
    def __init__(self, phone):

        chrome_driver_path = r'chromedriver.exe'
        option = webdriver.ChromeOptions()
        option.add_argument(r'--user-data-dir=D:\Chrome\\' + phone)  # 设置成用户自己的数据目录
        option.add_argument(r'--disable-software-rasterizer')

        os.environ["webdriver.chrome.driver"] = chrome_driver_path

        self.browser = webdriver.Chrome(
            executable_path=chrome_driver_path,
            options=option
        )

    def run(self):
        browser = self.browser

        self.openBrowser()
        print_log('打开浏览器，进入快手直播页')
        browser.implicitly_wait(60)  # 隐性等待，最长等30秒

        self.openLive()
        time.sleep(60)
        self.sendCookie()
        time.sleep(300)
        browser.quit()
        print_log('关闭浏览器')
        print_log('--------------------------------')

    def openBrowser(self):
        browser = self.browser
        # browser.maximize_window()
        # chrome浏览器打开快手直播页
        browser.get("https://live.kuaishou.com/")

    def retryOpenLive(self, retry=0):

        browser = self.browser

        # 主播尚未开播，则点击推荐直播
        if "主播尚未开播" in browser.page_source:
            if retry >= 10:
                print_log('主播尚未开播，重试失败：' + str(retry) + ' ' + browser.current_url)
                browser.quit()
                print_log('关闭浏览器')
                print_log('--------------------------------')
                return

            print_log('主播尚未开播，重试：' + str(retry) + ' ' + browser.current_url)
            previewLive = browser.find_elements_by_class_name('preview-video')
            previewLiveLen = len(previewLive)
            if previewLiveLen > 0:
                print(previewLiveLen)
                previewLive[previewLiveLen - 1].click()
            browser.refresh()
            time.sleep(5)

            retry = retry + 1
            return self.retryOpenLive(retry)

    def openLive(self):
        browser = self.browser

        browser.implicitly_wait(60)  # 隐性等待，最长等30秒

        enterLive = browser.find_element_by_class_name('enter-live-btn')

        # 点击进入直播间
        enterLive.click()
        time.sleep(10)

        self.retryOpenLive()

        print_log('获取当前链接：' + browser.current_url)

    def sendCookie(self):
        browser = self.browser

        cookies_list = browser.get_cookies()

        print_log('获取Cookie：')
        # print(cookies_list)

        cookies_str = ''
        for cookie_item in cookies_list:
            cookies_str = cookies_str + cookie_item['name'] + '=' + cookie_item['value'] + '; '

        print_log(cookies_str)

        if 'userId' not in cookies_str:
            browser.quit()
            print_log('关闭浏览器')
            print_log('--------------------------------')
            return

        response = requests.post(
            'https://ks.wenxingshuju.com/v2/api/kwaiOpen/saveCookie',
            {'cookie': cookies_str}
        )

        print_log('发送Cookie：' + response.text)


if __name__ == '__main__':

    phone = sys.argv[1]
    kwaiLiveWebDriver = KwaiLiveWebDriver(phone)
    try:
        kwaiLiveWebDriver.run()
    except Exception as e:
        print_log('抛出异常！' + str(e))

    kwaiLiveWebDriver.browser.quit()
    sys.exit(0)