2024-06-04 09:36:29 +09:00
|
|
|
import json
|
|
|
|
import discord
|
|
|
|
import os
|
|
|
|
import psycopg2
|
2024-06-04 22:53:31 +09:00
|
|
|
import datetime
|
2024-06-04 09:36:29 +09:00
|
|
|
|
2024-06-04 22:53:31 +09:00
|
|
|
class Bot(discord.Client):
|
2024-06-05 11:27:57 +09:00
|
|
|
def db_connect(self, host, db, port, user, password):
|
|
|
|
self.db = psycopg2.connect(f"host={host} dbname={db} port={port} user={user} password={password}")
|
2024-06-04 22:53:31 +09:00
|
|
|
|
|
|
|
async def on_ready(self):
|
|
|
|
print("ログイン成功")
|
2024-06-04 09:36:29 +09:00
|
|
|
|
2024-06-04 22:53:31 +09:00
|
|
|
async def on_message(self, message):
|
|
|
|
if message.author.bot:
|
|
|
|
return
|
|
|
|
|
|
|
|
if isinstance(message.channel, discord.DMChannel):
|
|
|
|
user_id = message.author.id
|
|
|
|
user_name = message.author.name
|
|
|
|
msg_split = message.content.split()
|
|
|
|
if msg_split[0] == "/password":
|
|
|
|
if len(msg_split) <= 2:
|
|
|
|
await message.channel.send("PC番号、もしくはデバイス番号が入力されていません。")
|
|
|
|
|
|
|
|
elif len(msg_split) == 3:
|
|
|
|
if msg_split[1].isdigit() and msg_split[2].isdigit():
|
|
|
|
if int(msg_split[1]) <= 10 and int(msg_split[1]) >= 1:
|
|
|
|
cursor = self.db.cursor()
|
|
|
|
cursor.execute("SELECT * FROM club_member WHERE discord_userid = %s", (str(message.author.id),))
|
|
|
|
member_info = cursor.fetchall()
|
2024-06-05 16:13:57 +09:00
|
|
|
cursor.execute("SELECT * FROM pc_usage_history WHERE member_id=%s ORDER BY id DESC LIMIT 1", (member_info[0][0],))
|
|
|
|
history_info = cursor.fetchall()
|
|
|
|
if not history_info[0][5]:
|
|
|
|
await message.channel.send(f"もう使用されているようです。解除するには /stop で使用終了をお知らせください。\n使用中のPC番号 | {str(history_info[0][2])}\n使用中のデバイス番号 | {str(history_info[0][3])}\n使用開始時刻 | {str(history_info[0][4])}\n使用目的 | {str(history_info[0][6])}")
|
|
|
|
else:
|
|
|
|
cursor.execute("INSERT INTO pc_usage_history ( member_id, pc_number, device_number, start_use_time ) VALUES ( %s, %s, %s, current_timestamp)", (member_info[0][0], int(msg_split[1]), int(msg_split[2])))
|
|
|
|
self.db.commit()
|
|
|
|
await message.channel.send(f"使用が開始されました。\nPC番号 | {msg_split[1]}\nデバイス番号 | {msg_split[2]}\n")
|
|
|
|
await self.get_channel(bot_config["log_channel_id"]).send(f'{member_info[0][2]}がPC{msg_split[1]}を使用しています')
|
2024-06-04 22:53:31 +09:00
|
|
|
else:
|
|
|
|
await message.channel.send("パソコンの台数が\n# だめです")
|
|
|
|
else:
|
|
|
|
await message.channel.send("構文が不正です。")
|
|
|
|
|
|
|
|
elif len(msg_split) >= 4:
|
|
|
|
if msg_split[1].isdigit() and msg_split[2].isdigit():
|
|
|
|
if int(msg_split[1]) <= 10 and int(msg_split[1]) >= 1:
|
|
|
|
cursor = self.db.cursor()
|
|
|
|
cursor.execute("SELECT * FROM club_member WHERE discord_userid = %s", (str(message.author.id),))
|
|
|
|
member_info = cursor.fetchall()
|
2024-06-05 16:13:57 +09:00
|
|
|
cursor.execute("SELECT * FROM pc_usage_history WHERE member_id=%s ORDER BY id DESC LIMIT 1", (member_info[0][0],))
|
|
|
|
history_info = cursor.fetchall()
|
|
|
|
if not history_info[0][5]:
|
|
|
|
await message.channel.send(f"もう使用されているようです。解除するには /stop で使用終了をお知らせください。\n使用中のPC番号 | {str(history_info[0][2])}\n使用中のデバイス番号 | {str(history_info[0][3])}\n使用開始時刻 | {str(history_info[0][4])}\n使用目的 | {str(history_info[0][6])}")
|
|
|
|
else:
|
|
|
|
cursor.execute("INSERT INTO pc_usage_history ( member_id, pc_number, device_number, start_use_time, use_detail ) VALUES ( %s, %s, %s, current_timestamp, %s)", (member_info[0][0], int(msg_split[1]), int(msg_split[2]), str(msg_split[3])))
|
|
|
|
self.db.commit()
|
|
|
|
await message.channel.send(f"使用が開始されました。\nPC番号 | {msg_split[1]}\nデバイス番号 | {msg_split[2]}\n")
|
|
|
|
await self.get_channel(bot_config["log_channel_id"]).send(f'{member_info[0][2]}がPC{msg_split[1]}を使用しています')
|
2024-06-04 22:53:31 +09:00
|
|
|
else:
|
|
|
|
await message.channel.send("パソコンの台数が\n# だめです")
|
|
|
|
else:
|
|
|
|
await message.channel.send("構文が不正です。")
|
|
|
|
|
|
|
|
|
|
|
|
elif msg_split[0] == "/stop":
|
|
|
|
cursor = self.db.cursor()
|
|
|
|
cursor.execute("SELECT * FROM club_member WHERE discord_userid = %s", (str(message.author.id),))
|
|
|
|
member_info = cursor.fetchall()
|
|
|
|
cursor.execute("SELECT * FROM pc_usage_history WHERE member_id=%s ORDER BY id DESC LIMIT 1", (member_info[0][0],))
|
|
|
|
history_info = cursor.fetchall()
|
2024-06-05 16:13:57 +09:00
|
|
|
if history_info[0][5]:
|
|
|
|
await message.channel.send("使用されていないようです...")
|
|
|
|
else:
|
|
|
|
cursor.execute("UPDATE pc_usage_history SET end_use_time = current_timestamp WHERE id = %s", (history_info[0][0],))
|
|
|
|
self.db.commit()
|
|
|
|
await message.channel.send(f"使用が終了されました。")
|
|
|
|
await self.get_channel(bot_config["log_channel_id"]).send(f'{member_info[0][2]}がPC{history_info[0][2]}の使用を終了しました')
|
2024-06-04 22:53:31 +09:00
|
|
|
|
|
|
|
else:
|
|
|
|
|
|
|
|
if message.content == "/hello":
|
|
|
|
await message.channel.send("こんにちは!!!!")
|
2024-06-05 16:13:57 +09:00
|
|
|
|
2024-06-04 22:53:31 +09:00
|
|
|
else:
|
|
|
|
await message.channel.send("どうしましたか?")
|
|
|
|
|
2024-06-04 19:53:24 +09:00
|
|
|
#cert = Cert()
|
2024-06-04 09:36:29 +09:00
|
|
|
|
2024-06-04 22:53:31 +09:00
|
|
|
config_dir_path = "./config/"
|
2024-06-04 19:53:24 +09:00
|
|
|
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)
|
2024-06-04 09:36:29 +09:00
|
|
|
|
2024-06-04 19:53:24 +09:00
|
|
|
db_config = {
|
|
|
|
"host": "localhost",
|
|
|
|
"db": "dislocker",
|
|
|
|
"username": "user",
|
2024-06-04 22:53:31 +09:00
|
|
|
"password": "example_pass",
|
|
|
|
"port": "5432"
|
2024-06-04 19:53:24 +09:00
|
|
|
}
|
|
|
|
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)
|
2024-06-04 09:36:29 +09:00
|
|
|
|
2024-06-04 19:53:24 +09:00
|
|
|
bot_config_path = config_dir_path + "bot.json"
|
2024-06-04 09:36:29 +09:00
|
|
|
if not os.path.isfile(bot_config_path):
|
2024-06-04 19:53:24 +09:00
|
|
|
if not os.path.isdir(config_dir_path):
|
|
|
|
os.mkdir(config_dir_path)
|
|
|
|
|
2024-06-04 09:36:29 +09:00
|
|
|
bot_config = {
|
2024-06-04 19:53:24 +09:00
|
|
|
"token": "TYPE HERE BOTS TOKEN KEY"
|
2024-06-05 16:13:57 +09:00
|
|
|
"log_channel_id" "TYPE HERE CHANNEL ID"
|
2024-06-04 09:36:29 +09:00
|
|
|
}
|
|
|
|
with open(bot_config_path, "w") as w:
|
|
|
|
json.dump(bot_config, w, indent=4)
|
|
|
|
|
|
|
|
elif os.path.isfile(bot_config_path):
|
2024-06-04 19:53:24 +09:00
|
|
|
with open(bot_config_path, "r") as r:
|
2024-06-04 09:36:29 +09:00
|
|
|
bot_config = json.load(r)
|
|
|
|
|
|
|
|
intents = discord.Intents.default()
|
|
|
|
intents.message_content = True
|
|
|
|
|
|
|
|
bot = Bot(intents=intents)
|
2024-06-05 11:27:57 +09:00
|
|
|
bot.db_connect(db_config["host"], db_config["db"], db_config["port"], db_config["username"], db_config["password"])
|
2024-06-04 09:36:29 +09:00
|
|
|
bot.run(bot_config["token"])
|