pull
This commit is contained in:
commit
20d9d15063
4 changed files with 116 additions and 28 deletions
5
.gitignore
vendored
5
.gitignore
vendored
|
@ -160,5 +160,10 @@ cython_debug/
|
|||
# option (not recommended) you can uncomment the following to ignore the entire idea folder.
|
||||
#.idea/
|
||||
|
||||
<<<<<<< HEAD
|
||||
config/*
|
||||
db/*
|
||||
=======
|
||||
config/
|
||||
db/
|
||||
>>>>>>> for_psql
|
||||
|
|
122
dislocker.py
122
dislocker.py
|
@ -1,24 +1,14 @@
|
|||
import json
|
||||
import discord
|
||||
import os
|
||||
<<<<<<< HEAD
|
||||
import sqlite3
|
||||
=======
|
||||
import psycopg2
|
||||
import datetime
|
||||
>>>>>>> for_psql
|
||||
|
||||
class Bot(discord.Client):
|
||||
async def on_ready(self):
|
||||
print("ログイン成功")
|
||||
|
||||
async def on_message(self, message):
|
||||
if message.author.bot:
|
||||
return
|
||||
|
||||
if message.content == "/hello":
|
||||
await message.channel.send("こんにちは!!!!")
|
||||
|
||||
if message.content == "/password":
|
||||
await message.channel.send("gjhioragiopwjhgiopweap")
|
||||
|
||||
|
||||
class Cert():
|
||||
class Database():
|
||||
def __init__(self):
|
||||
if not os.path.isfile("./db/dislocker.db"):
|
||||
initial = 1
|
||||
|
@ -55,23 +45,117 @@ class Cert():
|
|||
elif result[0][1] == username and result[0][2] == password:
|
||||
return 0
|
||||
|
||||
<<<<<<< HEAD
|
||||
|
||||
cert = Cert()
|
||||
=======
|
||||
class Bot(discord.Client):
|
||||
def db_connect(self, host, db, user, password):
|
||||
self.db = psycopg2.connect(f"host={host} dbname={db} user={user} password={password}")
|
||||
|
||||
bot_config_path = "./config/bot.json"
|
||||
async def on_ready(self):
|
||||
print("ログイン成功")
|
||||
|
||||
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()
|
||||
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")
|
||||
else:
|
||||
await message.channel.send("パソコンの台数が\n# だめです")
|
||||
else:
|
||||
await message.channel.send("構文が不正です。")
|
||||
>>>>>>> for_psql
|
||||
|
||||
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()
|
||||
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")
|
||||
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()
|
||||
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"使用が終了されました。")
|
||||
|
||||
else:
|
||||
|
||||
if message.content == "/hello":
|
||||
await message.channel.send("こんにちは!!!!")
|
||||
|
||||
else:
|
||||
await message.channel.send("どうしましたか?")
|
||||
|
||||
#cert = Cert()
|
||||
|
||||
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": ""
|
||||
"token": "TYPE HERE BOTS TOKEN KEY"
|
||||
}
|
||||
with open(bot_config_path, "w") as w:
|
||||
json.dump(bot_config, w, indent=4)
|
||||
|
||||
elif os.path.isfile(bot_config_path):
|
||||
with open("./config/bot.json", "r") as r:
|
||||
with open(bot_config_path, "r") as r:
|
||||
bot_config = json.load(r)
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.message_content = True
|
||||
|
||||
bot = Bot(intents=intents)
|
||||
bot.db_connect(db_config["host"], db_config["db"], db_config["username"], db_config["password"])
|
||||
bot.run(bot_config["token"])
|
|
@ -1,13 +1,8 @@
|
|||
# ログイン
|
||||
psql -U suti7 -d dislocker
|
||||
|
||||
CREATE TABLE user
|
||||
(user_id INTEGER PRIMARY KEY NOT NULL,
|
||||
discord_id INTEGER NOT NULL,
|
||||
discord_username VARCHAR(32) NOT NULL,
|
||||
PRIMARY KEY (user_id));
|
||||
CREATE TABLE club_member (id SERIAL NOT NULL, name VARCHAR(30) NOT NULL, discord_username VARCHAR(32) NOT NULL, discord_userid VARCHAR(18) NOT NULL, PRIMARY KEY (id));
|
||||
CREATE TABLE pc_usage_history (id SERIAL NOT NULL, member_id INTEGER NOT NULL, pc_number INTEGER NOT NULL, device_number INTEGER NOT NULL, start_use_time TIMESTAMP NOT NULL, end_use_time TIMESTAMP, use_detail VARCHAR(30), PRIMARY KEY (id), FOREIGN KEY (memb
|
||||
er_id) REFERENCES club_member(id));
|
||||
|
||||
CREATE TABLE user_list (user_id INTEGER NOT NULL, discord_username VARCHAR(32) NOT NULL, discord_userid INTEGER NOT NULL, PRIMARY KEY (user_id));
|
||||
|
||||
# テーブルを作る
|
||||
CREATE TABLE web_auth (id SERIAL, username VARCHAR(10) NOT NULL, password VARCHAR(32) NOT NULL, PRIMARY KEY(id));
|
||||
INSERTのとき文字列はシングルクォーテーションで囲む!!!
|
4
requirements.txt
Normal file
4
requirements.txt
Normal file
|
@ -0,0 +1,4 @@
|
|||
discord.py
|
||||
flask
|
||||
flask-login
|
||||
psycopg2
|
Loading…
Reference in a new issue