目錄

一、NovelAI?

二、UIautomation和pywin32

三、代碼?

1、AI畫圖相關

?2、QQ群消息抓取相關

四、代碼效果?

五、后記?

---

一、NovelAI?

NovelAI是一個用來畫二次元圖片的開源算法，部署方式見? b站鏈接

二、UIautomation和pywin32

UIautomation和pywin32是python的庫，主要用于窗口句柄的抓取和操作，本代碼中用于接收、發送QQ群消息?

三、代碼?

1、AI畫圖相關

通過使用NovelAI代碼包里的txt2img.py文件，實現從文字到圖片的轉化：

def txt2img(prompt: str, negative_prompt: str, prompt_style: str, prompt_style2: str, steps: int, sampler_index: int, restore_faces: bool, tiling: bool, n_iter: int, batch_size: int, cfg_scale: float, seed: int, subseed: int, subseed_strength: float, seed_resize_from_h: int, seed_resize_from_w: int, seed_enable_extras: bool, height: int, width: int, enable_hr: bool, scale_latent: bool, denoising_strength: float, *args):p = StableDiffusionProcessingTxt2Img(sd_model=shared.sd_model,outpath_samples=opts.outdir_samples or opts.outdir_txt2img_samples,outpath_grids=opts.outdir_grids or opts.outdir_txt2img_grids,prompt=prompt,styles=[prompt_style, prompt_style2],negative_prompt=negative_prompt,seed=seed,subseed=subseed,subseed_strength=subseed_strength,seed_resize_from_h=seed_resize_from_h,seed_resize_from_w=seed_resize_from_w,seed_enable_extras=seed_enable_extras,sampler_index=sampler_index,batch_size=batch_size,n_iter=n_iter,steps=steps,cfg_scale=cfg_scale,width=width,height=height,restore_faces=restore_faces,tiling=tiling,enable_hr=enable_hr,scale_latent=scale_latent if enable_hr else None,denoising_strength=denoising_strength if enable_hr else None,)if cmd_opts.enable_console_prompts:print(f"\ntxt2img: {prompt}", file=shared.progress_print_out)processed = modules.scripts.scripts_txt2img.run(p, *args)if processed is None:processed = process_images(p)shared.total_tqdm.clear()generation_info_js = processed.js()if opts.samples_log_stdout:print(generation_info_js)if opts.do_not_show_images:processed.images = []return processed.images, generation_info_js, plaintext_to_html(processed.info)

輸入對應的參數后，圖像數據存儲在以下位置

txt2img[0][0].save(fullfn)#fullfn是路徑

?2、QQ群消息抓取相關

接收方面

_get_all_hwnd(hwnd, mouse)用于遍歷QQ窗口下的所有控件，用于找到消息管理器窗口下的所有控件

def _get_all_hwnd(hwnd, mouse):if win32gui.IsWindow(hwnd) and win32gui.IsWindowEnabled(hwnd) and win32gui.IsWindowVisible(hwnd):hwnd_title.update({hwnd: win32gui.GetWindowText(hwnd)})

textrefresh(delay)用于點擊消息管理器的刷新按鈕，以更新群消息

def textrefresh(delay):win32gui.EnumWindows(_get_all_hwnd, 0)for wnd in hwnd_title.items():# print(wnd)if wnd[1] == '消息管理器':breaklong_position = win32api.MAKELONG(810, 130)win32api.SendMessage(int(wnd[0]), win32con.WM_LBUTTONDOWN, win32con.MK_LBUTTON, long_position)win32api.SendMessage(int(wnd[0]), win32con.WM_LBUTTONUP, win32con.MK_LBUTTON, long_position)sleep(delay)

?之后在一個while循環里不斷獲取最新的一條群消息，并用正則表達式提取指令，以用于后續操作

chat_window = auto.WindowControl(searchDepth=1, ClassName='TXGuiFoundation', Name='消息管理器') msg_list = chat_window.ListControl(Name='IEMsgView') #找到 list finalmsg = msg_list.GetLastChildControl() msg = finalmsg.Name # print(msg) obj = re.compile(r'.*?\((?P<QQnum>.*?)\)\d{1,}:\d{2}:\d{2}(?P<QQmsg>.*)', re.S) #可拿出 result = obj.findall(msg)

需要注意的是“點擊刷新按鈕”和“獲取群消息”是兩個事件。

不同點是“點擊刷新按鈕”使用的是win32api，win32gui，win32con庫，“獲取群消息”使用的是uiautomation庫。

相同點是這兩個操作都是針對“消息管理器”窗口的。

uiautomation庫和win32api，win32gui，win32con庫有很多共有的功能，兩者都是對控件進行操作的庫，博主在使用時有所取舍。

對于“點擊刷新按鈕”需求：uiautomation的點擊操作需要占用鼠標，很不方便，而win32庫里有后臺鼠標，因此用于實現“點擊刷新按鈕”的需求是合適的。

對于“獲取群消息”需求：win32庫需要遍歷所有子控件，博主認為這會提高運行時間，并且定位最后一條群消息還需要額外的篩選操作，而uiautomation里可以直接調用GetLastChildControl()方法得到最后一個子控件，而這個最后的子控件剛好是群內最新的消息。

至此程序已經完成了提取群消息里的指令的需求

發送方面

fs(fsgs, fsnr)用于向群聊天窗口發送文字信息

def fs(fsgs, fsnr):c.OpenClipboard()c.EmptyClipboard()c.SetClipboardData(b.CF_UNICODETEXT, fsnr)c.CloseClipboard()handle = a.FindWindow(None, fsgs)if handle != 0:a.SendMessage(handle, 770, 0, 0)a.SendMessage(handle, b.WM_KEYDOWN, b.VK_RETURN, 0)print('消息發送成功！')

sendImage(name,imgpath)用于向群聊天窗口發送圖片

def sendImage(name,imgpath):im = Image.open(imgpath)im.save('1.bmp')aString = windll.user32.LoadImageW(0, r"1.bmp", win32con.IMAGE_BITMAP, 0, 0, win32con.LR_LOADFROMFILE)#print(aString)if aString != 0: ## 由于圖片編碼問題 圖片載入失敗的話 aString 就等于0w.OpenClipboard()w.EmptyClipboard()w.SetClipboardData(win32con.CF_BITMAP, aString)# 關閉剪貼板w.CloseClipboard()# 獲取qq窗口句柄handle = win32gui.FindWindow(None, name)if handle == 0:print('未找到窗口！')# 顯示窗口win32gui.ShowWindow(handle, win32con.SW_SHOW)# time.sleep(0.2)# 把剪切板內容粘貼到qq窗口win32gui.SendMessage(handle, win32con.WM_PASTE, 0, 0)# time.sleep(0.2)# 按下后松開回車鍵，發送消息win32gui.SendMessage(handle, win32con.WM_KEYDOWN, win32con.VK_RETURN, 0)win32gui.SendMessage(handle, win32con.WM_KEYUP, win32con.VK_RETURN, 0)

至此程序已經完成了向群里發送文字和圖片的需求

以上是主要的代碼，后面還需要設置條件判斷群友是不是@了機器人，以及@機器人后的指令該怎么使用 ，等等的細節這里省略（不難）

四、代碼效果?

五、后記?

反正也沒人看，權當是個人學習記錄吧

僅供娛樂，如有侵權請聯系我刪除?

總結

以上是生活随笔為你收集整理的【NovelAI】在QQ群中部署AI画图机器人的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。