suti7yk5032/Dislocker
1
0
Fork 0
Code Issues 13 Pull requests Projects 3 Releases 2 Packages Wiki Activity Actions

起動時にワンタイムパスワードを削除する処理を追加

Browse source
This commit is contained in:
suti7yk5032 2024-09-26 21:22:08 +09:00
parent aa9379b2a6
commit 224492568a
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
dislocker_slash.py
View file

@ -66,7 +66,6 @@ class DL():
with open(self.server_config_path, "r", encoding="utf-8") as r: with open(self.server_config_path, "r", encoding="utf-8") as r:
self.server_config = json.load(r) self.server_config = json.load(r)
print("config ファイルを読み込みました。") print("config ファイルを読み込みました。")
if not os.path.isdir(self.export_dir_path): if not os.path.isdir(self.export_dir_path):
print("export ディレクトリが見つかりません... 作成します。") print("export ディレクトリが見つかりません... 作成します。")
@ -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: 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型で記述して、起動してください。") print("config ファイル内でチャンネルIDがint型で記述されていません。int型で記述して、起動してください。")
self.init_result = "not_int" self.init_result = "not_int"
if os.path.isfile(self.onetime_config_path):
print("ワンタイムパスワードが見つかりました。削除します。")
os.remove(self.onetime_config_path)
else: 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']}") 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() 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}} return {"result": 0, "about": "ok", "output_dict": {"onetime_password": onetime_password, "current_count": current_count, "max_count": max_count}}
except Exception as error: 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"} return {"result": 1, "about": "error"}