首页 > 编程语言 >用Python编写一个封装mstsc的RDP批量管理工具

用Python编写一个封装mstsc的RDP批量管理工具

时间:2023-03-24 13:58:11浏览次数:52  
标签:RDP currentQTableWidgetItem mstsc Python self hostjson tableWidget row 255

要实现的功能

1. 调用系统的mstsc命令来实现远程桌面
2. 确保连接过程不可见,实现直接连接的效果
3. 支持窗口和全屏连接
4. 支持手动添加新的桌面
5. 支持显示桌面列表
6. 支持删除列表
7. 隐藏桌面的地址和密码,只显示别名和用户名
8. 当选中项目时,取消隐藏

项目地址

点击传送 Ym-rdp

部分代码展示

import os
import win32crypt
import sys
import binascii
#导入QT组件
from PyQt5.QtWidgets import QMainWindow,QApplication,QHeaderView,QAbstractItemView,QTableWidgetItem,QMenu,QMessageBox,QTableWidget
from PyQt5 import QtCore
from PyQt5.QtGui import QIcon,QBrush,QColor,QCursor,QPalette,QPixmap
#导入主界面UI
from rdpui import Ui_Dialog
import query_ico_rc

class ConnDesk:
    def __init__(self,username,password,host,scre=1) -> None:
        '''
        scre: 0 全屏,1默认分辨率连接
        '''
        # 临时rdp文件
        self.rdpTemp = "rdp.rdp"

        self.username = username
        self.password = password
        self.host = host
        self.scre = scre

    def AddPwd(self):
        try:
            pwdHash = win32crypt.CryptProtectData(self.password.encode('utf-16-le'), u'psw', None, None, None, 0)
            self.password = binascii.hexlify(pwdHash).decode("utf-8")
            # rdp模板文件
            self.model = f'''
screen mode id:i:1
desktopwidth:i:1650
desktopheight:i:970
screen mode id:i:{self.scre}
session bpp:i:24
winposstr:s:2,3,188,8,1062,721
full address:s:{self.host}
compression:i:1
keyboardhook:i:2
audiomode:i:0
redirectdrives:i:0
redirectprinters:i:0
redirectcomports:i:0
redirectsmartcards:i:0
displayconnectionbar:i:1
autoreconnection enabled:i:1
authentication level:i:0
username:s:{self.username}
domain:s:MyDomain
alternate shell:s:
shell working directory:s:
password 51:b:{self.password}
disable wallpaper:i:1
disable full window drag:i:1
disable menu anims:i:1
disable themes:i:0
disable cursor setting:i:0
bitmapcachepersistenable:i:1
'''
            rdpTemp = open(self.rdpTemp,"w",encoding="utf-8")
            rdpTemp.write(self.model)
            rdpTemp.close()
            print("写入文件成功")
            return self.password
        except:
            return False

    def conn(self):
        cmd = f"mstsc {self.rdpTemp}"
        os.system(cmd)


