Info_edupage/Edu-kun_v.2.0-linux/extensions/Setup.py
Untriex Programming 9b6723e11e Initial commit
2021-08-31 22:06:02 +02:00

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))