Main.py

import os
import time

from watchdog.events import FileSystemEventHandler
from watchdog.observers import Observer

folderPath = os.getenv('APPDATA') + "\\HexChat\\"
serverConfigFile = open(folderPath + "servlist.conf", "r")
serverConfigFileText = serverConfigFile.read()
serverDomainPosition = serverConfigFileText.find("S=irc.ppy.sh")
logFileDestination = "a"
previousLogTextLen = 0
logTextToCheck = "a"
defaultTitle = "OFCT"
matchRefs = []

if serverDomainPosition == -1:
    serverDomainPosition = serverConfigFileText.find("S=cho.ppy.sh")

if serverDomainPosition == -1:
    print("HexChat에서 osu!irc 설정을 찾을 수 없습니다.")
    exit()

servername = serverConfigFileText[0:serverDomainPosition]
servername = servername[servername.rfind("N=") + 2:]
servername = servername[:servername.find("P=") - 1]

defaultTitle = input("타이머 기본 텍스트를 입력해 주세요 : ")
mpNumber = input("mp 번호를 입력해 주세요 : ")

logFileDestination = folderPath + "logs\\" + servername + "\\#mp_" + mpNumber + ".log"


# logFileDestination = folderPath + "logs\\" + servername + "\\test.log"
# logFileDestination = folderPath+"logs\\"+servername+"\\#osu.log"

def titleWrite(text):
    titleFile = open("./Sources/timerTitle.txt", "w", encoding="UTF8")
    titleFile.write(text)
    titleFile.close()


def timeWrite(text):
    timeFile = open("./Sources/time.txt", "w", encoding="UTF8")
    timeFile.write(text)
    timeFile.close()


def redWrite(text):
    redFile = open("./Sources/redTitle.txt", "w", encoding="UTF8")
    redFile.write(text)
    redFile.close()


def blueWrite(text):
    blueFile = open("./Sources/blueTitle.txt", "w", encoding="UTF8")
    blueFile.write(text)
    blueFile.close()


def resetText():
    titleWrite(defaultTitle)
    blueWrite("")
    redWrite("")
    timeWrite("")


def roster():
    titleWrite("로스터")
    redWrite("")
    blueWrite("")
    timer(90)


def pick(team):
    titleWrite("픽")

    if team:
        redWrite("Red")
    if not team:
        blueWrite("Blue")
    timer(90)


def ban(team):
    titleWrite("밴")

    if team:
        redWrite("Red")
    if not team:
        blueWrite("Blue")
    timer(120)


try:
    logFile = open(logFileDestination, "r", encoding="UTF8")
    previousLogTextLen = len(logFile.read())
    titleWrite(defaultTitle)
except FileNotFoundError:
    print("방 번호가 잘못되었거나, HexChat에서 방에 접속하지 않았습니다. 다시 확인해주세요.")
    time.sleep(5)
    exit()


def textCheck(loopCheck):
    global previousLogTextLen
    global logFile
    global logTextToCheck
    global matchRefs

    logFile = open(logFileDestination, "r", encoding="UTF8")
    logTextToCheck = logFile.read()[previousLogTextLen:]
    logFile.close()

    if not loopCheck:
        previousLogTextLen = previousLogTextLen + len(logTextToCheck)
        print(logTextToCheck)

    logTextToCheck = logTextToCheck.splitlines()

    for line in range(len(logTextToCheck)):
        if "\t!mp addref " in logTextToCheck[line] or "\t`addref " in logTextToCheck[line]:
            refereeNick = ""
            if "\t!mp addref " in logTextToCheck[line]:
                refereeNick = logTextToCheck[line][logTextToCheck[line].find("!mp addref ") + 11:]
            if "\t`addref " in logTextToCheck[line]:
                refereeNick = logTextToCheck[line][logTextToCheck[line].find("`addref ") + 8:]
            matchRefs.append(refereeNick)
            print(matchRefs)
            print("Added " + refereeNick + " to referees")

        for i in range(len(matchRefs)):
            if "<" + matchRefs[i] + ">\t`" in logTextToCheck[line] or "<" + matchRefs[i] + ">\t!mp" in logTextToCheck[
                line]:
                if "\t`" in logTextToCheck[line]:
                    command = logTextToCheck[line][logTextToCheck[line].find("\t`") + 2:]
                if "\t!mp " in logTextToCheck[line]:
                    command = logTextToCheck[line][logTextToCheck[line].find("\t!mp ") + 5:]

                if not loopCheck:
                    print(command)

                flag = True

                if "timer 90 r" in command:
                    if loopCheck:
                        return True
                    flag = False
                    resetText()
                    roster()

                if "timer 90 pr" in command:
                    if loopCheck:
                        return True
                    flag = False
                    resetText()
                    pick(True)

                if "timer 90 pb" in command:
                    if loopCheck:
                        return True
                    flag = False
                    resetText()
                    pick(False)

                if "timer 120 br" in command:
                    if loopCheck:
                        return True
                    flag = False
                    resetText()
                    ban(True)

                if "timer 120 bb" in command:
                    if loopCheck:
                        return True
                    flag = False
                    resetText()
                    ban(False)

                else:
                    if "timer " in command and flag == True:
                        try:
                            if loopCheck:
                                return True
                            customText = command[command.find("timer ") + 6:]
                            if not " " in customText:
                                customTime = int(customText)
                                resetText()
                                titleWrite("타이머")
                                timer(customTime)
                            if " " in customText:
                                customText = customText.split(" ", 1)
                                customTime = int(customText[0])
                                customText = customText[1]
                                resetText()
                                titleWrite(customText)
                                timer(customTime)
                        except ValueError:
                            print("입력값에 오류가 있습니다.")
                        except IndexError:
                            print("배열 인덱스 값 오류.")

                if "aborttimer" in command:
                    if loopCheck:
                        return True
                    resetText()

                if "text " in command:
                    try:
                        if loopCheck:
                            return True
                        customTitle = command[command.find("text ") + 5:]
                        resetText()
                        titleWrite(customTitle)
                    except ValueError:
                        print("입력값에 오류가 있습니다.")

                if "dt " in command:
                    try:
                        if loopCheck:
                            previousLogTextLen = previousLogTextLen + len(logTextToCheck[line])
                        global defaultTitle
                        print(command)
                        defaultTitle = command[command.find("dt ") + 3:]
                    except ValueError:
                        print("입력값에 오류가 있습니다.")

                if "reset" in command:
                    if loopCheck:
                        return True
                    resetText()


def timer(setTime):
    for i in range(setTime + 1):
        timeWrite(str(setTime - i))
        if textCheck(True):
            print("timer canceled")
            textCheck(False)
            return
        time.sleep(1)

    titleWrite("타이머 종료")
    redWrite("")
    blueWrite("")

    for i in range(20):
        time.sleep(0.5)
        if textCheck(True):
            textCheck(False)
            return

    resetText()


class MyHandler(FileSystemEventHandler):
    def on_modified(self, event):
        textCheck(False)


if __name__ == "__main__":
    event_handler = MyHandler()
    observer = Observer()
    observer.schedule(event_handler, path=folderPath + "logs\\" + servername + "\\", recursive=False)
    observer.start()

    try:
        while True:
            time.sleep(1)
    except KeyboardInterrupt:
        observer.stop()
    observer.join()