diff --git a/dislocker.py b/dislocker.py index 7fb720c..e3f77f6 100644 --- a/dislocker.py +++ b/dislocker.py @@ -53,9 +53,14 @@ class DL(): 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) + result = {"result": "error"} - 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']}") - + return result class Bot(discord.Client): @@ -231,8 +236,6 @@ class Bot(discord.Client): cursor.close() return result - - async def on_ready(self): print("ログイン成功") @@ -308,10 +311,9 @@ class Bot(discord.Client): elif export["result"] == "export_error": await message.channel.send("エクスポートに失敗しました。") - -intents = discord.Intents.default() -intents.message_content = True - dislocker = DL() -bot = Bot(intents=intents) -bot.run(dislocker.server_config['bot']['token']) +if dislocker["result"] == "ok": + intents = discord.Intents.default() + intents.message_content = True + bot = Bot(intents=intents) + bot.run(dislocker.server_config['bot']['token'])