#继承视窗和UI
class MainCode(QMainWindow,Ui_Dialog):

    def __init__(self):
        #初始化
        QMainWindow.__init__(self)
        Ui_Dialog.__init__(self)

        #将UI生成传到self
        self.setupUi(self)
        self.filedb = './ymhost.json'

        # 数据文件
        if os.path.exists(self.filedb):
            self.relodb()

        # label_5自定义标题栏图标
        self.label_5.setScaledContents(True)
        pixmap = QPixmap(':/app.ico')
        self.label_5.setPixmap(pixmap)

        # 引入全部样式
        self.setStyleSheet(MyStyle())

        # 隐藏标题栏
        self.setWindowFlags(QtCore.Qt.CustomizeWindowHint)
        # 去边框
        self.setWindowFlags(QtCore.Qt.FramelessWindowHint)
        # 将顶级画布设为透明
        self.setAttribute(QtCore.Qt.WA_TranslucentBackground)
        # 窗口透明度
        # self.setWindowOpacity(0.9)

        #不显示行名称
        self.tableWidget.verticalHeader().setVisible(False)
        #让表格铺满
        self.tableWidget.horizontalHeader().setSectionResizeMode(QHeaderView.Stretch)

        #绑定表格右键事件
        self.tableWidget.setContextMenuPolicy(QtCore.Qt.CustomContextMenu)
        self.tableWidget.customContextMenuRequested[QtCore.QPoint].connect(self.del_table_line)

        #表格数据更新时的事件
        self.tableWidget.itemChanged.connect(self.on_update)# 内容已更新时

        #绑定添加按钮的事件
        self.pushButton.clicked.connect(self.add_desk)

        # 最小化按钮
        self.pushButton_4.clicked.connect(self.showMinimized)
        # 关闭按钮
        self.pushButton_2.clicked.connect(self.close)


    def mousePressEvent(self, event): #鼠标拖拽窗口移动
        if event.button() == QtCore.Qt.LeftButton:
            self.m_flag = True
            self.m_Position = event.globalPos() - self.pos()  # 获取鼠标相对窗口的位置
            event.accept()
            self.setCursor(QCursor(QtCore.Qt.OpenHandCursor))  # 更改鼠标图标

    def mouseMoveEvent(self, QMouseEvent): #鼠标拖拽窗口移动
        if QtCore.Qt.LeftButton and self.m_flag:
            self.move(QMouseEvent.globalPos() - self.m_Position)  # 更改窗口位置
            QMouseEvent.accept()

    def mouseReleaseEvent(self, QMouseEvent): #鼠标拖拽窗口移动
        self.m_flag = False
        self.setCursor(QCursor(QtCore.Qt.ArrowCursor))


    def handleItemEntered(self,item):
        if item.column() == 1 or item.column() == 3:
            item.setForeground(QBrush(QColor(0,0,0)))

    def handleItemExited(self, item):
        if item.column() == 1 or item.column() == 3:
            item.setForeground(QBrush(QColor(255,255,255)))

    def cell_changed(self,row,column):
        print(row, column)

    def slot_currentCellChanged(self, row, column):
        print("点击了单元格:" + str(row) + "," + str(column))

    def on_update(self):
        if len(self.tableWidget.selectedItems()) > 1:
            for currentQTableWidgetItem in self.tableWidget.selectedItems():
                if currentQTableWidgetItem.column() == 1:
                    try:
                        hostjson[addline]["host"] = currentQTableWidgetItem.text()
                    except:
                        pass
                    finally:
                        currentQTableWidgetItem.setForeground(QBrush(QColor(255,255,255))) # 隐藏地址
                if currentQTableWidgetItem.column() == 3:
                    try:
                        hostjson[addline]["password"] = currentQTableWidgetItem.text()
                    except:
                        pass
                    finally:
                        currentQTableWidgetItem.setForeground(QBrush(QColor(255,255,255))) # 隐藏密码
            return
        if os.path.exists("./ymhost.json"):
            try:
                hostjson = eval(open(self.filedb,"r",encoding="utf-8").read())
            except:
                hostjson = {"0":{}}
            filedata = open(self.filedb,"w",encoding="utf-8")
        else:
            hostjson = {}
            filedata = open(self.filedb,"w",encoding="utf-8")
        # print(hostjson)
        for currentQTableWidgetItem in self.tableWidget.selectedItems():
            addline = str(currentQTableWidgetItem.row())
            # print(currentQTableWidgetItem.row(), currentQTableWidgetItem.column(), currentQTableWidgetItem.text())
            # print(hostjson[currentQTableWidgetItem.row()]["name"])
            # 更新的行
            if currentQTableWidgetItem.column() == 0:
                # print(hostjson[currentQTableWidgetItem.row()]["name"],currentQTableWidgetItem.text())
                hostjson[addline]["name"] = currentQTableWidgetItem.text()
            if currentQTableWidgetItem.column() == 1:
                try:
                    hostjson[addline]["host"] = currentQTableWidgetItem.text()
                except:
                    pass
                finally:
                    currentQTableWidgetItem.setForeground(QBrush(QColor(255,255,255))) # 隐藏地址
            if currentQTableWidgetItem.column() == 2:
                hostjson[addline]["username"] = currentQTableWidgetItem.text()
            if currentQTableWidgetItem.column() == 3:
                try:
                    hostjson[addline]["password"] = currentQTableWidgetItem.text()
                except:
                    pass
                finally:
                    currentQTableWidgetItem.setForeground(QBrush(QColor(255,255,255))) # 隐藏密码
        filedata.write(str(hostjson))
        filedata.close()
        self.tableWidget.update()

    # 重载数据
    def relodb(self,**args):
        if os.path.exists("./ymhost.json"):
            data = open(self.filedb,"r",encoding="utf-8")
            try:
                data = eval(data.read())
                for i in data:
                    i = data[i]
                    # 跳过空数据
                    if len(i) == 0:
                        continue
                    name,host,username,password = i["name"],i["host"],i["username"],i["password"]
                    info = [name,host,username,password]
                    row = self.tableWidget.rowCount()
                    self.tableWidget.insertRow(row)
                    for listed in range(len(info)):
                        item = QTableWidgetItem(str(info[listed]))
                        # 将地址和密码置为与背景色相同,即隐藏地址和密码
                        if listed == 1 or listed == 3:
                            item.setForeground(QBrush(QColor(255,255,255)))
                        item.setTextAlignment(QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter)
                        self.tableWidget.setItem(row,listed,item)
                    self.tableWidget.update()
            except:
                print("加载数据出错了")

    # 修改关闭程序的事件
    def closeEvent(self, event):
        print("保存并退出程序")
        event.accept()

        # 提示框例子
        # result = QtWidgets.QMessageBox.question(self, "将关闭程序", "是否保存修改的数据并关闭程序?", QtWidgets.QMessageBox.Yes | QtWidgets.QMessageBox.No)
        # if(result == QtWidgets.QMessageBox.Yes):
        #     print("保存退出")
        #     event.accept()
        # else:
        #     print("取消关闭")
        #     event.ignore()

    # 单元格被双击时
    def on_click(self):
        # for currentQTableWidgetItem in self.tableWidget.selectedItems():
        #     print(currentQTableWidgetItem.row(), currentQTableWidgetItem.column(), currentQTableWidgetItem.text())
        pass

     #右键表格功能
    def del_table_line(self, pos):
        pop_menu = QMenu()
        conn_desk = pop_menu.addAction('默认连接')
        conn_maxdesk = pop_menu.addAction('全屏连接')
        # edit_desk = pop_menu.addAction('编辑桌面')
        del_desk = pop_menu.addAction('删除桌面')
        # export_desk = pop_menu.addAction('导出桌面')

        # 获取右键菜单中当前被点击的是哪一项
        action = pop_menu.exec_(self.tableWidget.mapToGlobal(pos))

        # 选中连接桌面
        if action == conn_desk:
            print("普通连接")
            # 获取当前选中的行
            row = self.tableWidget.currentRow()
            # 获取当前选中行的主机地址
            host = self.tableWidget.item(row , 1).text()
            username = self.tableWidget.item(row , 2).text()
            password = self.tableWidget.item(row , 3).text()
            new = ConnDesk(username,password,host)
            print("密码:",new.AddPwd())
            # 开启连接
            new.conn()

        # 选中全屏连接
        if action == conn_maxdesk:
            print("全屏连接")
            # 获取当前选中的行
            row = self.tableWidget.currentRow()
            # 获取当前选中行的主机地址
            host = self.tableWidget.item(row , 1).text()
            username = self.tableWidget.item(row , 2).text()
            password = self.tableWidget.item(row , 3).text()
            new = ConnDesk(username,password,host,0)
            print("密码:",new.AddPwd())
            # 开启连接
            new.conn()

        # 选中删除桌面
        if action == del_desk:
            selected_items = self.tableWidget.selectedItems()
            if len(selected_items) == 0:  # 说明没有选中任何行
                return
            selected_items = [selected_items[i] for i in range(len(selected_items)-1, -1, -7)]
            # 将选定行的行号降序排序,从索引大的行开始删除
            for items in selected_items:
                self.tableWidget.removeRow(self.tableWidget.indexFromItem(items).row())

            # 从数据文件中删除,若列表无数据,则会在上一步结束执行
            hostjson = eval(open(self.filedb,"r",encoding="utf-8").read())
            filedata = open(self.filedb,"w",encoding="utf-8")
            # 要删除的数据对应索引
            rowline = str(self.tableWidget.rowCount())
            hostjson.pop(rowline)
            filedata.write(str(hostjson))
            filedata.close()
            self.tableWidget.update()

    # 添加一条记录
    def add_desk(self):
            name = self.lineEdit.text()
            host = self.lineEdit_2.text()
            username = self.lineEdit_3.text()
            password = self.lineEdit_4.text()
            if len(name) == 0 or len(host) == 0 or len(username) == 0 or len(password) == 0:
                QMessageBox.information(self, "新增失败", "有不完整的内容",QMessageBox.Yes)
                return
            info = [name,host,username,password]
            row = self.tableWidget.rowCount()
            self.tableWidget.insertRow(row)
            if os.path.exists("./ymhost.json"):
                # print("文件存在")
                try:
                    hostjson = eval(open(self.filedb,"r",encoding="utf-8").read())
                    if len(hostjson) == 0:
                        # print("长度为0")
                        hostjson = {"0":{}}
                except:
                    hostjson = {"0":{}}
                filedata = open(self.filedb,"w",encoding="utf-8")
            else:
                # print("文件不存在")
                hostjson = {"0":{}}
                filedata = open(self.filedb,"w",encoding="utf-8")
            addlen = str(self.tableWidget.rowCount() -1)
            hostjson[addlen] = {}
            for listed in range(len(info)):
                # print("新增内容:",info[listed])
                item = QTableWidgetItem(str(info[listed]))
                item.setTextAlignment(QtCore.Qt.AlignHCenter | QtCore.Qt.AlignVCenter)
                self.tableWidget.setItem(row,listed,item)
                if listed == 0:
                    hostjson[addlen]["name"] = info[listed]
                if listed == 1:
                    hostjson[addlen]["host"] = info[listed]
                    item.setForeground(QBrush(QColor(255,255,255))) # 隐藏地址
                if listed == 2:
                    hostjson[addlen]["username"] = info[listed]
                if listed == 3:
                    hostjson[addlen]["password"] = info[listed]
                    item.setForeground(QBrush(QColor(255,255,255))) # 隐藏密码
            # print(hostjson)
            filedata.write(str(hostjson))
            filedata.close()
            self.tableWidget.update()

