initial codebase

Author: crash

.gitignore

@@ -0,0 +1,2 @@
+/config.cfg
+/__pycache__

AeAPI.py

@@ -0,0 +1,82 @@
+import threading
+import json
+import urllib.request
+import urllib.parse
+
+class AeAPI:
+	url = "http://ae27ff.localhost/api/"
+	appname = 'Unknown-PyAeAPI-App'
+	appid = ''
+	appver = '1.0'
+	apiver='1.0'
+
+	@classmethod	
+	def query_preparedata(cls,data_fields):
+		if data_fields is None:
+			return None
+		json_fields = {k: json.dumps(v) for k,v in data_fields.items()}
+		return urllib.parse.urlencode(json_fields).encode('utf-8');
+
+	@classmethod
+	def query(cls,qname,fields={},data_fields={}):
+		result = {}
+		try:
+			fields['query']=qname
+			fields['appid']=cls.appid
+			arguments = urllib.parse.urlencode(fields)
+			qurl = cls.url + "?"+ arguments
+			req = urllib.request.Request(
+				qurl,
+				data=cls.query_preparedata(data_fields),
+				headers={
+					'User-Agent':'PyAeAPI/'+cls.apiver+' '+cls.appname+'/'+cls.appver,
+					"Content-Type": "application/x-www-form-urlencoded"
+				}
+			)
+			result = json.loads(urllib.request.urlopen(req).read().decode('utf-8'))
+		except Exception as e:
+			result={'neterror':True,'error':e}
+		else:
+			result['neterror']=False;
+		return result
+
+	@classmethod
+	def usersonrank(cls,rank):
+		return cls.query('usersonrank',{'rank':rank})
+
+	@classmethod
+	def progressions(cls,user='*',limit=100):
+		return cls.query('progressions',{'user':user,'limit':limit})
+
+	@classmethod
+	def lastrankup(cls):
+		return cls.query('lastrankup')
+
+	@classmethod
+	def userrank(cls,user):
+		return cls.query('userrank',{'user':user})
+
+	@classmethod
+	def bulkrank(cls,users):
+		return cls.query('bulkrank',{},{'users':users});	
+
+	@classmethod
+	def activity(cls,days=None):
+		if days is None:
+			return cls.query('activity')
+		return cls.query('activity',{'days':days})
+
+#	timer_func = None	
+#	def timershim():
+#		if AeAPI.timer_func is not None:
+#			print('run timer')
+#			AeAPI.timer_func()
+#			threading.Timer(60, AeAPI.timershim).start()
+#
+#
+#	def timer(timerfunc):
+#		print('set timer func')
+#		AeAPI.timer_func = timerfunc		
+#		AeAPI.timershim()
+
+

AeRankings.py

@@ -0,0 +1,37 @@
+import json
+
+from AeAPI import *
+
+class AeRankings:
+	filename = "rankings.json"
+	rankings = {}
+	updated = False	
+
+	@classmethod
+	def load(cls):
+		try:
+			cls.rankings = json.load(open(cls.filename))
+		except FileNotFoundError:
+			print("Rankings file not initialized.");
+
+	@classmethod
+	def save(cls):
+		json.dump(cls.rankings, open(cls.filename, 'w'))
+
+	@classmethod
+	def update(cls):
+		users = list(cls.rankings.keys())
+		result = AeAPI.bulkrank(users)
+		entries = result['users']
+		for entry in entries:
+			cls.rankings[entry['user']] = entry['rank']
+		cls.updated = True
+
+	@classmethod
+	def adduser(cls,user):
+		cls.rankings[user]=-1 # add user regardless of API exception
+		result = AeAPI.userrank(user)
+		#print("userrank dump: ")
+		#print(result)
+		cls.rankings[user]=result['rank']
+		cls.updated = True

DiscordPins.py

