exec.py

# Finding Text at file and directory
from rich.console import Console
import os.path
import sys
import os
#import time
# Time look
#start = time.time()
# Global variable
console = Console()
exts = [] # 찾은 결과 값을 저장해요
count = 0 # 반복횟수를 저장해요

# Functions
def Isfile(file):
    if os.path.isfile(file):
        return 1
    else:
        return -1
def Isdir(path):
    if os.path.isdir(path):
        return 1
    else:
        return -1
def Table(text):
    n = len(text)
    table = [0] * n
    m = 0
    for i in range(1, n):
        while m > 0 and text[i] != text[m]:
            m = table[m-1]
        if text[i] == text[m]:
            m += 1
            table[i] = m
    return table
def Find(target, text):
    c = 0 # count
    m = 0
    if text == '':
        text = ' ' # Non의 값은 _(스페이스바 1개)로 값을 바꾸어요
    table = Table(text)
    for i in range(len(target)):
        while m > 0 and target[i] != text[m]:
            m = table[m-1]
        if target[i] == text[m]:
            m += 1
            if m == len(text):
                c = 1 # conut = 1
                m = table[m-1]
    return c
def FindFile(path, text):
    if Isfile(path) == -1:
        return 0
    file = open(path, "rb")
    load = file.read() # Bytes load
    file.close()
    read = load.decode(encoding="utf-8") # Bytes -> Str
    if read.find(' ')!=0:
        read += ' '
    if Find(read, text) != 0:
        return 1
    else:
        return -1
def FindDir(path, text):
    global count, exts
    try:
        if Isdir(path) == -1:
            return 0 # No directory
        dir_list = os.listdir(path)
        # 읽지 못하는 디렉토리들
        if '.git' in dir_list:
            dir_list.remove('.git')
        if '.github' in dir_list:
            dir_list.remove('.github')
        if '.DS_Store' in dir_list:
            dir_list.remove('.DS_Store')
        if '.localized' in dir_list:
            dir_list.remove('.localized')
        if '.vscode' in dir_list:
            dir_list.remove('.vscode')
        if '.gitignore' in dir_list:
            dir_list.remove('.gitignore')
        if 'node_modules' in dir_list:
            dir_list.remove('node_modules')
        if '.env' in dir_list:
            dir_list.remove('.env')
        if 'build' in dir_list:
            dir_list.remove('build')
        readable_dir_list = []
        readable_file_exts = ['.py', '.md', '.txt' ,'LICENSE', '.txt', '.rb', '.1', '.html'] # 이 읽어도 되는 파일들의 리스트 변수에 읽어도 되는 파일의 확장자나 이름을 입력하면 되요!
        for i in dir_list:
            if Isfile(path + '/' + i) == 1: # Isfile로 파일인지 아닌지를 확인해요
                if Find(i, '.swp') != 0:
                    continue # .swp 파일이 나오면 밑부분의 코드를 실행하지 않고 위의 코드로 올라가요
                for j in readable_file_exts:
                    if Find(i, j) != 0:
                        readable_dir_list.append(i) # dir_list에서 읽을 수 있는 파일들을 readable_dir_list에 리스트에 저장해요
            else: # dir
                readable_dir_list.append(i)
        dir_list = readable_dir_list # dir_list를 읽을 수 있는 파일들의 리스트로 초기화 해주어요
        leng = len(dir_list) # value
        for i in range(1, leng+1):
            name = "".join(dir_list[i-1])
            file = path + '/' + dir_list[i-1]
            if Isfile(file) == 1: # Isfile로 파일인지 아닌지를 확인해요
                find = FindFile(file, text)
                if find != -1:
                    count = 0 + 1
                    ext = "%s\n"%file # \n을 추가해서 "".join(exts)해도 줄 바꿈이 괜찮은거에요
                    exts.append(ext)
            else: # dir
                dir = FindDir(file, text)
        if count == 0:
            return -1 # 파일에 찾는 문자열이 없다면
        return "".join(exts) # No return of value (1: extant)
    except UnicodeDecodeError: # 만약 못읽는 파일이 나오면 못읽는 파일은 읽지 않아요 # Python의 모듈 Pickle로 만든 파일은 고유의 설정때문에 ft는 읽지를 못하고, 이 오류가 발생하게 되요
        pass
    except PermissionError: # 파일의 권한이 맞지 않을 경우, 읽지 않아요
        return sys.exit(1)
