diff --git a/dislocker.py b/dislocker.py index dcf692d..ab09a42 100644 --- a/dislocker.py +++ b/dislocker.py @@ -12,52 +12,45 @@ class DL(): self.config_dir_path = "./config/" self.export_dir_path = "./export/" self.server_config_path = self.config_dir_path + "server.json" - - if not os.path.isdir(self.config_dir_path): - print("config ディレクトリが見つかりません... 作成します。") - try: + try: + if not os.path.isdir(self.config_dir_path): + print("config ディレクトリが見つかりません... 作成します。") os.mkdir(self.config_dir_path) - except: - print("config ディレクトリを作成できませんでした。") - if not os.path.isfile(self.server_config_path): - print("config ファイルが見つかりません... 作成します。") - self.server_config = { - "db": { - "host": "localhost", - "port": "5432", - "db_name": "dislocker", - "username": "user", - "password": "password" - }, - "bot": { - "token": "TYPE HERE BOTS TOKEN KEY", - "log_channel_id" : "TYPE HERE CHANNEL ID", - "config_channel_id": "TYPE HERE CHANNEL ID" + if not os.path.isfile(self.server_config_path): + print("config ファイルが見つかりません... 作成します。") + self.server_config = { + "db": { + "host": "localhost", + "port": "5432", + "db_name": "dislocker", + "username": "user", + "password": "password" + }, + "bot": { + "token": "TYPE HERE BOTS TOKEN KEY", + "log_channel_id" : "TYPE HERE CHANNEL ID", + "config_channel_id": "TYPE HERE CHANNEL ID" + } } - } - try: with open(self.server_config_path, "w") as w: json.dump(self.server_config, w, indent=4) - except: - print("config ファイルを作成できませんでした。") - elif os.path.isfile(self.server_config_path): - with open(self.server_config_path, "r") as r: - self.server_config = json.load(r) - if not os.path.isdir(self.export_dir_path): - print("export ディレクトリが見つかりません... 作成します。") - try: + elif os.path.isfile(self.server_config_path): + with open(self.server_config_path, "r") as r: + self.server_config = json.load(r) + + if not os.path.isdir(self.export_dir_path): + print("export ディレクトリが見つかりません... 作成します。") os.mkdir(self.export_dir_path) - except: - print("export ディレクトリを作成できませんでした。") - try: + 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']}") result = {"result": "ok"} + except (Exception) as error: - print("データベースに接続できません。\nエラー内容\n" + error) + print("初回処理でエラーが発生しました。\nエラー内容\n" + error) result = {"result": "error"} finally: