データベース接続時の例外処理を追加
This commit is contained in:
parent
f7f7033ea5
commit
4ecf209e7e
1 changed files with 12 additions and 10 deletions
22
dislocker.py
22
dislocker.py
|
@ -53,9 +53,14 @@ class DL():
|
||||||
os.mkdir(self.export_dir_path)
|
os.mkdir(self.export_dir_path)
|
||||||
except:
|
except:
|
||||||
print("export ディレクトリを作成できませんでした。")
|
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):
|
class Bot(discord.Client):
|
||||||
|
@ -231,8 +236,6 @@ class Bot(discord.Client):
|
||||||
cursor.close()
|
cursor.close()
|
||||||
|
|
||||||
return result
|
return result
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
async def on_ready(self):
|
async def on_ready(self):
|
||||||
print("ログイン成功")
|
print("ログイン成功")
|
||||||
|
@ -308,10 +311,9 @@ class Bot(discord.Client):
|
||||||
elif export["result"] == "export_error":
|
elif export["result"] == "export_error":
|
||||||
await message.channel.send("エクスポートに失敗しました。")
|
await message.channel.send("エクスポートに失敗しました。")
|
||||||
|
|
||||||
|
|
||||||
intents = discord.Intents.default()
|
|
||||||
intents.message_content = True
|
|
||||||
|
|
||||||
dislocker = DL()
|
dislocker = DL()
|
||||||
bot = Bot(intents=intents)
|
if dislocker["result"] == "ok":
|
||||||
bot.run(dislocker.server_config['bot']['token'])
|
intents = discord.Intents.default()
|
||||||
|
intents.message_content = True
|
||||||
|
bot = Bot(intents=intents)
|
||||||
|
bot.run(dislocker.server_config['bot']['token'])
|
||||||
|
|
Loading…
Reference in a new issue