使用Python快速启动多个PC客户端

作为一枚游戏测试,日常工作的时候,经常需要同时开着多个游戏账号进行测试。

以Unity打包出来的PC客户端为例,每次都需要双击运行exe程序,等着游戏启动后,因为同时开着多个客户端,所以还需要拖动客户端,来调整游戏在屏幕上的位置,防止遮挡,方便查看。

操作流程差不多就像这样子(本文用记事本来替代游戏客户端):

这个操作其实还是挺繁琐的,尤其你开了3个以上客户端的时候,能不能将上边的操作做成工具来自动完成呢?

首先我们分析下这个操作的流程

那接下来用Python各个击破吧~

启动游戏

启动游戏只需要一句代码~

subprocess.Popen(r"C:\Windows\system32\notepad.exe")

subprocess是Python自带的子进程管理模块,定义有数个创建子进程的函数,也提供了一些管理标准流(standard stream)和管道(pipe)的工具,从而在进程间使用文本通信。简单理解就是,通过CMD敲的命令,都基本可以用subprocess来实现批量处理。

计算客户端的坐标位置

基本思路:

  • 确认游戏的窗口大小,如800*450
  • 计算整个屏幕大小,可以放多少个800*450的客户端,并计算出每个客户端的坐标位置

如何获取屏幕大小呢?Python在Windows平台有一个库叫pywin32,它为Python提供访问Windows API的扩展,提供了齐全的Windows常量、接口、线程以及COM机制等。

安装pywin32,在sourceforge下载对于你电脑Py版本的exe安装包,直接运行即可。

pywin32提供了一个方法win32gui.GetClientRect,可以获取窗口坐标及大小,只需要将桌面窗口句柄作为参数传进去即可。

所以获取桌面分辨率的方式就是:

_, _, width, height = win32gui.GetClientRect(win32gui.GetDesktopWindow())

接下来就是计算在桌面上,可以摆放多少个游戏客户端了,就像下图这个样子。

这个不复杂,就直接上代码了,计算出桌面可以摆放的客户端的左上角坐标,保存到一个列表里边备用

game_width = 450
game_height = 800

_, _, width, height = win32gui.GetClientRect(win32gui.GetDesktopWindow())

x_num = int(width / game_width)
y_num = int(height / game_height)

result = list()

for y in range(y_num):
    for x in range(x_num):
        result.append((x * game_width, y * game_height, game_width, game_height))

print(result)

output: [(0, 0, 450, 800), (450, 0, 450, 800), (900, 0, 450, 800), (1350, 0, 450, 800)]

调整窗口大小并拖动到指定位置


win32gui.MoveWindow(hwnd, x, y, width, height, bRepaint) 提供了一个移动窗口的方式,函数的几个参数分别表示句柄,起始点x坐标,y坐标,宽度,高度,是否重绘界面。

app_pid = subprocess.Popen(r"C:\Windows\system32\notepad.exe").pid  # 获取进程pid
time.sleep(1)
hwnd_list = []
win32gui.EnumWindows(lambda _hwnd, _hwnd_list: _hwnd_list.append(_hwnd), hwnd_list)  # 获取当前全部的窗口句柄
app_hwnd = None
for hwnd in hwnd_list:
    hid, pid = win32process.GetWindowThreadProcessId(hwnd)  # 根据窗口句柄取出窗口的hid和pid
    if pid == app_pid:
        app_hwnd = hwnd
        break
if not app_hwnd:
    raise Exception("没有找到hwnd")

app_position = result.pop(0)
win32gui.MoveWindow(app_hwnd, *app_position, True)  # 移动窗口位置

再封装下


import win32gui
import win32process
import subprocess
import time


def calculate_app_positions(app_width, app_height, scale):
    _, _, width, height = win32gui.GetClientRect(win32gui.GetDesktopWindow())
    final_app_width = int(app_width * scale)
    final_app_height = int(app_height * scale)
    x_num = int(width / final_app_width)
    y_num = int(height / final_app_height)

    positions = list()

    for y in range(y_num):
        for x in range(x_num):
            positions.append((x * final_app_width, y * final_app_height, final_app_width, final_app_height))
    return positions


def start_app(app_path, num, app_width, app_height, scale):
    app_positions = calculate_app_positions(app_width, app_height, scale)
    for i in range(num):
        app_pid = subprocess.Popen(app_path).pid  # 获取进程pid
        time.sleep(1)
        hwnd_list = []
        win32gui.EnumWindows(lambda _hwnd, _hwnd_list: _hwnd_list.append(_hwnd), hwnd_list)  # 获取当前全部的窗口句柄
        app_hwnd = None
        for hwnd in hwnd_list:
            hid, pid = win32process.GetWindowThreadProcessId(hwnd)  # 根据窗口句柄取出窗口的hid和pid
            if pid == app_pid:
                app_hwnd = hwnd
                break
        if not app_hwnd:
            raise Exception("没有找到hwnd")

        app_position = app_positions.pop(0)
        win32gui.MoveWindow(app_hwnd, *app_position, True)  # 移动窗口位置


if __name__ == '__main__':

    app_path = r"C:\Windows\system32\notepad.exe"
    app_width = 450
    app_height = 800

    result = input("输入要打开的客户端数量和显示比例,默认比例是1.0,例子:3 0.75\n")
    result = result.split()
    if len(result) == 2:
        num, scale = result
    else:
        num = result[0]
        scale = 1
    start_app(app_path, int(num), app_width, app_height, float(scale))

最终运行效果~

你可能感兴趣的:(使用Python快速启动多个PC客户端)