Python自动化运维|Python操控CMD登录服务器|Python操控程序

uiautomation库学习

介绍

uiautomation这个模块是由一个大佬利用自己空闲时间开发的库。主要根据Win32这个微软的标准库来封装了一些能够操作桌面应用程序的方法。

uiautomation封装了微软UIAutomation API，支持自动化Win32，MFC，WPF，Modern UI(Metro UI), Qt, IE, Firefox(version<=56 or >=60, Firefox57是第一个Rust开发版本,前几个Rust开发版本个人测试发现不支持),Chrome和基于Electron开发的应用程序(Chrome浏览器和Electron应用需要加启动参数–force-renderer-accessibility才能支持UIAutomation).

使用

uiautomation中封装了Windows UIAutomation中的各个Control和Pattern.
Control类型有ButtonControl, TextControl, TreeControl等等。
Pattern类型有ExpandCollapsePattern,InvokePattern等等。
实际使用时，要用Control或Pattern对象来获取控件信息或操作控件。
uiautomation根据你提供的控件属性在控件树中从上往下查找控件。

实际上是更具微软的结构树，来查找可以控制的组件。
在结构树当中查找，查找到控件后对控件进行操作。
searchFromControl = None, 从哪个控件开始查找，如果为None，从根节点Desktop开始查找
searchDepth = 0xFFFFFFFF, 搜索深度
searchInterval = SEARCH_INTERVAL, 搜索间隔
foundIndex = 1 ，搜索到的满足搜索条件的控件索引，索引从1开始
Name 控件名字
SubName 控件部分名字
RegexName 使用re.match匹配符合正则表达式的名字，Name,SubName,RegexName只能使用一个，不能同时使用
ClassName 类名字
AutomationId 控件AutomationId
ControlType 控件类型
Depth 控件相对于searchFromControl的精确深度
Compare 自定义比较函数function(control: Control, depth: int)->bool

searchDepth和Depth的区别是：
searchDepth在指定的深度范围内（包括1~searchDepth层中的所有子孙控件）搜索第一个满足搜索条件的控件
Depth只在Depth所在的深度（如果Depth>1，排除1~searchDepth-1层中的所有子孙控件）搜索第一个满足搜索条件的控件

在编写代码时可以配合inspect.exe使用
在这个软件当中可以查清楚的查看到结构树，便于定位到需要的控件
在这里插入图片描述

简单的使用例子 – 在CMD当中使用SSH连接服务器，输入指令获取结果

通过Win + R 打开CMD窗口：

auto.SendKeys('{Win}{R}{enter}')  # 通过Win+R打开CMD窗口
cmdShellWindow = auto.WindowControl(searchDepth=1,ClassName="ConsoleWindowClass")  # 获取CMD窗口控制
cmdShellWindow.SetTopmost(True)  # 将窗口设置为最上层

使用SSH，连接服务器

loginCmd = f'ssh {user}@{ip}'  # user为linux服务器的用户名，ip为服务器地址 
auto.SendKeys(loginCmd + "{enter}")

执行指令，获取结果

for line in open(cmdFile, 'r', encoding='utf-8').readlines():
    # 获取CMD界面结果
    head = getCommentCMD(CMD)
    auto.SendKeys('clear{enter}')
    line = line.replace('\n', '') if '\n' in line else line
    auto.SendKeys(line)
    auto.SendKeys('{enter}')
    time.sleep(0.5)
    endRes = waitResult(CMD, head)
    if endRes:
        txtContent = CMD.DocumentControl(searchDepth=1).GetTextPattern().DocumentRange.GetText()
        txtContent = txtContent.split('clear')[-1]
        writeResult(txtContent, line)
        continue
    else:
        writeResult('Falled!!')
        return

完整code

# -*- coding: utf-8 -*-
import time
import uiautomation as auto


def waitResult(CMD, head, waitTime=31):
    for count in range(1, waitTime + 1):
        time.sleep(1)
        end = getCommentCMD(CMD)
        print(f'head:{head}')
        print(f'end:{end}')
        if head == end:
            continue
        else:
            return True
    return False


