웹싸이트 패스워드 관리 프로그램 업그레이드

Days29에서 생성한 웹싸이트 패스워드 관리 프로그램의 저장파일을 JSON 형식으로 수정하고,

최초 파일이 없을 경우 등 에러에 대한 예외처리를 적용한 업그레이드 버전을 작성함.

 

이는 Search 기능을 포함한 최종버전의 -1 버전임.

 

from tkinter import *
from tkinter import messagebox
from random import choice, randint, shuffle
import pyperclip
import json  # json확장자 파일을 wirte(dump)하고, read(load)하고, update(update)하기 위한 모듈 import

# -----------------gen password----------------------#
def gen():
    password_entry.delete(0,20)  #생성 전에 화면의 패스워드 엔트리엣 삭제
    letters = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z']
    numbers = ['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
    symbols = ['!', '#', '$', '%', '&', '(', ')', '*', '+']

    # for char in range(nr_letters):
    #   password_list.append(random.choice(letters))
    #
    # for char in range(nr_symbols):
    #   password_list += random.choice(symbols)
    #
    # for char in range(nr_numbers):
    #   password_list += random.choice(numbers)

    # 반복문을 리스트 컨프리핸션으로 구현
    password_letters = [choice(letters) for _ in range(randint(6,8))]
    password_numbers = [choice(numbers) for _ in range(randint(1,2))]
    password_symbers = [choice(symbols) for _ in range(randint(1,2))]

    password_list = password_letters + password_numbers + password_symbers
    shuffle(password_list)

    # password = ""
    # for char in password_list:
    #   password += char

    # 반복문을 join 모듈로 구현
    password = "".join(password_list)

    # 콘솔 확인용
    print(f"{password_list}\n")
    print(f"Your password is: {password}")

    # 패스워드 생성과 동시에 클립보드에 복사
    pyperclip.copy(password)

    # 패스워드 위치에 생성한 패스워드 표현
    password_entry.insert(0,password)

# -----------------save password_label----------------------#
def save():
    website = website_entry.get()
    email = email_entry.get()
    password = password_entry.get()
    new_data = {
        website:{
            "email" : email,
            "password" : password,
        }
    }

    if len(website) == 0 or len(email)==0 or len(password)==0:
        messagebox.showinfo(title="Warning", message="Please don't leave any fields empty!")
    else:
        try:
            # data.json 파일을 read 해보는데,
            with open("data.json", "r") as data_file:  ## data_file=open("data.json","a")  #이렇게 열면 close()로 닫아야 함
                # Reading old data
                data = json.load(data_file)

        except FileNotFoundError:
            # data.json 파일이 없어서 FileNotFoundError가 나면 write 모드로 열면서 생성하고
            with open("data.json","w") as data_file:
                json.dump(new_data, data_file, indent=4)

        else:
            # Updating old data with new data
            data.update(new_data)

            with open("data.json","w") as data_file:
                # Saving updated data
                json.dump(data, data_file, indent=4)

        finally:
            print(data)
            website_entry.delete(0, END)
            email_entry.delete(0, END)
            password_entry.delete(0, END)

window = Tk()
window.title("Password Manager - powered by kiostory(1Sep2022)")
window.config(padx=20, pady=20)

canvas = Canvas(height=200, width=200)
logo_img = PhotoImage(file="logo.png")
canvas.create_image(100, 100, image=logo_img)
# logo_label=Label(text="MyPass")
canvas.grid(row=0, column=1)

# Labels
website_label = Label(text="Website:")
website_label.grid(row=1, column=0)

email_label = Label(text="Email/Username:")
email_label.grid(row=2, column=0)

password_label = Label(text="Password:")
password_label.grid(row=3, column=0)

# Entries
website_entry = Entry(width=35)
website_entry.grid(row=1, column=1, columnspan=2)
website_entry.focus()
email_entry = Entry(width=35)
email_entry.grid(row=2, column=1, columnspan=2)
email_entry.insert(0, "")
password_entry = Entry(width=21)
password_entry.grid(row=3, column=1)

# Buttons
generate_password_button = Button(text="Generate Password", command=gen)
generate_password_button.grid(row=3, column=2)
add_button = Button(text="Add", width=36, command=save)
add_button.grid(row=4, column=1, columnspan=2)

window.mainloop()