python pandas를 이용한 csv 데이터 검색 프로그램

csv파일을 이용한 검색 프로그램

검색에서 불필요한 항목은 제거하고 보고싶은 내용만 정렬 후 검색

def search_phone_number(csv_path, search_value):
    df = pd.read_csv(csv_path, encoding='ANSI')
    columns_to_exclude = ['v_port', 'MAC addr', 'mode', 'attached UC phone', 'Detail Type']
    search_result = df[df['Phone Number'].astype(str).str.contains(str(search_value))].drop(columns=columns_to_exclude)
    search_result = search_result.sort_values(by='Phone Number', ascending=True)
    return search_result


def search_ip(csv_path, search_value):
    df = pd.read_csv(csv_path, encoding='ANSI')
    columns_to_exclude = ['v_port', 'MAC addr', 'mode', 'attached UC phone', 'Detail Type']
    search_result = df[df['IP address'].astype(str).str.startswith(str(search_value))].drop(columns=columns_to_exclude)
    # IP 주소를 숫자로 변환하여 정렬
    search_result['IP address'] = search_result['IP address'].apply(lambda x: sum([256**i*int(j) for i, j in enumerate(x.split('.')[::-1])]))
    search_result = search_result.sort_values(by='IP address', ascending=True)
    # 다시 IP 주소 형식으로 변환
    search_result['IP address'] = search_result['IP address'].apply(lambda x: '.'.join([str((x >> i) % 256) for i in (24, 16, 8, 0)]))
    return search_result


def search_directory(csv_path, search_value):
    df = pd.read_csv(csv_path, encoding='ANSI')
    columns_to_exclude = ['v_port', 'MAC addr', 'mode', 'attached UC phone', 'Detail Type']
    search_result = df[df['Directory'].astype(str).str.contains(str(search_value))]
    search_result = search_result.drop(columns=columns_to_exclude)
    search_result = search_result.sort_values(by='Directory', ascending=True)
    return search_result


csv_path = "c:/domain/ipt/user_profile.csv"
df = pd.read_csv(csv_path, encoding='ANSI')


while True:
    
# 프로그램에 대한 정보
    os.system('cls' if os.name == 'nt' else 'clear')
    print('''
=========================================================

전화번호 / IP / 사용자 검색 프로그램 입니다.

1. 전화번호 검색

2. IP검색

3. 디렉토리 (사용자) 검색

9. 종료

=========================================================
''')
    
    option = input("\n검색 옵션을 선택하세요. : ")
    
    if option == "1":
        os.system('cls' if os.name == 'nt' else 'clear')
        search_value = input("\n전화번호 : ")
        result = search_phone_number(csv_path, search_value)
        if len(result) > 0:
            print("검색 결과:")
            print(result.to_string(index=False))  # 모든 행 데이터를 출력하기 위해 index를 제외하고 출력합니다.
        else:
            print("\n검색 결과가 없습니다.\n")
        
        input("\n계속하려면 엔터 키를 눌러주세요.\n")


    elif option == "2":
        os.system('cls' if os.name == 'nt' else 'clear')
        search_value = input("\nIP Address : ")
        result = search_ip(csv_path, search_value)
        if len(result) > 0:
            print("검색 결과:")
            print(result.to_string(index=False))  # 모든 행 데이터를 출력하기 위해 index를 제외하고 출력합니다.
        else:
            print("\n검색 결과가 없습니다.\n")
        
        input("\n계속하려면 엔터 키를 눌러주세요.\n")


    elif option == "3":
        os.system('cls' if os.name == 'nt' else 'clear')
        search_value = input("\n디렉토리 : ")
        result = search_directory(csv_path, search_value)
        if len(result) > 0:
            print("검색 결과:")
            print(result.to_string(index=False))  # 모든 행 데이터를 출력하기 위해 index를 제외하고 출력합니다.
        else:
            print("\n검색 결과가 없습니다.\n")
        
        input("\n계속하려면 엔터 키를 눌러주세요.\n")



    elif option == "9":
        print("\n프로그램을 종료합니다.\n")
        break
    
    else:
        print("\n올바른 옵션을 선택해주세요.\n")