From 4ecf209e7ea6e9416f95a976f1239821e9294acb Mon Sep 17 00:00:00 2001
From: suti7yk5032 <yuman1965969@gmail.com>
Date: Sun, 7 Jul 2024 15:43:43 +0900
Subject: [PATCH] =?UTF-8?q?=E3=83=87=E3=83=BC=E3=82=BF=E3=83=99=E3=83=BC?=
 =?UTF-8?q?=E3=82=B9=E6=8E=A5=E7=B6=9A=E6=99=82=E3=81=AE=E4=BE=8B=E5=A4=96?=
 =?UTF-8?q?=E5=87=A6=E7=90=86=E3=82=92=E8=BF=BD=E5=8A=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dislocker.py | 22 ++++++++++++----------
 1 file changed, 12 insertions(+), 10 deletions(-)

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'])