def writeResult(res, cmd=None):
    resFile = './resLog.txt'
    resLog = open(resFile, 'a+', encoding='utf-8')
    content = res.split('\n')
    resLog.write('-' * 10 + str(cmd) + '-' * 10 + '\n')
    for line in content:
        line = line.strip()
        if len(line) == 0:
            continue
        resLog.write(line)
        resLog.write('\n')
    resLog.write('-' * 40 + '\n')
    resLog.write('\n')
    resLog.close()


def mainCMDWindowd(commandFile):
    auto.SendKeys('{Win}{R}{enter}')  # 通过Win+R打开CMD窗口
    cmdShellWindow = auto.WindowControl(searchDepth=1, ClassName="ConsoleWindowClass")  # 获取CMD窗口控制
    cmdShellWindow.SetTopmost(True)
    res = loginSSH(cmdShellWindow, 'xxx', '0.0.0.0', 'xxxx') # 这里为服务用户名、ip、密码
    if res:
        executeCMD(cmdShellWindow, commandFile)
        cmdShellWindow.TitleBarControl().ButtonControl(Name="关闭").Click()


def executeCMD(CMD, cmdFile):
    for line in open(cmdFile, 'r', encoding='utf-8').readlines():
        # 获取CMD界面结果
        head = getCommentCMD(CMD)
        auto.SendKeys('clear{enter}')
        line = line.replace('\n', '') if '\n' in line else line
        auto.SendKeys(line)
        auto.SendKeys('{enter}')
        time.sleep(0.5)
        endRes = waitResult(CMD, head)
        if endRes:
            txtContent = CMD.DocumentControl(searchDepth=1).GetTextPattern().DocumentRange.GetText()
            txtContent = txtContent.split('clear')[-1]
            writeResult(txtContent, line)
            continue
        else:
            writeResult('Falled!!')
            return
        # 获取指令执行后的结果


def getCommentCMD(CMD):
    txtElement = CMD.DocumentControl(searchDepth=1).GetTextPattern().DocumentRange.GetText()
    res = []
    for line in txtElement.split('\n'):
        line = line.strip()
        if len(line) == 0:
            continue
        res.append(line)
    return res[-2]


def loginSSH(CMD, user, ip, pwd):
    loginCmd = f'ssh {user}@{ip}'
    head = getCommentCMD(CMD)
    auto.SendKeys(loginCmd + "{enter}")
    endRes = waitResult(CMD, head)
    if endRes:
        auto.SendKeys(pwd + "{enter}")
        return True
    else:
        return False


if __name__ == '__main__':
    cmdfile = './CMD.txt'
    mainCMDWindowd(cmdfile)

代码使用方法：

需要准备好要在服务器上运行的指令，放在同目录下的CMD.txt文本文件当中

如：

扫描二维码关注公众号，回复： 15509260 查看本文章

CMD.txt:

ls
pwd
cd clash/
cat config.yaml

最终的运行结果会自动生成在同目录下的resLog.txt文件当中：

resLog.txt

----------ls----------
[root@GitLabHost ~]# ls
clash  pydInstaller  python3  qqchatgpt
[root@GitLabHost ~]#
----------------------------------------

----------pwd----------
[root@GitLabHost ~]# pwd
/root
[root@GitLabHost ~]#
----------------------------------------

----------cd clash/----------
[root@GitLabHost ~]# cd clash/
[root@GitLabHost clash]#
----------------------------------------

----------cat config.yaml----------
surance.com,Proxy
xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
# GeoIP China
- GEOIP,CN,Domestic
- MATCH,Others
[root@GitLabHost clash]#
----------------------------------------

运行效果：

需要自己实践才能知道。

代码实现功能：

自动化登录服务器，在服务器上输入内容获取结果，实现自动化运维的方法。

总结：

通过一个简单的案例，提供了一个通过Python控制桌面应用程序的方法，通过uiautomation模块可以拓展到控制QQ、微信、IE浏览器等等，更进一步的解放人的双手。但因为查找的方法，在树结构当中查找目标的效率有点慢，需要一层一层的查找，缺少优化的方法。

Python自动化运维|Python通过CMD登录服务器|Python操控程序