From bf61a59b277b96fb38d7c9c7e9329d52d5ca1e52 Mon Sep 17 00:00:00 2001 From: Roeniss Moon Date: Sat, 10 Feb 2024 01:33:43 +0900 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20=ED=95=A8=EC=88=98=20=EC=9D=B4?= =?UTF-8?q?=EB=A6=84=20=EB=B0=8F=20=EC=9C=84=EC=B9=98=20=EC=A1=B0=EC=A0=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/dhapi/router/credentials_provider.py | 30 ++++++++++++------------ 1 file changed, 15 insertions(+), 15 deletions(-) diff --git a/src/dhapi/router/credentials_provider.py b/src/dhapi/router/credentials_provider.py index d68845c..59a843d 100644 --- a/src/dhapi/router/credentials_provider.py +++ b/src/dhapi/router/credentials_provider.py @@ -7,17 +7,9 @@ logger = logging.getLogger(__name__) -def read_credentials_file(profile_name): - with open(os.path.expanduser("~/.dhapi/credentials"), "r", encoding="UTF-8") as f: - file = f.read() - config = tomli.loads(file) - credentials = config.get(profile_name) - return credentials - - def get_credentials(profile_name): try: - credentials = read_credentials_file(profile_name) + credentials = _read_credentials_file(profile_name) except FileNotFoundError: print("❌ ~/.dhapi/credentials 파일을 찾을 수 없습니다. 파일을 생성하고 프로필을 추가하시겠습니까? [Y/n] ", end="") answer = input().strip().lower() @@ -25,29 +17,37 @@ def get_credentials(profile_name): print("📝 입력된 프로필 이름을 사용하시겠습니까? [Y/n]", end="") answer = input().strip().lower() if answer in ["y", "yes", ""]: - add_credentials(profile_name) + _add_credentials(profile_name) else: print("📝 프로필 이름을 입력하세요: ", end="") profile_name = input().strip() - add_credentials(profile_name) + _add_credentials(profile_name) else: raise FileNotFoundError("~/.dhapi/credentials 파일을 찾을 수 없습니다.") - credentials = read_credentials_file(profile_name) + credentials = _read_credentials_file(profile_name) if credentials is None: print(f"❌'{profile_name}' 프로필을 찾지 못했습니다. 추가하시겠습니까? [Y/n] ", end="") answer = input().strip().lower() if answer in ["y", "yes", ""]: - add_credentials(profile_name) - credentials = read_credentials_file(profile_name) + _add_credentials(profile_name) + credentials = _read_credentials_file(profile_name) return credentials raise ValueError(f"~/.dhapi/credentials 파일에서 '{profile_name}' 프로필을 찾지 못했습니다.") return credentials -def add_credentials(profile_name): +def _read_credentials_file(profile_name): + with open(os.path.expanduser("~/.dhapi/credentials"), "r", encoding="UTF-8") as f: + file = f.read() + config = tomli.loads(file) + credentials = config.get(profile_name) + return credentials + + +def _add_credentials(profile_name): print("📝 사용자 ID를 입력하세요: ", end="") user_id = input().strip() print("📝 사용자 비밀번호를 입력하세요: ", end="")