folding@SUSTech.py

from tkinter import *
import os
import sys
import subprocess
import threading
from login import *
import logging
from multiprocessing import Queue

global GUI_log_here
GUI_log_here = False

global login_method

global username

global thread_should_end

class TextHandler(logging.Handler):
    def __init__(self, txtOutput, level=logging.DEBUG, formatter=None):
        logging.Handler.__init__(self, level)
        self.txtOutput = txtOutput

    def emit(self, record):
        global GUI_log_here
        if GUI_log_here:
            insertLog(self.txtOutput, self.format(record))


global login_ok
login_ok = False

file_handler = logging.FileHandler(filename='FASLog.txt')
file_handler.setLevel(logging.DEBUG)

global formatter
formatter = logging.Formatter('%(asctime)s - %(levelname)s - "%(message)s" in - %(funcName)s - Line:%(lineno)s\n')
file_handler.setFormatter(formatter)

b_logger = logging.getLogger('test')
b_logger.setLevel(logging.DEBUG)
b_logger.addHandler(file_handler)

c_logger = logging.getLogger('echo')
c_logger.setLevel(logging.DEBUG)
global echo_formatter
echo_formatter = logging.Formatter('%(message)s')

# window for login 
def login_clicked():
    global login_screen
    login_screen = Toplevel(main_screen)
    login_screen.title("登录")
    login_screen.geometry("300x270")
    login_screen.resizable(0, 0) #Don't allow resizing in the x or y direction
    login_screen.iconbitmap("icon.ico")
    Label(login_screen, text="").pack()
    Label(login_screen, text="SUSTech CAS身份认证").pack()
    Label(login_screen, text="").pack()

    global username_verify
    global password_verify

    username_verify = StringVar()
    password_verify = StringVar()

    global username_login_entry
    global password_login_entry

    Label(login_screen, text="用户名 * ").pack()
    username_login_entry = Entry(login_screen, textvariable=username_verify)
    username_login_entry.pack()
    Label(login_screen, text="").pack()
    Label(login_screen, text="密码 * ").pack()
    password_login_entry = Entry(login_screen, textvariable=password_verify, show= '*')
    password_login_entry.pack()
    Label(login_screen, text="").pack()
    btn = Button(login_screen, text="登录", width=10, height=1, command=login_verify, cursor="hand2").pack()
    login_screen.bind('<Return>', login_verify)

    # fetch buffer usrn
    f = open("./FASconfig.conf")
    usrn_buf = f.read().strip()
    f.close()
    if usrn_buf == '':
        username_login_entry.focus()
    else:
        username_login_entry.insert(0, usrn_buf)
        password_login_entry.focus()
 
    b_logger.info("Initializing Login Window...")


# Implementing event on login button 
def login_verify(event=None):
    global username
    global password
    username = username_verify.get()
    password = password_verify.get()
    # buffer usrn
    f = open("./FASconfig.conf", "w")
    f.write(username)
    f.close()

    b_logger.debug("username: {}".format(username))
    b_logger.info("Verifying username and password...")

    # Try login
    if not Login(username, password):
        login_failure()
        b_logger.warning("login failure.")
    else:
        global config
        config = GetConfig(username, password, b_logger)
        if not config:
            config_failure()
            b_logger.warning("pulling config failure.")
        else:
            global login_ok
            login_ok = True
            b_logger.debug("Login successful, exiting window...")
            # ask for user name permission
            ask_for_permission()

# ask whether the user agrees to use sid as his/her user name
def ask_for_permission():
    global permission_screen
    permission_screen = Toplevel(login_screen)
    permission_screen.title("登录方式")
    permission_screen.geometry("300x210")
    permission_screen.resizable(0, 0) #Don't allow resizing in the x or y direction
    permission_screen.iconbitmap("icon.ico")
    Label(permission_screen, text="Folding@SUSTech", bg="blue", fg="white", width="300", height="2", font=("Calibri Bold", 13)).pack()
    Label(permission_screen, text="").pack()
    Button(permission_screen, text="匿名登录", height="2", width="30", command=anonymous_login, cursor="hand2").pack()
    Label(permission_screen, text="").pack()
    Button(permission_screen, text="使用学号登录（推荐）", height="2", width="30", command=sid_login, cursor="hand2").pack()
    b_logger.info("Asking for login approach...")

def anonymous_login():
    global login_method
    login_method = "anonymous"
    b_logger.debug("'Anonymous Login' selected")
    continue2LoginSuccess()

def sid_login():
    global login_method
    login_method = "sid"
    b_logger.debug("'SID Login' selected")
    continue2LoginSuccess()

def continue2LoginSuccess():
    delete_main_screen()
    print("正在写入配置文件...")
    b_logger.info("Writing config file")
    global config
    WriteConfig(config)
    print("配置文件写入成功，开始运行")
    b_logger.info("Config file written, starting execution...")

# Designing popup for login invalid password
def login_failure():
    global login_failure_screen
    login_failure_screen = Toplevel(login_screen)
    login_failure_screen.title("认证信息无效！")
    login_failure_screen.geometry("300x66")
    login_failure_screen.iconbitmap("icon.ico")
    Label(login_failure_screen, text="请检查用户名和密码").pack()
    Button(login_failure_screen, text="确定", command=delete_login_failure).pack()

# Designing popup for config_failure
def config_failure():
    global config_failure_screen
    config_failure_screen = Toplevel(login_screen)
    config_failure_screen.title("Failed to get config file.")
    config_failure_screen.geometry("300x66")
    config_failure_screen.iconbitmap("icon.ico")
    Label(config_failure_screen, text="Failed to get config file.").pack()
    Button(config_failure_screen, text="OK", command=delete_config_failure).pack()

