利用Python自动化脚本登录csdn，调用AI大模型帮我写csdn 知识博客然后自动发布！！！成功了，AI生成的文章质量比我自己呕心沥血写的文章的平均质量分还高10分！！

zero笨小孩

已于 2024-04-16 11:51:56 修改

阅读量1.6k

点赞数 54

分类专栏： python 文章标签：自动化 Python AI自动化人工智能 AI编程 AI写作 ai python

于 2024-04-15 09:32:29 首次发布

本文链接：https://blog.csdn.net/saxx2/article/details/137687634

版权

python 专栏收录该内容

3 篇文章 0 订阅

订阅专栏

作为一个前端开发工程师，最近也在尝试自学Python 自动化相关知识，突然想到能否通过自动化脚本去登录csdn,然后调用AI去写文章博客，再通过脚本去发布，一个字”干“！

最终效果：

以下是我的python脚本的大致流程，因为我刚学python,代码质量仅供参考：

使用webdriver创建一个浏览器窗口

 # 创建一个Chrome浏览器实例
    driver = webdriver.Chrome(chrome_options)

    driver.get("https://www.csdn.net/")  # 打开网页
    driver.maximize_window()  # 最大化

    wait = WebDriverWait(driver, 10) # 等待网页加载完成

获取登录按钮，触发其点击事件，显示登录tab;因为登录弹框是嵌入在Iframe中的，所以这里需要切换网页文档，然后调用login方法

 # 等待登录按钮出现
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".toolbar-btn-loginfun"))).click()

    # 切换到login iframe 中
    login_iframe = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "iframe[name='passport_iframe']")))
    driver.switch_to.frame(login_iframe)
    login(wait)

开始登录流程，获取账号密码输入框，用脚本输入文本，点击登录按钮

# 等待登录框出现
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".login-box-tabs-items > span:last-child"))).click()

    # 输入用户名和密码
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[autocomplete='username']"))).send_keys(
        "1xxxxx0692")
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[autocomplete='current-password']"))).send_keys(
        "110xxxx612zl")
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".icon-nocheck"))).click()
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".login-form-item button"))).click()

登录成功后，进入编辑页面

   
        driver.get("https://mp.csdn.net/mp_blog/creation/editor")  # 打开网页
        wait = WebDriverWait(driver, 10)

设置当前文章所属的专栏

 # 设置专栏
        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".tag__item-list"))).click()
        wait.until(EC.visibility_of_element_located(
            (By.CSS_SELECTOR, ".tag__options-list > .tag__option-box:first-child"))).click()
        driver.execute_script("document.body.click()")

设置标签

        def set_tag(text):

            # 设置标签
            wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".tag__btn-tag"))).click()
            wait.until(EC.visibility_of_element_located(
                (By.CSS_SELECTOR, ".mark_selection_box_header input"))).send_keys(
                text)
            time.sleep(1)
            # 按下回车键
            wait.until(EC.visibility_of_element_located(
                (By.CSS_SELECTOR, ".mark_selection_box_header input"))).send_keys(Keys.ENTER)
            time.sleep(1)
            driver.execute_script("document.body.click()")

        for aritcle_type in aritcle_types:
            set_tag(aritcle_type)

调用AI模型，获取文章标题和内容，并使用脚本插入到页面中,这里需要注意的是文章内容编辑元素也是嵌入在Iframe中的，所以这里也需要切换文档到iframe中，然后在获取元素

content = get_csdn_article(article_title)
        print(content)
        title_textarea = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".input__title textarea")))
        title_textarea.send_keys(content['title'])
        # 切换到login iframe 中
        content_iframe = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "iframe.cke_wysiwyg_frame")))
        driver.switch_to.frame(content_iframe)
        content_body = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".cke_editable>p")))
        # 添加子元素
        driver.execute_script('''
                        var childElement = document.createElement("div");
                        childElement.innerHTML = arguments[0];
                        arguments[1].innerHTML = '';
                        arguments[1].appendChild(childElement)''', content['content'], content_body)
        time.sleep(20)
        content_body.send_keys("[[以上内容均由AI自动化生成发布,仅供参考,谢谢您的访问]]")

快行的起来吧，我已经写了十几篇了，但质量仅供参考

完整代码如下：

import time
from selenium import webdriver
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
import os
import json
import sys
from wenxin_ai import get_csdn_article_by_wenxin
from zhipu_ai import get_csdn_article, get_csdn_article_title