def MyStyle():
    return """
        QPushButton {
            background-color: #FFFFFF;
            border-radius: 5px;
            margin: 5px;
        }
        QFrame {
            background-color: #C9DEE6;
            border-radius: 15px;
        }
        QFrame#frame_3 {
            background-color: #9BC8DB;
            border-radius: 15px;
        }
        QLabel#label_5 {
            background-color: #FFFFFF;
            margin: 5px;
            border-radius: 5px;
        }
        QLabel#label,#label_2,#label_3,#label_4 {
            background-color: #ffffff;
            border-radius: 5px;
            margin-left: 5px;
            margin-right: 5px;
        }
        QLineEdit {
            background-color: #ffffff;
            border-radius: 5px;
            margin-left: 5px;
            margin-right: 5px;
            padding-left: 2px;
            padding-right: 2px;
        }
        QTableWidget {
            background-color: rgb(255,255,255);
            border-radius: 15px;
        }
        """

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainCode()
    window.setWindowIcon(QIcon(":/app.ico"))
    window.setFixedSize(window.width(), window.height())
    window.show()
    sys.exit(app.exec_())

预览

image

下载地址

https://github.com/HG-ha/Ym-rdp/releases/download/v0.1.4/RDP-win.exe

标签:RDP,currentQTableWidgetItem,mstsc,Python,self,hostjson,tableWidget,row,255
From: https://www.cnblogs.com/ymer/p/17251306.html

相关文章