# Deleting popups
def delete_main_screen():
    main_screen.destroy()

def delete_login_failure():
    login_failure_screen.destroy()

def delete_config_failure():
    config_failure_screen.destroy()

# Designing Main(first) window
def main_account_screen():
    global main_screen
    main_screen = Tk()
    main_screen.geometry("300x150")
    main_screen.resizable(0, 0) #Don't allow resizing in the x or y direction
    main_screen.title("Folding@SUSTech")
    main_screen.iconbitmap("icon.ico")
    Label(text="Folding@SUSTech", bg="blue", fg="white", width="300", height="2", font=("Calibri Bold", 13)).pack()
    Label(text="").pack()
    Button(text="认证身份", height="2", width="30", command=login_clicked, cursor="hand2").pack()
    b_logger.info("Initializing Main Window...")
    main_screen.mainloop()

fah_client_ref = None

# Designing FAS window
def fas_screen():
    global paused
    global fah_client_ref
    paused = False
    b_logger.info("Starting FAHClient with config.xml...")
    global login_method
    global start_config
    if login_method == 'sid':
        global username
        start_config = "FAHClient --config ./config.xml --user {}".format(username)
    elif login_method == 'anonymous':
        start_config = "FAHClient --config ./config.xml"
    else:
        b_logger.error("Fail to select any login method.")
        sys.exit(-1)
    #fah_client_ref = launchWithoutConsole(start_config)
    launchPIPEproc(start_config)
    
    global fas_screen
    fas_screen = Tk()
    fas_screen.geometry("700x400")
    fas_screen.resizable(0, 0) #Don't allow resizing in the x or y direction
    fas_screen.title("Folding@SUSTech")
    fas_screen.iconbitmap("icon.ico")
    Label(text="Folding@SUSTech", bg="blue", fg="white", width="300", height="2", font=("Calibri Bold", 13)).pack()
    Label(text="").pack()
    global btn_txt
    btn_txt = StringVar()
    btn = Button(textvariable=btn_txt, height="2", width="30", command=toggleFolding, cursor="hand2",
                 bg="white", fg="black", activeforeground="white", activebackground="black").pack()
    b_logger.info("Initializing FAHCLient with a pre-unpause signal...")
    
    Label(text="").pack()

    txtFrame = Frame(fas_screen, borderwidth=1, relief=SUNKEN)
    txtOutput = Text(txtFrame, wrap=WORD, height=18, width=90, borderwidth=0, bg="black", fg="white", state=DISABLED)
    vscroll = Scrollbar(txtFrame, orient=VERTICAL, command=txtOutput.yview)
    txtOutput['yscroll'] = vscroll.set
    vscroll.pack(side="right", fill="y")
    txtOutput.pack(side="left", fill="both", expand=True)
    txtFrame.pack()

    global formatter
    th = TextHandler(txtOutput)
    th.setFormatter(formatter)
    b_logger.addHandler(th)
    global echo_formatter
    th_echo = TextHandler(txtOutput)
    th_echo.setFormatter(echo_formatter)
    c_logger.addHandler(th_echo)

    global GUI_log_here
    GUI_log_here = True
    unpause()
    fas_screen.mainloop()

def insertLog(txtOutput, str):
    txtOutput.config(state=NORMAL)
    txtOutput.insert(END, str)
    txtOutput.config(state=DISABLED)
    txtOutput.yview_moveto(1)
    

def toggleFolding():
    unpause() if paused else pause()

def unpause():
    #subprocess.Popen("FAHClient --send-unpause")
    launchWithoutConsole("FAHClient --send-unpause")
    global paused
    # print("Unpausing!!! - {}".format(paused))
    paused = False
    global btn_txt
    btn_txt.set("正在运行，点击暂停")
    b_logger.debug("Unpaused")

def pause():
    #subprocess.Popen("FAHClient --send-pause")
    launchWithoutConsole("FAHClient --send-pause")
    global paused
    # print("pausing!!! - {}".format(paused))
    paused = True
    global btn_txt
    btn_txt.set("已暂停，点击继续")
    b_logger.debug("Paused")

def launchWithoutConsole(command):
    startupinfo = subprocess.STARTUPINFO()
    startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
    return subprocess.Popen(command, startupinfo=startupinfo)

def launchPIPEproc(command):
    startupinfo = subprocess.STARTUPINFO()
    startupinfo.dwFlags |= subprocess.STARTF_USESHOWWINDOW
    global fah_client_ref
    fah_client_ref = subprocess.Popen(command, startupinfo=startupinfo, stdout=subprocess.PIPE)
    #fah_client_ref = subprocess.Popen(command, startupinfo=startupinfo)
    global thread_should_end
    thread_should_end = False
    send_thread(fah_client_ref, command).start()

class send_thread(threading.Thread):
    def __init__(self, fah_client_ref, command):
        threading.Thread.__init__(self)
        self.fah_client_ref = fah_client_ref
        self.command = command
    
    def run(self):
        for stdout_line in iter(self.fah_client_ref.stdout.readline, ""):
            c_logger.debug(stdout_line.decode())
            global thread_should_end
            if thread_should_end:
                break
        fah_client_ref.stdout.close()

if __name__ == "__main__":
    main_account_screen()
    if login_ok:
        fas_screen()
        GUI_log_here = False
        print("退出")
        thread_should_end = True
        b_logger.info("Finish")
        fah_client_ref.terminate()