Initial commit
This commit is contained in:
Untriex Programming
@@ -0,0 +1,407 @@
|
||||
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))
|
||||
Reference in New Issue
Block a user