352 lines
13 KiB
Python
352 lines
13 KiB
Python
import random
|
||
import threading
|
||
import time
|
||
from enum import Enum
|
||
import wda
|
||
import os
|
||
from Utils.AiUtils import AiUtils
|
||
from Utils.ControlUtils import ControlUtils
|
||
from Utils.LogManager import LogManager
|
||
from Entity.Variables import anchorList, removeModelFromAnchorList, prologueList
|
||
from Utils.Requester import Requester
|
||
|
||
|
||
# 脚本管理类
|
||
class ScriptManager():
|
||
|
||
# 单利对象
|
||
_instance = None # 类变量，用于存储单例实例
|
||
|
||
def __new__(cls):
|
||
# 如果实例不存在，则创建一个新实例
|
||
if cls._instance is None:
|
||
cls._instance = super(ScriptManager, cls).__new__(cls)
|
||
# 返回已存在的实例
|
||
return cls._instance
|
||
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.initialized = True # 标记已初始化
|
||
|
||
# 养号
|
||
def growAccount(self, udid, event):
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
session.appium_settings({"snapshotMaxDepth": 0})
|
||
|
||
# 先关闭Tik Tok
|
||
ControlUtils.closeTikTok(session, udid)
|
||
time.sleep(1)
|
||
|
||
# 重新打开Tik Tok
|
||
ControlUtils.openTikTok(session, udid)
|
||
time.sleep(3)
|
||
|
||
# 创建udid名称的目录
|
||
AiUtils.makeUdidDir(udid)
|
||
|
||
# 假设此时有个开关
|
||
while not event.is_set():
|
||
try:
|
||
img = client.screenshot()
|
||
filePath = f"resources/{udid}/bgv.png"
|
||
img.save(filePath)
|
||
LogManager.info("保存屏幕图像成功", udid)
|
||
print("保存了背景图")
|
||
time.sleep(1)
|
||
except Exception as e:
|
||
LogManager.error(e, udid)
|
||
print(e)
|
||
|
||
try:
|
||
# 判断视频类型
|
||
addX, addY = AiUtils.findImageInScreen("add", udid)
|
||
# 多次获取结果是否一致，如果有不一致的结果就切换视频
|
||
isSame = False
|
||
for i in range(2):
|
||
tx, ty = AiUtils.findImageInScreen("add", udid)
|
||
if addX == tx and addY == ty:
|
||
isSame = True
|
||
time.sleep(1)
|
||
else:
|
||
isSame = False
|
||
break
|
||
|
||
# 如果找到普通视频
|
||
if addX > 0 and isSame:
|
||
needLike = random.randint(0, 10)
|
||
# 查找首页按钮
|
||
homeButton = AiUtils.findHomeButton(udid)
|
||
if homeButton:
|
||
print("有首页按钮，查看视频")
|
||
videoTime = random.randint(5, 15)
|
||
time.sleep(videoTime)
|
||
|
||
# 百分之三的概率点赞
|
||
if needLike < 3:
|
||
print("点赞")
|
||
ControlUtils.clickLike(session, udid)
|
||
|
||
print("继续观看视频")
|
||
videoTime = random.randint(10, 30)
|
||
time.sleep(videoTime)
|
||
print("准备划到下一个视频")
|
||
client.swipe_up()
|
||
else:
|
||
print("找不到首页按钮。出错了")
|
||
else:
|
||
nextTime = random.randint(1, 5)
|
||
time.sleep(nextTime)
|
||
client.swipe_up()
|
||
except Exception as e:
|
||
print(f"发生异常：{e}")
|
||
client.swipe_up()
|
||
|
||
|
||
# 观看直播
|
||
def watchLiveForGrowth(self, udid, event):
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
|
||
session.appium_settings({"snapshotMaxDepth": 15})
|
||
# 先关闭Tik Tok
|
||
ControlUtils.closeTikTok(session, udid)
|
||
time.sleep(1)
|
||
|
||
# 重新打开Tik Tok
|
||
ControlUtils.openTikTok(session, udid)
|
||
time.sleep(3)
|
||
# 进入直播
|
||
live_button = session(xpath='//XCUIElementTypeButton[@name="直播"]')
|
||
if live_button.exists:
|
||
live_button.click()
|
||
time.sleep(20)
|
||
|
||
size = session.window_size()
|
||
width, height = size.width, size.height
|
||
print(f"屏幕的宽高是：{width},{height}")
|
||
|
||
# 可选：重新拉起 session，规避偶发 Stale 会话
|
||
session = client.session()
|
||
# session.appium_settings({"snapshotMaxDepth": 25})
|
||
|
||
while not event.is_set():
|
||
try:
|
||
time.sleep(3)
|
||
|
||
# 如果处于 PK（分数条），直接划走
|
||
if session(xpath='//XCUIElementTypeOther[@name="kGBLInteractionViewMatchScoreBar"]').exists:
|
||
print("✅ 当前是 PK，跳过")
|
||
session.swipe_up()
|
||
continue
|
||
|
||
# 数直播显示区域窗口（主画面 + 连麦小窗）
|
||
count = AiUtils.count_add_by_xml(session)
|
||
print(f"检测到直播显示区域窗口数：{count}")
|
||
|
||
if count > 1:
|
||
print("❌ 多窗口（有人连麦/分屏），划走")
|
||
session.swipe_up()
|
||
continue
|
||
else:
|
||
print("✅ 单窗口（只有一个主播），(目前是20%概率)开始点赞")
|
||
|
||
# 点赞（仍保留中途转PK的保护）
|
||
if random.random() >= 0.89:
|
||
print("开始点赞")
|
||
for _ in range(random.randint(10, 30)):
|
||
if session(xpath='//XCUIElementTypeOther[@name="kGBLInteractionViewMatchScoreBar"]').exists:
|
||
print("❗ 中途开始 PK，停止点赞并跳过")
|
||
session.swipe_up()
|
||
break
|
||
x = width // 3 + random.randint(-10, 10)
|
||
y = height // 3 + random.randint(10, 20)
|
||
print("双击坐标：", x, y)
|
||
session.double_tap(x, y)
|
||
|
||
print("--------------------------------------------")
|
||
time.sleep(random.randint(100, 300))
|
||
session.swipe_up()
|
||
|
||
except Exception as e:
|
||
print("循环异常，重试：", repr(e))
|
||
# 轻量恢复：重新获取 session，避免因为快照或元素句柄失效卡死
|
||
try:
|
||
session = client.session()
|
||
except Exception:
|
||
time.sleep(2)
|
||
session = client.session()
|
||
|
||
|
||
|
||
# 关注打招呼以及回复主播消息
|
||
def greetNewFollowers(self, udid, needReply, event):
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
|
||
# 先关闭Tik Tok
|
||
ControlUtils.closeTikTok(session, udid)
|
||
time.sleep(1)
|
||
|
||
# 重新打开Tik Tok
|
||
ControlUtils.openTikTok(session, udid)
|
||
time.sleep(3)
|
||
|
||
# 设置查找深度
|
||
session.appium_settings({"snapshotMaxDepth": 15})
|
||
# 点击搜索按钮
|
||
ControlUtils.clickSearch(session)
|
||
|
||
# 返回上一步
|
||
def goBack(count):
|
||
# 如需回复消息
|
||
if needReply:
|
||
count += 1
|
||
for i in range(count):
|
||
ControlUtils.clickBack(session)
|
||
time.sleep(2)
|
||
|
||
# 循环条件。1、 循环关闭 2、 数据处理完毕
|
||
while not event.is_set() or len(anchorList) > 0:
|
||
if needReply:
|
||
print("如果需要回复主播消息。走此逻辑")
|
||
if AiUtils.getUnReadMsgCount(session) > 0:
|
||
# 此处调用另外一个方法处理
|
||
|
||
# 设置查找深度
|
||
session.appium_settings({"snapshotMaxDepth": 15})
|
||
# 点击搜索按钮
|
||
ControlUtils.clickSearch(session)
|
||
|
||
# 查找输入框
|
||
input = session.xpath('//XCUIElementTypeSearchField')
|
||
# 如果找到了输入框，就点击并且输入内容
|
||
input.click()
|
||
# 稍作停顿
|
||
time.sleep(1)
|
||
|
||
# 获取一个主播
|
||
anchor = anchorList[0]
|
||
aid = anchor.anchorId
|
||
anchorCountry = anchor.country
|
||
|
||
input.clear_text()
|
||
time.sleep(2)
|
||
# 输入主播id
|
||
input.set_text(aid + "\n")
|
||
|
||
# 切换UI查找深度
|
||
session.appium_settings({"snapshotMaxDepth": 25})
|
||
|
||
# 定位 "关注" 按钮 通过关注按钮的位置点击主播首页
|
||
follow_button = session.xpath('//XCUIElementTypeOther[@name="zhang1231511"]//XCUIElementTypeButton[@enabled="true" and @visible="true"]')
|
||
time.sleep(2)
|
||
if follow_button.exists:
|
||
print("找到关注按钮！")
|
||
x = follow_button.bounds.x - 200
|
||
y = follow_button.bounds.y
|
||
client.click(x, y)
|
||
else:
|
||
goBack(1)
|
||
removeModelFromAnchorList(anchor)
|
||
print("未找到关注按钮")
|
||
continue
|
||
time.sleep(3)
|
||
|
||
# 找到并点击第一个视频
|
||
cellClickResult = ControlUtils.clickFirstVideoFromDetailPage(session)
|
||
|
||
# 观看主播视频
|
||
def viewAnchorVideo():
|
||
print("开始查看视频")
|
||
count = 2
|
||
while count != 0:
|
||
time.sleep(5)
|
||
print("条件满足，继续查看")
|
||
img = client.screenshot()
|
||
time.sleep(1)
|
||
filePath = f"resources/{udid}/bgv.png"
|
||
img.save(filePath)
|
||
LogManager.info("保存屏幕图像成功", udid)
|
||
print("保存了背景图")
|
||
time.sleep(2)
|
||
# 查找add图标
|
||
addX, addY = AiUtils.findImageInScreen("add", udid)
|
||
if addX != -1:
|
||
r = ControlUtils.clickLike(session, udid)
|
||
# 点赞成功。总数量减1，否则就滑到下一个视频。原因太多。不好判断。不如直接下一个视频
|
||
if r == True:
|
||
count -= 1
|
||
else:
|
||
client.swipe_up()
|
||
else:
|
||
print("没找有效视频")
|
||
client.swipe_up()
|
||
break
|
||
|
||
# 假装看几秒视频
|
||
time.sleep(5)
|
||
client.swipe_up()
|
||
|
||
if count == 0:
|
||
ControlUtils.clickBack(session)
|
||
break
|
||
|
||
# 如果打开视频失败。说明该主播没有视频
|
||
if cellClickResult == True:
|
||
print("点击了视频")
|
||
session.appium_settings({"snapshotMaxDepth": 15})
|
||
time.sleep(2)
|
||
print("重新设置了匹配深度")
|
||
viewAnchorVideo()
|
||
time.sleep(2)
|
||
print("视频看完了")
|
||
session.appium_settings({"snapshotMaxDepth": 25})
|
||
# 点击关注按钮
|
||
followButton = AiUtils.getFollowButton(session)
|
||
if followButton.exists:
|
||
followButton.click()
|
||
else:
|
||
goBack(2)
|
||
return
|
||
|
||
time.sleep(2)
|
||
AiUtils.getSendMesageButton(session)
|
||
time.sleep(2)
|
||
|
||
# 查找聊天界面中的输入框节点
|
||
chatInput = session.xpath("//*[className='XCUIElementTypeTextView']")
|
||
if chatInput.exists:
|
||
print("找到了")
|
||
# 准备发送一条信息
|
||
chatInput.click()
|
||
time.sleep(2)
|
||
text = random.choice(prologueList)
|
||
msg = Requester.translation(text, anchorCountry)
|
||
chatInput.set_text(msg + "\n")
|
||
time.sleep(1)
|
||
else:
|
||
print("无法发送信息，换个主播")
|
||
# 接着下一个主播
|
||
goBack(3)
|
||
|
||
else:
|
||
print("没找到主播的第一个视频")
|
||
# 删除当前数据
|
||
removeModelFromAnchorList(anchor)
|
||
goBack(2)
|
||
|
||
|
||
|
||
def test(self, udid):
|
||
client = wda.USBClient(udid)
|
||
session = client.session()
|
||
session.appium_settings({"snapshotMaxDepth": 10})
|
||
|
||
print(client.source())
|
||
|
||
# manager = ScriptManager()
|
||
# manager.test("eca000fcb6f55d7ed9b4c524055214c26a7de7aa")
|
||
# manager.growAccount("eca000fcb6f55d7ed9b4c524055214c26a7de7aa")
|
||
|