frp GUI客户端

之前的frp客户端还是命令行的形式，朋友问我能不能做GUI
我打算试试，就花了几天学了下 tkinter，试着做了一个：

import os
from tkinter import *
from tkinter import messagebox


class Application(Frame):

    def __init__(self, master=None):
        super().__init__(master)
        self.master = master
        self.pack()
        self.createWidget()

    def createWidget(self):
        # 输入ip
        self.labelIp = Label(self, text="本地IP")
        self.labelIp.grid(row=0, column=0, sticky=SW)
        self.ip = StringVar()
        self.entryIp = Entry(self, textvariable=self.ip)
        self.entryIp.grid(row=1, column=0)
        self.ip.set("127.0.0.1")

        # 输入端口
        self.labelPort = Label(self, text="端口")
        self.labelPort.grid(row=0, column=1, sticky=SW)
        self.port = StringVar()
        self.entryPort = Entry(self, textvariable=self.port)
        self.entryPort.grid(row=1, column=1)

        # 隧道名称
        self.labelName = Label(self, text="隧道名称")
        self.labelName.grid(row=2, column=0, sticky=SW)
        self.name = StringVar()
        self.entryName = Entry(self, textvariable=self.name)
        self.entryName.grid(row=3, column=0)

        # 隧道类型
        self.labeltyp = Label(self, text="隧道类型")
        self.labeltyp.grid(row=2, column=1, sticky=SW)
        self.typ = StringVar(self)
        self.typ.set("TCP")
        self.om01 = OptionMenu(self, self.typ, "TCP", "UDP")
        self.om01.grid(row=3, column=1)

        self.btn01 = Button(self, text="创建", command=self.establish)
        self.btn01.grid(row=4, column=0, sticky=EW)
        self.btn02 = Button(self, text="启动", command=self.run)
        self.btn02.grid(row=4, column=1, sticky=W)
        self.btn03 = Button(self, text="停止", command=self.end)
        self.btn03.grid(row=4, column=1, sticky=E)

    def establish(self):
        if self.typ.get() == "TCP":
            typ = "tcp"
        if self.typ.get() == "UDP":
            typ = "udp"
        with open("frpc.ini", "w") as f:
            list = ['[common]\n',
                    'server_addr = 这里是服务器地址\n',
                    'server_port = 7000\n',
                    'token = 这里是密钥\n'
                    '\n',
                    '[' + self.name.get() + ']\n',
                    'type = ' + typ + '\n',
                    'local_ip = ' + self.ip.get() + '\n',
                    'local_port = ' + self.port.get() + '\n',
                    'remote_port = 6001']
            f.writelines(list)

    def run(self):
        os.popen('frpc.exe')
        messagebox.showinfo("启动成功", "您的本地地址：" + self.ip.get() + ":" + self.port.get() + "已映射到服务器地址：[这里是服务器地址]:6001")

    def end(self):
        os.system('chcp 65001')
        os.system('taskkill /F /IM frpc.exe')


if __name__ == '__main__':
    root = Tk()
    root.geometry("300x150+600+250")
    root.title("FRPGUI")
    app = Application(master=root)
    def end():
        os.system('taskkill /F /IM frpc.exe')
        root.destroy()
    root.protocol('WM_DELETE_WINDOW', end)
    root.mainloop()

效果：

目前还不太美观
学着视频里的老师用的面向对象的方式写的，比一般的写法要复杂些，花了好些时间才理解

遇到的问题

当然在写代码的时候也遇到了很多问题

启动

一开始的启动方法，我用的是：

def run(self):
    os.system('frpc.exe')

但是点击启动后窗口直接卡住未响应了

猜测是因为os.system 会将运行结果输出在命令行导致GUI程序无法正确运行
又或者是因为GUI程序需要等os.system运行结束才能继续运行，而frpc.exe是持续运行的
于是我选择将代码改成：

def run(self):
    os.system('start cmd /K frpc.exe')

打开一个新的cmd窗口来运行frpc.exe
程序果然能正常运行了，但是一个突然出现一个CMD的大黑框真的很丑诶。如果还是会出现CMD窗口，我这GUI界面不是白做了？
尝试用代码将CMD窗口隐藏，但是还是会有一瞬间出现CMD窗口
查阅了os模块的资料后我发现，只要用os.popen来运行frpc.exe就不会卡住

def run(self):
    os.popen('frpc.exe')

猜测是因为os.popen不会将运行结果打印出来，等于是直接在后台运行

停止

虽然加了停止按钮，但是也不能排除不小心把窗口直接关掉的情况
这时frpc.exe仍然在运行，只能在任务管理器里关闭或者是手动kill进程
于是我加了一段代码：

def end():
    os.system('taskkill /F /IM frpc.exe')
    root.destroy()
root.protocol('WM_DELETE_WINDOW', end)

监听窗口关闭事件，在窗口关闭前先运行taskkill /F /IM frpc.exe来杀死frpc.exe进程

检测是否成功启动

目前我还没有做到，不管是否成功启动，只要按下按钮都会提示启动成功

原因是当我试图打印出os.popen的值时：

def run(self):
    cmd = os.popen('frpc.exe')
    print(cmd.read())

窗口又无响应了
甚至我只是给一个变量赋值为cmd.read()都不行：

def run(self):
    cmd = os.popen('frpc.exe')
    test = cmd.read()

这……
这个地方卡了我好久，我开摆了

就这么过了几天
我重新翻看frp的文档时发现，客户端的配置文件中有一个可选参数是log_file，可以将日志写入到指定的文件中
那么我再从文件中获取日志不就可以了？
问题解决！

def run(self):
    os.popen('frpc.exe')
    time.sleep(0.5)  # 等待0.5秒
    with open("frpc.log", "r") as f:
        self.log = str(f.readlines()[-3:])  # 获取最新的三行
        print(self.log)
        if "[W]" in self.log:
            os.system('taskkill /F /IM frpc.exe')
            if "port already" in self.log:
                messagebox.showerror("启动失败", "远程端口已被使用，请改用其他端口\n\n程序已禁用自选远程端口功能，请联系小叶子")
            elif "proxy name" in self.log:
                messagebox.showerror("启动失败", "隧道名称" + self.name.get() + "已被使用，请检查并更换一个隧道名称")
            else:
                messagebox.showerror("启动失败", "出现了没有预料的错误，请检查日志文件frpc.log，将日志文件发给小叶子")
        else:
            messagebox.showinfo("启动成功", "您的本地地址：" + self.ip.get() + ":" + self.port.get() +
                                "已映射到服务器地址：[这里是服务器地址]:6001")

睡眠0.5秒是因为，不等待一会的话读取出来的依然是上次的日志