def TextInput():
    return sys.argv[1]
def PathInput():
    return sys.argv[2]
def NowDir():
    return os.path.dirname(os.path.abspath(__file__))
def TextLower(text):
    return text.lower()
def TextUpper(text):
    return text.upper()
def ManHelp():
    ManVer()
    now = NowDir()
    help_file = now + "/doc/help.txt"
    isfile = Isfile(help_file)
    if isfile == -1:
        print("Error: No help.txt file")
        return -1
    file = open(help_file, "rb")
    load = file.read()
    file.close()
    print(load.decode(encoding="utf-8")) # Prints
def ManVer():
    now = NowDir()
    ver_file = now + "/doc/version.txt"
    isfile = Isfile(ver_file)
    if isfile == -1:
        print("Error: No version.txt file")
        return -1
    file = open(ver_file, "rb")
    load = file.read()
    file.close()
    print(load.decode(encoding="utf-8")[:2] + ' ' + load.decode(encoding="utf-8")[3:]) # Prints
    
def ManIndexErrorHelp(flag):
    print("error: No '%s' flags"%flag)
    now = NowDir()
    err_file = now + "/doc/error_help.txt"
    isfile = Isfile(err_file)
    if isfile == -1:
        print("Error: No error_help.txt file")
        return -1
    file = open(err_file, "rb")
    load = file.read()
    file.close()
    print(load.decode(encoding="utf-8")) # Print
def Flags():
    global text
    cp_text = text # 복사본으로 이용해요
    for i in range(1, 3):
        if i == 1:
            cp_text = TextLower(cp_text)
        else:
            cp_text = TextUpper(cp_text)
        if (cp_text.find('-h')==0 or cp_text.find('--h')==0):
            ManHelp()
            return sys.exit(0)
        if (cp_text.find('-v')==0 or cp_text.find('--v')==0):
            ManVer()
            return sys.exit(0)
def Options():
    global path
    if (path.find("!")==0 or path.find("*")==0):
        path = os.getcwd()
def Print(text, path):
    #global dir
    file_list = []
    try:
        for i in range(1, 3):
            if i == 1:
                text = TextLower(text)
            else:
                text = TextUpper(text)
            dir = FindDir(path, text)
            if dir == -1: # -1 of dir
                return print(-1)
            elif dir == 0: # 0 of dir or file: 1 or -1
                file = FindFile(path, text)
                if file == 0:
                    return print(0)
                else: # file of 1 or -1
                    file_list.append(file)
                for i in range(1, 3):
                    if file_list[i-1] == 1:
                        print(1)
                        sys.exit(0)
                print(-1)
            else: # 1
                if i == 2:
                    dir = dir.split() # Str -> List
                    set_ = set(dir) # 순서가 바뀌어요
                    dir = list(set_)
                    dir.sort()
                    leng = len(dir)
                    for i in range(1, leng+1):
                        if Find("".join(dir[i-1]).replace(path+'/', ""), '/')!=0:
                            list_leng = len("".join(dir[i-1]).replace(path+'/', "").split('/'))
                            for j in range(0, list_leng):
                                if j <= list_leng-2:
                                    console.print("".join(dir[i-1]).replace(path+'/', "").split('/')[j], style="blue", end="/")
                            console.print("".join(dir[i-1]).replace(path+'/', "").split('/')[j], style="green")
                        else:
                            console.print("".join(dir[i-1]).replace(path+'/', ""), style="yellow")
                    print(i)
    except IndexError:
        pass
    except AttributeError:
        pass

try:
    text, path = ' ', ' ' # NameError 방지해요
    text = TextInput()
    # Flags
    Flags()
    path = PathInput()
    if len(sys.argv) > 3: # 다중 인자 입력 방지
        sys.exit(1)
    # Options
    Options()
    # Print Find
    Print(text, path)
    # Time Look
    #print("ExecTime: ", time.time() - start)
except IndexError: # sys.argv의 인자가 충분히 제공 되지 않았을때
    if (text.find("-")==0):
        ManIndexErrorHelp(text)
    else:
        path = os.getcwd()
        Print(text, path)
        #Time Look
        #print("ExecTime: ", time.time() - start)

# (0): rich text를 읽지 않게끔 설정하기
# (0): pip로 설치가능하도록 설정하기
# (0): man기능 추가하기
# (0): doc -v이렇게 하는거는 별로임