import winreg
import os
tcp = "SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\Wds\\rdpwd\\Tds\\tcp"
RDP_Tcp = "SYSTEM\\CurrentControlSet\\Control\\Terminal Server\\WinStations\\RDP-Tcp"
reg1 = winreg.CreateKey(winreg.HKEY_LOCAL_MACHINE, tcp)
port1 = winreg.QueryValueEx(reg1, 'PortNumber')
print("原端口:" + str(port1[0]))
print("新端口:")
myport = input()
winreg.SetValueEx(reg1, 'PortNumber', 0, winreg.REG_DWORD, int(myport))
port1 = winreg.QueryValueEx(reg1, 'PortNumber')
reg2 = winreg.CreateKey(winreg.HKEY_LOCAL_MACHINE, RDP_Tcp)
winreg.SetValueEx(reg2, 'PortNumber', 0, winreg.REG_DWORD, int(myport))
port1 = winreg.QueryValueEx(reg1, 'PortNumber')
print("修改成功")
os.system("pause")