def login(wait):
    argv_local = sys.argv[0] if len(sys.argv) >= 1 else 2
    # 等待登录框出现
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".login-box-tabs-items > span:last-child"))).click()

    # 输入用户名和密码
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[autocomplete='username']"))).send_keys(
        "18180276692")
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "input[autocomplete='current-password']"))).send_keys(
        "1103661612zl")
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".icon-nocheck"))).click()
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".login-form-item button"))).click()

    time.sleep(10)

    def create_csdn_blog(article_title="", aritcle_types=None):
        if aritcle_types is None:
            aritcle_types = ["面试", "前端", "javaScript", "AI", "Python", "自动化脚本"]
        driver.get("https://mp.csdn.net/mp_blog/creation/editor")  # 打开网页
        wait = WebDriverWait(driver, 10)
        driver.execute_script("window.scrollTo(0, 2000)")
        time.sleep(5)
        driver.execute_script("window.scrollTo(0, 2000)")
        # 设置专栏
        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".tag__item-list"))).click()
        wait.until(EC.visibility_of_element_located(
            (By.CSS_SELECTOR, ".tag__options-list > .tag__option-box:first-child"))).click()
        driver.execute_script("document.body.click()")
        time.sleep(2)

        def set_tag(text):

            # 设置标签
            wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".tag__btn-tag"))).click()
            wait.until(EC.visibility_of_element_located(
                (By.CSS_SELECTOR, ".mark_selection_box_header input"))).send_keys(
                text)
            time.sleep(1)
            # 按下回车键
            wait.until(EC.visibility_of_element_located(
                (By.CSS_SELECTOR, ".mark_selection_box_header input"))).send_keys(Keys.ENTER)
            time.sleep(1)
            driver.execute_script("document.body.click()")

        for aritcle_type in aritcle_types:
            set_tag(aritcle_type)
            # content = get_csdn_article_by_wenxin(article_title)
        # 设置标题和内容
        content = get_csdn_article(article_title)
        print(content)
        title_textarea = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".input__title textarea")))
        title_textarea.send_keys(content['title'])
        # 切换到login iframe 中
        content_iframe = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "iframe.cke_wysiwyg_frame")))
        driver.switch_to.frame(content_iframe)
        content_body = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".cke_editable>p")))
        # 添加子元素
        driver.execute_script('''
                        var childElement = document.createElement("div");
                        childElement.innerHTML = arguments[0];
                        arguments[1].innerHTML = '';
                        arguments[1].appendChild(childElement)''', content['content'], content_body)
        time.sleep(20)
        content_body.send_keys("[[以上内容均由AI自动化生成发布,仅供参考,谢谢您的访问]]")
        driver.switch_to.default_content()
        time.sleep(2)
        driver.execute_script("document.body.click()")
        driver.execute_script("window.scrollTo(0, 20000)")
        time.sleep(2)
        # 提取摘要
        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".btn-getdistill"))).click()
        time.sleep(5)
        # 提交
        wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".btn-box .btn-outline-danger"))).click()
        time.sleep(5)
        try:
            time.sleep(5)
            print("开始创建文章", article_title)
        except Exception as e:
            print("登录失效，重新登录")
            login(wait)

    if argv_local == 1:
        try:
            argv_title = sys.argv[1]  # title
            argv_aritcle_types = sys.argv[1:]  # aritcle_types
            create_csdn_blog(argv_title, argv_aritcle_types)
        except Exception as e:
            create_csdn_blog()
    else:
        with open('local_questions.json', 'r', encoding='utf-8') as file:
            datas = json.load(file)
            for data in datas['values']:
                print(data["title"], data["tags"])
                create_csdn_blog(data["title"], data["tags"])


if __name__ == '__main__':
    chrome_options = Options()
    chrome_options.add_argument("--disable-save-password-bubble")
    chrome_options.add_experimental_option("excludeSwitches", ["enable-automation"])
    chrome_options.add_argument('--disable-blink-features=AutomationControlled')

    # 创建一个Chrome浏览器实例
    driver = webdriver.Chrome(chrome_options)

    driver.get("https://www.csdn.net/")  # 打开网页
    driver.maximize_window()  # 最大化

    wait = WebDriverWait(driver, 10) # 等待网页加载完成

    # 等待登录按钮出现
    wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, ".toolbar-btn-loginfun"))).click()

    # 切换到login iframe 中
    login_iframe = wait.until(EC.visibility_of_element_located((By.CSS_SELECTOR, "iframe[name='passport_iframe']")))
    driver.switch_to.frame(login_iframe)
    login(wait)

from zhipuai import ZhipuAI
import json
import markdown


client = ZhipuAI(api_key="432b814e2xxxxxxxxxx4kvde0vFLhFkSC")  # 填写您自己的APIKey




def get_csdn_article_title():
    # 读取json文件
    with open('question.json', 'r', encoding='utf-8') as file:
        #
        datas = json.load(file)
        messages = []
        content = "我是一位前端开发面试官，我需要一道优质的面试题，你前面已经出过的题目,请一定不要再次出现了,知识范围请从[Vue、React、Html、css、js、浏览器知识、计算机网络、webpack、工程化、性能优化]中随机选择一个内容，请给我题目（标题），题要有针对性，聚焦某个特定的知识点，不要太宽泛，标题尽量精简不要太长,不要出现题目要求，只给我题目就行，我自己去补充内容,所有标题以'AI题库：'开头‘,最重要的是你前面已经存在过的题目,请一定不要在出现了,这是最重要的"
        for data in datas['values']:
            # 合并两个数组
            messages.append({"role": "user", "content": content})
            messages.append({"role": "assistant", "content": data})
        messages.append({"role": "user", "content": content})
        print(messages)
        response = client.chat.completions.create(
            model="glm-3-turbo",  # 填写需要调用的模型名称
            temperature=0.99,
            top_p=0.99,
            messages=messages
        )

        datas['values'].append(response.choices[0].message.content)
        with open('question.json', 'w', encoding='utf-8') as file:
            json.dump(datas, file, indent=4, ensure_ascii=False)
        return response.choices[0].message.content


# 获取一篇csdn文章 自定义标题 or 随机ai标题
def get_csdn_article(title):
    if title:
        article_title = title
    else:
        article_title = get_csdn_article_title()
    response = client.chat.completions.create(
        model="glm-3-turbo",  # 填写需要调用的模型名称
        temperature=0.99,
        max_tokens=2048,
        messages=[
            {"role": "user",
             "content": "我是一位程序员,请帮我写一篇关于" + article_title + "的高质量csdn博客,要求内容丰富 准确 严谨 正确,适当扩展内容,尽可能去示例说明细节,最好是在最前面给我加上目录方便阅读,谢谢你的帮助"},
        ],
    )
    html = markdown.markdown(response.choices[0].message.content)
    html = '<h3 style="margin:20px 0;color:#e2855c">以下内容均由AI自动化生成发布,仅供参考,谢谢您的访问</h3><br>' + html + '<br><br>'
    print(html)
    return {
        "title": title,
        "content": html
    }