起動時にワンタイムパスワードを削除する処理を追加
This commit is contained in:
parent
aa9379b2a6
commit
224492568a
1 changed files with 6 additions and 2 deletions
|
@ -67,7 +67,6 @@ class DL():
|
|||
self.server_config = json.load(r)
|
||||
print("config ファイルを読み込みました。")
|
||||
|
||||
|
||||
if not os.path.isdir(self.export_dir_path):
|
||||
print("export ディレクトリが見つかりません... 作成します。")
|
||||
os.mkdir(self.export_dir_path)
|
||||
|
@ -79,6 +78,11 @@ class DL():
|
|||
if type(self.server_config["bot"]["log_channel_id"]) is not int or type(self.server_config["bot"]["config_channel_id"]) is not int:
|
||||
print("config ファイル内でチャンネルIDがint型で記述されていません。int型で記述して、起動してください。")
|
||||
self.init_result = "not_int"
|
||||
|
||||
if os.path.isfile(self.onetime_config_path):
|
||||
print("ワンタイムパスワードが見つかりました。削除します。")
|
||||
os.remove(self.onetime_config_path)
|
||||
|
||||
else:
|
||||
self.db = psycopg2.connect(f"host={self.server_config['db']['host']} dbname={self.server_config['db']['db_name']} port={self.server_config['db']['port']} user={self.server_config['db']['username']} password={self.server_config['db']['password']}")
|
||||
cursor = self.db.cursor()
|
||||
|
@ -745,7 +749,7 @@ class DL():
|
|||
return {"result": 0, "about": "ok", "output_dict": {"onetime_password": onetime_password, "current_count": current_count, "max_count": max_count}}
|
||||
|
||||
except Exception as error:
|
||||
self.log(title=f"[ERROR] PC登録用のワンタイムパスワード発行中にエラーが発生しました。 {str(error.__class__.__name__)}", message=str(error.args), flag=1)
|
||||
self.log(title=f"[ERROR] デバイス登録用のワンタイムパスワード発行中にエラーが発生しました。 {str(error.__class__.__name__)}", message=str(error.args), flag=1)
|
||||
return {"result": 1, "about": "error"}
|
||||
|
||||
|
||||
|
|
Loading…
Reference in a new issue