from DrissionPage import Chromium, ChromiumOptions
from DrissionPage.errors import ElementNotFoundError

from database.tcontentdispatch.model import TContentDispatch
from log.log_manager import log


class Toutiao:
    def __init__(self, article: TContentDispatch):
        self.article = article
        co = ChromiumOptions().auto_port()  # auto_port后需要重新登录账号
        # co = ChromiumOptions()
        self.browser = Chromium(addr_or_opts=co)
        self.tab = self.browser.latest_tab
        self.tab.get('https://mp.toutiao.com/')

    def need_login(self):
        self.tab.wait(5)
        try:
            login_dialog = self.tab.ele('.login-wrap')
            log(login_dialog.html)
            return True
        except ElementNotFoundError:
            return False

    def login_with_password(self):
        try:
            # 1. 点击‘密码登录’按钮，切换到密码登录界面
            password_login_btn = self.tab.ele('.web-login-other-login-method__list__item__icon web-login-other-login-method__list__item__icon__account_pwd')
            password_login_btn.click()
            # 2. 输入账号
            username_input_box = self.tab.ele('.web-login-normal-input__input')
            username_input_box.input('17704081680')
            # 3. 输入密码
            password_input_box = self.tab.ele('.web-login-button-input__input')
            password_input_box.input('G*9dkvm834;.,[')
            # 4. 勾选同意协议
            agree_checkbox = self.tab.ele('.web-login-confirm-info__checkbox')
            agree_checkbox.click()
            # 5. 点击登录按钮
            login_btn = self.tab.ele('.web-login-button')
            login_btn.click()
        except ElementNotFoundError:
            log('尝试进行账号登录，但登录界面元素未找到')


    def publish(self):
        try:
            # 1. 处理登录
            if self.need_login():
                self.login_with_password()
            # 2. 新建文章
            new_article_btn = self.tab.ele('.byte-menu-inline base_creation_tab').ele('.byte-menu-item')
            new_article_btn.click()
            # 3. 输入标题
            title_input_box = self.tab.ele('.editor-title autofit-textarea-wrapper').ele('tag:textarea')
            title_input_box.input(self.article.title)
            # 4. 输入正文
            content_input_area = self.tab.ele('.ProseMirror')
            if self.article.ai_content:
                content_input_area.input(self.article.ai_content)
            else:
                content_input_area.input(self.article.content)
            # 5. 等待10秒，文章会自动保存到草稿箱
            self.tab.wait(10)
        except (ElementNotFoundError, AttributeError):
            log('发布文章出现异常')
        finally:
            # 6. 结束
            self.finish()

    def finish(self):
        # 关闭浏览器
        self.browser.quit()


if __name__ == '__main__':
    article = TContentDispatch()
    article.title = '今日新鲜事'
    article.content = '当地时间29日，阿塞拜疆总统阿利耶夫称，阿克套空难原因系飞机“遭受来自地面的攻击受损”，飞机在俄境内格罗兹尼附近尾部遭地面射击严重破坏且失去控制。'
    toutiao = Toutiao(article)
    toutiao.publish()