設定がひとつのjsonファイルで完結するように

dislocker.py

@@ -6,6 +6,42 @@ import hashlib
 import string
 import random
 
+class Dislocker():
+    def __init__(self):
+        self.config_dir_path = "./config/"
+        self.server_config_path = self.config_dir_path + "server.json"
+
+        if not os.path.isdir(self.config_dir_path):
+            print("config ディレクトリが見つかりません... 作成します。")
+            os.mkdir(self.config_dir_path)
+        
+        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"
+                }
+            }
+
+            with open(self.server_config_path, "w") as w:
+                json.dump(self.server_config, w, indent=4)
+
+        elif os.path.isfile(self.server_config_path):
+            with open(self.server_config_path, "r") as r:
+                self.server_config = json.load(r)
+
+        
+
+
 class Bot(discord.Client):
     def db_connect(self, host, db, port, user, password):
         self.db = psycopg2.connect(f"host={host} dbname={db} port={port} user={user} password={password}")
@@ -146,7 +182,7 @@ class Bot(discord.Client):
                                     await message.channel.send(f"使用が開始されました。\nパスワード | {register["password"]}\nPC番号 | {msg_split[1]}\nデバイス番号 | {msg_split[2]}")
                                 elif len(msg_split) == 4:
                                     await message.channel.send(f"使用が開始されました。\nパスワード | {register["password"]}\nPC番号 | {msg_split[1]}\nデバイス番号 | {msg_split[2]}\n使用目的 | {msg_split[3]}")
-                                await self.get_channel(bot_config["log_channel_id"]).send(f'{register["name"]} さんがPC {msg_split[1]} を使用しています')
+                                await self.get_channel(dislocker.server_config["bot"]["log_channel_id"]).send(f'{register["name"]} さんがPC {msg_split[1]} を使用しています')
                             elif register["result"] == "user_data_not_found":
                                 await message.channel.send("ユーザーとして登録されていないようです。管理者に問い合わせてください。")
                             elif register["result"] == "pc_already_in_use_by_you":
@@ -164,9 +200,9 @@ class Bot(discord.Client):
                     await message.channel.send("使用されていないようです...")
                 elif stop["result"] == "ok":
                     await message.channel.send(f"PC番号 {stop["pc_number"]} の使用が終了されました。")
-                    await self.get_channel(bot_config["log_channel_id"]).send(f'{stop["name"]} さんがPC {stop["pc_number"]} の使用を終了しました')
+                    await self.get_channel(dislocker.server_config["bot"]["log_channel_id"]).send(f'{stop["name"]} さんがPC {stop["pc_number"]} の使用を終了しました')
 
-        elif message.channel.id == bot_config["config_channel_id"]:
+        elif message.channel.id == dislocker.server_config["bot"]["config_channel_id"]:
             msg_split = message.content.split()
             if msg_split[0] == "/register":
                 if len(msg_split) <= 3:
@@ -184,45 +220,10 @@ class Bot(discord.Client):
                     await message.channel.send("なんでかわからんけど不正です。")
 
 
-config_dir_path = "./config/"
-db_config_path = config_dir_path + "db.json"
-if not os.path.isfile(db_config_path):
-    if not os.path.isdir(config_dir_path):
-        os.mkdir(config_dir_path)
-
-    db_config = {
-        "host": "localhost",
-        "db": "dislocker",
-        "username": "user",
-        "password": "example_pass",
-        "port": "5432"
-    }
-    with open(db_config_path, "w") as w:
-        json.dump(db_config, w, indent=4)
-elif os.path.isfile(db_config_path):
-    with open(db_config_path, "r") as r:
-        db_config = json.load(r)
-
-bot_config_path = config_dir_path + "bot.json"
-if not os.path.isfile(bot_config_path):
-    if not os.path.isdir(config_dir_path):
-        os.mkdir(config_dir_path)
-
-    bot_config = {
-        "token": "TYPE HERE BOTS TOKEN KEY",
-        "log_channel_id" : "TYPE HERE CHANNEL ID",
-        "config_channel_id": "TYPE HERE CHANNEL ID"
-    }
-    with open(bot_config_path, "w") as w:
-        json.dump(bot_config, w, indent=4)
-
-elif os.path.isfile(bot_config_path):
-    with open(bot_config_path, "r") as r:
-        bot_config = json.load(r)
-
 intents = discord.Intents.default()
 intents.message_content = True
 
+dislocker = Dislocker()
 bot = Bot(intents=intents)
-bot.db_connect(db_config["host"], db_config["db"], db_config["port"], db_config["username"], db_config["password"])
+bot.db_connect(dislocker.server_config["db"]["host"], dislocker.server_config["db"]["db_name"], dislocker.server_config["db"]["port"], dislocker.server_config["db"]["username"], dislocker.server_config["db"]["password"])
-bot.run(bot_config["token"])
+bot.run(dislocker.server_config["bot"]["token"])