@@ -0,0 +1,82 @@
+import discord
+import json
+
+class DiscordPins:
+	filename = "pins.json"
+	client = None
+	pins = {}
+
+	@classmethod
+	def load(cls,client):
+		cls.client = client
+		try:
+			cls.pins = json.load(open(cls.filename))
+		except FileNotFoundError:
+			print("Pins file not initialized.");
+
+	@classmethod
+	def save(cls):
+                json.dump(cls.pins, open(cls.filename, 'w'))
+
+	@classmethod
+	def getmidfromcid(cls, channelid):
+		try:
+			return cls.pins[channelid]
+		except KeyError:
+			return None
+
+	@classmethod
+	def addid(cls,channelid,messageid):
+		cls.pins[channelid]=messageid
+		#cls.pins.append( {"channel":channelid, "message":messageid} );
+
+	@classmethod
+	async def getfromid(cls,channelid,messageid=None):
+		if messageid == None:
+			messageid = cls.getmidfromcid(channelid)
+		if messageid == None:
+			return None
+		try:
+			channel = cls.client.get_channel(channelid)
+			if channel == None:
+				print("couldn't find channel "+str(channelid))
+				return None
+			return await cls.client.get_message(channel, messageid)
+		except discord.NotFound:
+			return None
+
+	@classmethod
+	async def removeid(cls,channelid):
+		message = await cls.getfromid(channelid)
+		if message == None:
+			return False 
+		await cls.client.delete_message(message)
+		cls.pins.pop(channelid,None)
+		return True	
+
+	@classmethod
+	async def remove(cls,channel):
+		return await cls.removeid(channel.id)			
+
+	@classmethod
+	def add(cls,message):
+		cls.addid(message.channel.id,message.id)
+
+	@classmethod
+	async def editall(cls,new_content=None,embed=None):
+		keys = list(cls.pins.keys())
+		for channelid in keys:
+			print("edit message for channel: " + channelid)
+			message = await cls.getfromid(channelid)
+			print("   message found: "+str(message))
+			if message == None:
+				cls.pins.pop(channelid,None)
+				continue
+			await cls.client.edit_message(message, new_content, embed=embed)
+
+	@classmethod
+	async def create(cls,channel,content=None,embed=None):
+		message = await cls.client.send_message(channel,content,embed=embed)
+		cls.add(message)
+		return message
+	

config.cfg.example

@@ -0,0 +1,8 @@
+[discord]
+privkey=YOUR_DISCORD_BOT_PRIVKEY_HERE
+admins=YOUR_DISCORD_ID_HERE
+
+[ae27ff]
+appid=YOUR_AE27FF_APPID_KEY_HERE
+appname=ae-Bot
+appver=0.1

main.py

