suti7yk5032/Dislocker
1
0
Fork 0
Code Issues 13 Pull requests Projects 3 Releases 2 Packages Wiki Activity Actions

tryを修正

Browse source
This commit is contained in:
suti7yk5032 2024-07-07 15:52:46 +09:00
parent 73502817e0
commit bf418936d4
1 changed files with 27 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

61
dislocker.py
View file

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