geetest.py

import os
import json
# import win32api,win32con 移除不需要的功能 By FriendshipEnder 4/19
from selenium import webdriver
from selenium.webdriver.common.by import By
from time import sleep
import pyperclip # 新增剪贴板功能 By FriendshipEnder 4/19

class dealCode():
	def __init__(self,specificID=None):

		self.specificID = specificID
		self.cookies = {}
		self.u = ""

	def load_cookies(self):
		if not os.path.exists("user_data.json"):
			print("未找到用户数据文件")
			os.system("pause")
			exit()
		with open("user_data.json","r") as r:
			try:
				data = json.load(r)
				if len(data) == 0:
					raise ""
			except:
				print("数据文件错误，请登陆后尝试")
			if self.specificID:
				data = data[self.specificID][1]
			else:
				data = data[list(data.keys())[0]][1]
			for item in data.split(";"):
				i = item.split("=")
				if len(i) == 2:
					self.WebDriver.add_cookie(cookie_dict={"domain": ".bilibili.com" ,'name' : i[0].strip(), 'value' : i[1].strip()})
	
	def init_browser(self):
		options = webdriver.EdgeOptions()
		options.add_argument("--log-level=3")
		self.WebDriver = webdriver.Edge(options=options)
		self.WebDriver.get("https://bilibili.com")
		self.WebDriver.delete_all_cookies()

	def load_code(self,url):
		print("token需要验证！！正在加载验证码，请完成验证后关闭网页！！")
		self.init_browser()
		self.load_cookies()
		self.WebDriver.get(url)
		while True:
			sleep(0.1)
			try:
				self.WebDriver.execute_script('javascript:void(0);')
			except:
				return 1

	def test(self):
		self.load_code("https://bilibili.com")

	def mult_work(self):
		self.init_browser()
		self.load_cookies()
		pyperclip.copy("")   # 先向剪贴板写入空白字符, 用于检测验证码是否更新了剪贴板 By FriendshipEnder 4/19
		while pyperclip.paste() == "":   # 检测剪贴板是否更新 By FriendshipEnder 4/19
			sleep(0.25)
			if not os.path.exists("url"):
				print("请在脚本目录打开此程序")
				os.system("pause")
			a = open("url","r")
			u = a.read()
			a.close()
			if u and u.strip() != self.u:
				self.u = u
				# 不再用msgbox那个提醒了, 自动弹窗显示验证码 By FriendshipEnder 4/19
				self.WebDriver.get(u)
			# print("剪贴板: ",pyperclip.paste()) # 输出剪贴板内容 By FriendshipEnder 4/19

if __name__ == '__main__':
	dealCode().mult_work()