@@ -0,0 +1,164 @@
+import discord
+import asyncio
+import json
+import configparser
+
+from AeAPI import * 
+from AeRankings import *
+from DiscordPins import *
+
+config = configparser.ConfigParser()
+client = discord.Client()
+clientkey = None
+description = '''Placeholder commands description'''
+
+def getconfig(section,key,default):
+	if not config.has_option(section,key):
+		return default
+	return config.get(section,key)	
+
+def load():
+	global client
+	global clientkey
+	config.read('config.cfg')
+	AeAPI.appid = getconfig('ae27ff','appid','ae27ff-example-app-id-000-0000-000000000000')
+	AeAPI.appname = getconfig('ae27ff','appname','ae-Bot')
+	AeAPI.appver = getconfig('ae27ff','appver','0.1')
+	clientkey = getconfig('discord','privkey','NO-PRIVKEY-SPECIFIED')
+	AeRankings.load()
+	DiscordPins.load(client)
+
+def save():
+	DiscordPins.save()
+	AeRankings.save()
+
+def start():
+	client.loop.create_task(my_background_task())
+	client.run(clientkey)
+
+def stop():
+	save()
+
+
+
+
+
+##########################################################################
+
+def get_activity_string():
+	gstatus='error'
+	try:
+		gstatus = "ae27ff - "+str(AeAPI.activity()['active'])+" users. "+str(AeAPI.activity(1)['active'])+" on today."
+		print(gstatus)
+	except Exception as e:
+		print(e)
+		print(AeAPI.activity())
+		print(AeAPI.activity(1))
+	return gstatus
+
+
+def link(text,uri=""):
+	return "["+text+"](https://ae27ff.meme.tips"+uri+")"
+
+def field(level,users):
+	f = {}
+	userlinks = "";
+	for user in users:
+		userlinks+=link(user,"/u/"+user)+" "
+	f['name'] = "__**Level "+str(level)+"**__";
+	f['value'] = userlinks
+	f['inline']=False
+	return f
+
+def userstofields():
+	fields = []
+	for rank in range(100,0,-1):
+		users = [u for (u,r) in AeRankings.rankings.items() if r == rank]
+		if len(users)>0:
+			fields.append(field(rank,users))
+	return fields
+
+
+async def my_background_task():
+	prefix="!ae"
+	preEmbedMessage = '**Add your name to the rankings!** \nJust use the command: ```http\n'+prefix+'adduser YourAE27FFName```\n\n**Solved a new level and want to show your progress on the rankings?** \nJust use the command: ```http\n'+prefix+'update```\n\nFor bot managing commands (only for moderators) use: ```http\n'+prefix+'modHelp```';
+
+#	embed = discord.Embed(
+#		title="***AE27FF RANKINGS***",
+#		type="rich",
+#		description="List of member's current levels on ",
+#		colour=11413503,
+
+
+	embed = discord.Embed(**{
+		"title": "**AE27FF RANKINGS**",
+  		"description": "Member levels on " + link("ae27ff"),
+		"color": 11413503,
+	});
+
+	embed.set_footer(**{
+		"icon_url": "https://cdn.discordapp.com/avatars/431930297328730114/da26cbbc0f89134763b03396783a96cb.png",
+		"text": "Bot inspired by Luis"
+	});
+
+
+
+	await client.wait_until_ready()
+	while not client.is_closed:
+		AeRankings.save()
+		DiscordPins.save()
+		AeRankings.update()
+		embed.clear_fields()
+		for field in userstofields():
+			embed.add_field(**field)
+		await DiscordPins.editall(preEmbedMessage,embed)
+		await client.change_presence(game=discord.Game(name=get_activity_string()))
+		await asyncio.sleep(60) # task runs every 60 seconds
+
+
+
+@client.event
+async def on_ready():
+	print('Logged in as')
+	print(client.user.name)
+	print(client.user.id)
+	print('------')
+	AeRankings.adduser("crashdemons")
+	AeRankings.adduser("amone")
+	AeRankings.update()
+	print(AeRankings.rankings)
+	await client.change_presence(game=discord.Game(name='ae27ff'))
+
+@client.event
+async def on_message(message):
+	if message.content.startswith('!aecreate'):
+		tmp = await DiscordPins.create(message.channel,"Loading content...")
+	elif message.content.startswith('!aedelete'):
+		result = await DiscordPins.remove(message.channel)
+		if result:
+			await client.send_message(message.channel,"Channel message deleted.")
+		else:
+			await client.send_message(message.channel,"Channel message not found.")
+	elif message.content.startswith('!aeadduser'):
+		param = message.content[len('!aeadduser '):]
+		if len(param)>0:
+			AeRankings.adduser(param)
+			await client.send_message(message.channel,"Added "+str(param)+" to rankings")
+		else:
+			await client.send_message(message.channel,"Which user?")
+			
+
+#this is standard, but the example command is never fired (when using client = commands.Bot(...)), so the discord.ext syntax seems useless(?)
+#@bot.command()
+#async def aeadduser(user: str):
+#	AeRankings.adduser(user)
+#	await bot.say("Added user "+str(user)+" to rankings.")
+
+
+
+
+
+######################################################################
+
+load()
+start()