비밀번호 관리 매니저 만들기 using Python

 

비밀번호 관리 매니저 만들기 using Python

 

Introduction

우리는 여러 웹사이트에 회원 가입 했습니다. 여러 아이디, 여러 패스워드 .. 보안에 취약합니다. 

주기적인 비밀번호 변경 관리와 암호화가 필요합니다.

 

필요 라이브러리

pip install cryptography

Full Code

from cryptography.fernet import Fernet
 
 
class PasswordManager:
 
    def __init__(self):
        self.key = None
        self.password_file = None
        self.password_dict = {}
 
    
    def create_key(self, path):
        self.key = Fernet.generate_key()
        with open(path, 'wb') as f:
            f.write(self.key)
 
    
    def load_key(self, path):
        with open(path, 'rb') as f:
            self.key = f.read()
 
    
    def create_password_file(self, path, initial_values=None):
        self.password_file = path    
 
        if initial_values is not None:
            for key, value in initial_values.items():
                self.add_password(key, value)
 
 
    def load_password_file(self, path):
        self.password_file = path
        
        with open(path, 'r') as f:
            for line in f:
                site, encrypted = line.split(":")
 
 
    def add_password(self, site, password):
        self.password_dict[site] = password
        
        if self.password_file is not None:
            with open(self.password_file, 'a+') as f:
                encrypted = Fernet(self.key).encrypt(password.encode())
                f.write(site + ":" + encrypted.decode() + "\n")
 
    
    def get_password(self, site):
        return self.password_dict[site]
 
 
 
def main():
    password = {
 
        "email": "1234567",
        "facebook": "myfbpassword",
        "YouTube": "helloworld123",
        "something": "myfavoritepassword_123"
 
    }
 
 
    pm = PasswordManager()
    print("""What do you want to do?
    (1) Create a new key
    (2) Load an existing key
    (3) Create new password file
    (4) Load existing password file
    (5) Add a new password
    (6) Get a password
    (q) Quit
    """)
 
    done = False
        
    while not done: 
        choice = input("Enter your choice: ")
        if choice == "1":
            path = input("Enter path: ")
            pm.create_key(path)
 
        elif choice == "2":
            path = input("Enter path: ")            
            pm.load_key(path) 
 
        elif choice == "3":
            path = input("Enter path: ")
            pm.create_password_file(path, password)
        elif choice == "4":
            path = input("Enter path: ")
            pm.load_password_file(path)
        elif choice == "5":
            path = input("Enter the site: ")
            password = input("Enter the password: ")
            pm.add_password(site, password)
        elif choice == "6":
            site = input("What site do you want: ")
            print(f"Password for {site} is {pm.get_password(site)}")
        elif choice == "q":
            done = True
            print("Bye")
        else:
            print("Invalid choice!")
 
 
if __name__ == "__main__":
    main()
 
 

 

실행

새로 생성한 key

새로 생성한 패스워드 파일

key와 패스워드 파일을 생성했습니다.