408 lines
14 KiB
Python
408 lines
14 KiB
Python
|
import json
|
||
|
|
import string
|
||
|
|
import time
|
||
|
|
import requests
|
||
|
|
import discord
|
||
|
|
from discord.ext import commands
|
||
|
|
|
||
|
|
|
||
|
|
class Setup(commands.Cog):
|
||
|
|
|
||
|
|
def __init__(self, client):
|
||
|
|
self.client = client
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def add_client(self, ctx, name, password, save_number_0_1_2):
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json", "r") as f:
|
||
|
|
accounts = json.load(f)
|
||
|
|
|
||
|
|
save_num = int(save_number_0_1_2)-1
|
||
|
|
if save_num == -1:
|
||
|
|
save_num = ""
|
||
|
|
|
||
|
|
server_accounts = accounts[str(ctx.guild.id)]
|
||
|
|
|
||
|
|
server_accounts[name] = {"name": name,
|
||
|
|
"password": weird_shuffle(password),
|
||
|
|
"file": f"edu{save_num}.json"}
|
||
|
|
|
||
|
|
accounts[str(ctx.guild.id)] = server_accounts
|
||
|
|
|
||
|
|
await ctx.channel.purge(limit=1)
|
||
|
|
|
||
|
|
await ctx.send(f"Client for account {name} added.")
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json", "w") as f:
|
||
|
|
json.dump(accounts, f, indent=4)
|
||
|
|
|
||
|
|
await self.h_m(ctx)
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def clear_clients(self, ctx):
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json", "r") as f:
|
||
|
|
accounts = json.load(f)
|
||
|
|
|
||
|
|
accounts[str(ctx.guild.id)] = {}
|
||
|
|
|
||
|
|
await ctx.send("Cleared clients.")
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json", "w") as f:
|
||
|
|
json.dump(accounts, f, indent=4)
|
||
|
|
|
||
|
|
# removing accounts
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu.json", "r") as f:
|
||
|
|
save = json.load(f)
|
||
|
|
|
||
|
|
save[str(ctx.guild.id)] = {}
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu.json", "w") as f:
|
||
|
|
json.dump(save, f, indent=4)
|
||
|
|
|
||
|
|
# removing accounts
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu0.json", "r") as f:
|
||
|
|
save = json.load(f)
|
||
|
|
|
||
|
|
save[str(ctx.guild.id)] = {}
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu0.json", "w") as f:
|
||
|
|
json.dump(save, f, indent=4)
|
||
|
|
|
||
|
|
# removing accounts
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu1.json", "r") as f:
|
||
|
|
save = json.load(f)
|
||
|
|
|
||
|
|
save[str(ctx.guild.id)] = {}
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu1.json", "w") as f:
|
||
|
|
json.dump(save, f, indent=4)
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def set_channel(self, ctx, channel_name):
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/channels.json", "r") as f:
|
||
|
|
channels = json.load(f)
|
||
|
|
|
||
|
|
channels[str(ctx.guild.id)] = channel_name
|
||
|
|
|
||
|
|
for channel in ctx.guild.channels:
|
||
|
|
if channel.name == channel_name:
|
||
|
|
await channel.send("This channel is now set as the channel where bot sends messages(notifications).\n"
|
||
|
|
f"<@{ctx.author.id}>")
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/channels.json", "w") as f:
|
||
|
|
json.dump(channels, f, indent=4)
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def debug(self, ctx, True_False):
|
||
|
|
if True_False in ["True", "true"]:
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/setup.json", "r") as f:
|
||
|
|
setupX = json.load(f)
|
||
|
|
|
||
|
|
setupX[str(ctx.guild.id)]['debug'] = True
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/setup.json", "w") as f:
|
||
|
|
json.dump(setupX, f, indent=4)
|
||
|
|
|
||
|
|
await ctx.send("Debug is now on.")
|
||
|
|
elif True_False in ["False", "false"]:
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/setup.json", "r") as f:
|
||
|
|
setupX = json.load(f)
|
||
|
|
|
||
|
|
setupX[str(ctx.guild.id)]['debug'] = False
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/setup.json", "w") as f:
|
||
|
|
json.dump(setupX, f, indent=4)
|
||
|
|
|
||
|
|
await ctx.send("Debug is now off.")
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def h_m(self, ctx):
|
||
|
|
update_database(ctx)
|
||
|
|
|
||
|
|
no_edu0 = False
|
||
|
|
no_edu1 = False
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu.json", "r") as f:
|
||
|
|
edu = json.load(f)
|
||
|
|
edu = edu[str(ctx.guild.id)]
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu0.json", "r") as f:
|
||
|
|
try:
|
||
|
|
edu0 = json.load(f)
|
||
|
|
edu0 = edu0[str(ctx.guild.id)]
|
||
|
|
if len(edu0.keys()) == 0:
|
||
|
|
raise Exception("nothing inside")
|
||
|
|
except:
|
||
|
|
no_edu0 = True
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/edu1.json", "r") as f:
|
||
|
|
try:
|
||
|
|
edu1 = json.load(f)
|
||
|
|
edu1 = edu1[str(ctx.guild.id)]
|
||
|
|
if len(edu1.keys()) == 0:
|
||
|
|
raise Exception("nothing inside")
|
||
|
|
except:
|
||
|
|
no_edu1 = True
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/old.json", "r") as f:
|
||
|
|
oldMain = json.load(f)
|
||
|
|
|
||
|
|
old = oldMain[str(ctx.guild.id)]
|
||
|
|
|
||
|
|
new_e_superids = []
|
||
|
|
new_timelineids = []
|
||
|
|
|
||
|
|
# handling homework
|
||
|
|
messages = edu['homeworks']
|
||
|
|
to_show = []
|
||
|
|
for message in messages:
|
||
|
|
try:
|
||
|
|
if message["e_superid"] not in old["homework"].keys():
|
||
|
|
to_show.append(message)
|
||
|
|
new_e_superids.append(message["e_superid"])
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
to_show0 = []
|
||
|
|
if no_edu0 is False:
|
||
|
|
messages = edu0['homeworks']
|
||
|
|
for message in messages:
|
||
|
|
try:
|
||
|
|
if message["e_superid"] not in old["homework"].keys():
|
||
|
|
if message["e_superid"] not in new_e_superids:
|
||
|
|
new_e_superids.append(message["e_superid"])
|
||
|
|
to_show0.append(message)
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
to_show1 = []
|
||
|
|
if no_edu1 is False:
|
||
|
|
messages = edu1['homeworks']
|
||
|
|
for message in messages:
|
||
|
|
try:
|
||
|
|
if message["e_superid"] not in old["homework"].keys():
|
||
|
|
if message["e_superid"] not in new_e_superids:
|
||
|
|
new_e_superids.append(message["e_superid"])
|
||
|
|
to_show1.append(message)
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
except:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# handling messages
|
||
|
|
# getting client names to remove private messages
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json", "r") as f:
|
||
|
|
accounts = json.load(f)
|
||
|
|
accounts = accounts[str(ctx.guild.id)]
|
||
|
|
|
||
|
|
names = [accounts[x]["name"] for x in accounts.keys()]
|
||
|
|
|
||
|
|
for name in range(len(names)):
|
||
|
|
i = 0
|
||
|
|
li = 0
|
||
|
|
for char in names[name]:
|
||
|
|
if char in string.ascii_uppercase:
|
||
|
|
i += 1
|
||
|
|
if i == 2:
|
||
|
|
names[name] = f"{names[name][:li]} {names[name][li:]}"
|
||
|
|
li += 1
|
||
|
|
|
||
|
|
# looking for messages
|
||
|
|
to_show = []
|
||
|
|
messages = edu["timelineItems"]
|
||
|
|
for message in messages:
|
||
|
|
if message["timelineid"] not in old['messages'].keys():
|
||
|
|
if message['typ'] == "sprava":
|
||
|
|
if message["user_meno"] not in names:
|
||
|
|
to_show.append(message)
|
||
|
|
new_timelineids.append(message["timelineid"])
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
|
||
|
|
to_show0 = []
|
||
|
|
if no_edu0 is False:
|
||
|
|
messages = edu0["timelineItems"]
|
||
|
|
for message in messages:
|
||
|
|
if message["timelineid"] not in old['messages'].keys():
|
||
|
|
if message["timelineid"] not in new_timelineids:
|
||
|
|
if message['typ'] == "sprava":
|
||
|
|
if message["user_meno"] not in names:
|
||
|
|
to_show0.append(message)
|
||
|
|
new_timelineids.append(message["timelineid"])
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
|
||
|
|
to_show1 = []
|
||
|
|
if no_edu1 is False:
|
||
|
|
messages = edu1["timelineItems"]
|
||
|
|
for message in messages:
|
||
|
|
if message["timelineid"] not in old['messages'].keys():
|
||
|
|
if message["timelineid"] not in new_timelineids:
|
||
|
|
if message['typ'] == "sprava":
|
||
|
|
if message["user_meno"] not in names:
|
||
|
|
to_show1.append(message)
|
||
|
|
new_timelineids.append(message["timelineid"])
|
||
|
|
else:
|
||
|
|
pass
|
||
|
|
|
||
|
|
# appending new messages to old ones
|
||
|
|
for iD in new_e_superids:
|
||
|
|
old["homework"][iD] = iD
|
||
|
|
for iD in new_timelineids:
|
||
|
|
old["messages"][iD] = iD
|
||
|
|
|
||
|
|
oldMain[str(ctx.guild.id)] = old
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/old.json", "w") as f:
|
||
|
|
json.dump(oldMain, f, indent=4)
|
||
|
|
|
||
|
|
await ctx.send("Done")
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def filter_name(self, ctx, *, name):
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/names.json", "r") as f:
|
||
|
|
names = json.load(f)
|
||
|
|
|
||
|
|
try:
|
||
|
|
names[str(ctx.guild.id)]
|
||
|
|
except Exception as e:
|
||
|
|
names[str(ctx.guild.id)] = []
|
||
|
|
|
||
|
|
names[str(ctx.guild.id)].append(str(name))
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/names.json", "w") as f:
|
||
|
|
json.dump(names, f, indent=4)
|
||
|
|
|
||
|
|
await ctx.send(f"Name {name} added to filter.")
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def clear_filter(self, ctx):
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/names.json", "r") as f:
|
||
|
|
names = json.load(f)
|
||
|
|
|
||
|
|
names.pop(str(ctx.guild.id))
|
||
|
|
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/names.json", "w") as f:
|
||
|
|
json.dump(names, f, indent=4)
|
||
|
|
|
||
|
|
await ctx.send("Filter cleared.")
|
||
|
|
|
||
|
|
@commands.command()
|
||
|
|
@commands.has_permissions(administrator=True)
|
||
|
|
async def HOW_TO_SETUP(self, ctx):
|
||
|
|
tutorial = discord.Embed(title="Tutorial")
|
||
|
|
|
||
|
|
tutorial.add_field(name="1. setting channel",
|
||
|
|
value="use &set_channel <channel-name> to set channel where bot should send notifications.")
|
||
|
|
tutorial.add_field(name="2. adding clients",
|
||
|
|
value="use &add_client <username-on-edupage> <password-on-edupage> <number of save>\n"
|
||
|
|
"Number of save can be either 0, 1 or 2, one save can have maximum of 1 account.\n"
|
||
|
|
"If something isnt working u can use &clear_clients to reset.")
|
||
|
|
tutorial.add_field(name="3. filtering",
|
||
|
|
value="This is for special cases, normally personal messages are filtered automatically"
|
||
|
|
" using edupage username but in some cases edupage uses different name than username"
|
||
|
|
"for messages. \nTo fix this we use &filter_name <name_to_filter> that filters private "
|
||
|
|
"messages with that name, parameter <name_to_filter> can contain spaces.\n"
|
||
|
|
"If you made a mistake u can use &clear_filter to reset filtered names.")
|
||
|
|
tutorial.add_field(name="4. not working",
|
||
|
|
value="If bot starts spamming messages and isnt stopping(this shouldnt happen)\n"
|
||
|
|
"try using &h_m.\n"
|
||
|
|
"Lastly theres &debug <true/false>, use this if u want to know if bot is dead or u did"
|
||
|
|
" a mistake while adding clients, if u did a mistake it should start sending message "
|
||
|
|
"'A problem has occurred' somewhere.")
|
||
|
|
await ctx.send(content=None, embed=tutorial)
|
||
|
|
|
||
|
|
|
||
|
|
def update_database(ctx):
|
||
|
|
# getting accounts
|
||
|
|
with open("/home/dietpi/Edu-kun_v.2.0/saves/accounts.json") as f:
|
||
|
|
accounts = json.load(f)[str(ctx.guild.id)]
|
||
|
|
|
||
|
|
for account in accounts.keys():
|
||
|
|
for i in range(6):
|
||
|
|
try:
|
||
|
|
url = 'https://spspb.edupage.org/login/edubarLogin.php'
|
||
|
|
headers = {
|
||
|
|
"User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:82.0) Gecko/20100101 Firefox/82.0"}
|
||
|
|
values = {'username': f"{accounts[account]['name']}",
|
||
|
|
'password': f"{weird_shuffle(accounts[account]['password'])}"}
|
||
|
|
edupage = requests.session()
|
||
|
|
edupage.post(url, data=values, headers=headers)
|
||
|
|
r = edupage.get("https://spspb.edupage.org/timeline/", headers=headers)
|
||
|
|
x = r.text.split("\n")
|
||
|
|
lines_to_del = []
|
||
|
|
for line in x:
|
||
|
|
if len(line) < 10000:
|
||
|
|
lines_to_del.append(line)
|
||
|
|
|
||
|
|
for line in lines_to_del:
|
||
|
|
del x[x.index(line)]
|
||
|
|
|
||
|
|
x = list(x[0])
|
||
|
|
while x[0] != "{":
|
||
|
|
del x[0]
|
||
|
|
while x[-1] != "}":
|
||
|
|
del x[-1]
|
||
|
|
x = "".join(x)
|
||
|
|
x = x.replace("false", "False").replace("true", "True").replace("null", "None")
|
||
|
|
var = eval(x)
|
||
|
|
|
||
|
|
with open(f"/home/dietpi/Edu-kun_v.2.0/saves/{accounts[account]['file']}", "r") as f:
|
||
|
|
save = json.load(f)
|
||
|
|
|
||
|
|
save[str(ctx.guild.id)] = var
|
||
|
|
|
||
|
|
with open(f"/home/dietpi/Edu-kun_v.2.0/saves/{accounts[account]['file']}", "w") as f:
|
||
|
|
json.dump(save, f, indent=4)
|
||
|
|
break
|
||
|
|
except:
|
||
|
|
time.sleep(0.5)
|
||
|
|
|
||
|
|
|
||
|
|
def weird_shuffle(message):
|
||
|
|
x = message
|
||
|
|
o = []
|
||
|
|
e = []
|
||
|
|
ind = []
|
||
|
|
gug = 0
|
||
|
|
for i in x:
|
||
|
|
o.append(i)
|
||
|
|
f = len(x) / 2
|
||
|
|
if f % 1 == 0:
|
||
|
|
o.append(" ")
|
||
|
|
gug = 1
|
||
|
|
p = len(o)
|
||
|
|
for u in range(0, p):
|
||
|
|
ind.append(u)
|
||
|
|
for ef in range(0, p):
|
||
|
|
if max(ind) - ef * 2 < 0:
|
||
|
|
break
|
||
|
|
e.append(o[max(ind) - ef * 2])
|
||
|
|
if min(ind) + 1 + (ef * 2) > max(ind):
|
||
|
|
break
|
||
|
|
e.append(o[min(ind) + 1 + (ef * 2)])
|
||
|
|
zet = ["", p]
|
||
|
|
if gug == 1:
|
||
|
|
e.remove(e[0])
|
||
|
|
zet[0] = " "
|
||
|
|
zet[1] = p - 1
|
||
|
|
gug = ""
|
||
|
|
for ups in range(0, zet[1]):
|
||
|
|
gug += e[ups]
|
||
|
|
return zet[0] + gug
|
||
|
|
|
||
|
|
|
||
|
|
def setup(client):
|
||
|
|
client.add_cog(Setup(client))
|