level.py

import asyncio, json, time, random
import discord.utils
import math, re
from commands import is_blocked
from operator import itemgetter

user_data = []
level_data = []
# level_data: {"Name"="<name>", "LevelPoints" = <points>}
FILE_PATH = "level_data.json"

LEVEL_MAX = 100
LEVEL_RANKS = ["Newbie", "Rookie", "General", "Lieutenant", "Major", "Colonel", "Commandant", "Captain", "Master", "God"]
LEVEL_EXPERIENCE_NEEDED = 350
LEVEL_RANDOM_VALUE_MIN = 15
LEVEL_RANDOM_VALUE_MAX = 40
LEVEL_TIMER = 300

clock = 0


def load():
	import os
	if os.path.exists(FILE_PATH):
		content = open(FILE_PATH, "r").read()
		data = json.loads(content)
		if data:
			global level_data
			level_data = data

	global clock
	clock = time.time()


def save():
	f = open(FILE_PATH, "w")
	data = json.dumps(level_data)
	f.write(data)
	f.close()


def process_message(message):
	if is_blocked(message.channel):
		return
	if message.content.startswith("!status"):
		return
	for user in user_data:
		if message.author.id in user["id"]:
			user["nof"] += 1
			return
	user_data.append({"id": message.author.id, "nof": 1})


def tick():
	global clock, LEVEL_TIMER, level_data
	if time.time() - clock >= LEVEL_TIMER:
		for user in user_data:
			if user["nof"] > 0:
				userID = user["id"]
				found = False
				for leveluser in level_data:
					if "id" in leveluser and userID == leveluser["id"]:
						leveluser["exp"] += random.randint(LEVEL_RANDOM_VALUE_MIN, LEVEL_RANDOM_VALUE_MAX)
						found = True

				if not found:
					level_data.append({"id": user["id"], "exp": random.randint(LEVEL_RANDOM_VALUE_MIN, LEVEL_RANDOM_VALUE_MAX)})
		clock = time.time()
		user_data.clear()


@asyncio.coroutine
def status(message, client):
	@asyncio.coroutine
	def print_status(client, author, channel, values):
		name = author.name
		if author.nick:
			name = author.nick

		# Configure embedded message
		title = values[1]
		description = "Level: " + values[0] + "\nExp: " + values[2] + "/" + str(LEVEL_EXPERIENCE_NEEDED) +"\nRank: " + values[3]

		# Embed the message
		em = discord.Embed(title=title, description=description, colour=0x5942f4)
		em.set_author(name=author.name, icon_url=author.avatar_url)

		# Post the embedded message
		yield from client.send_message(channel, embed=em)

	if is_blocked(message.channel):
		return

	if not message.server:
		yield from client.send_message(message.author, "You can't do that here, you must send from a server.")
		return

	if message.content.lower() == "!status" or message.content.lower() == "!status ":
		values = get_data(message.author.id)
		yield from print_status(client=client, author=message.author, channel=message.channel, values=values)
		return
	else:
		if message.mentions:
			values = get_data(message.mentions[0].id)
			yield from print_status(client=client, author=message.mentions[0], channel=message.channel, values=values)
			return

		args = message.content[8:]
		names = re.findall('"([^"]*)"', args)

		if len(names) == 0:
			names = args.split(" ")
			if len(names) == 0:
				names = [args]

		print(names)
		x = 0
		if len(names) > 2:
			x = 2
		else:
			x = len(names)
		for i in range(0, x):
			member = discord.utils.get(message.server.members, name=names[i])
			if member:
				values = get_data(member.id)
				yield from print_status(client=client, author=member, channel=message.channel, values=values)
			else:
				member = discord.utils.get(message.server.members, nick=names[i])
				if member:
					values = get_data(member.id)
					yield from print_status(client=client, author=member, channel=message.channel, values=values)
				else:
					yield from client.send_message(message.channel, "Member '" + names[i] + "' not found!")


def get_data(clientid):
	global level_data
	newlist = sorted(level_data, key=itemgetter('exp'), reverse=True)
	data = []
	place = 1
	for user in newlist:
		if 'id' in user and user['id'] == clientid:
			data.append(str(math.floor(1 + (user["exp"] / LEVEL_EXPERIENCE_NEEDED))))
			rank = int(math.floor(int(data[0]) / 10))
			if rank >= 9: rank = 9
			if rank < 0: rank = 0
			data.append(LEVEL_RANKS[rank])
			data.append(str(user["exp"] - (int(data[0]) - 1) * LEVEL_EXPERIENCE_NEEDED))
			data.append(str(place))
			return data
		place += 1

	data.append("1")
	data.append(LEVEL_RANKS[0])
	data.append("0")
	data.append("N/A")
	return data


@asyncio.coroutine
def give_exp(message, client):
	# check syntax
	data = message.content.split(" ")
	if data and len(data) >= 3:
		try:
			amount = int(data[1])
		except ValueError:
			yield from client.send_message(message.channel, "Syntax incorrect")
			return

		if amount < 0:
			yield from client.send_message(message.channel, "Amount can't be negative.")
			return

		name = data[2]
		for i in range(3, len(data)):
			name += " " + data[i]

		found = False

		if message.mentions:
			for leveluser in level_data:
				if "id" in leveluser and message.mentions[0].id == leveluser["id"]:
					leveluser["exp"] += amount
					found = True

			if not found:
				level_data.append({"id": message.mentions[0].id, "exp": amount})

			return
		member = discord.utils.get(message.server.members, name=name)
		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					leveluser["exp"] += amount
					found = True

			if not found:
				level_data.append({"id": member.id, "exp": amount})
			return

		member = discord.utils.get(message.server.members, nick=name)

		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					leveluser["exp"] += amount
					found = True

			if not found:
				level_data.append({"id": member.id, "exp": amount})
			return
		yield from client.send_message(message.channel, "Member not found.")
	else:
		yield from client.send_message(message.channel, "Syntax incorrect.")


@asyncio.coroutine
def take_exp(message, client):
	# check syntax
	data = message.content.split(" ")
	if data and len(data) >= 3:
		try:
			amount = int(data[1])
		except ValueError:
			yield from client.send_message(message.channel, "Syntax incorrect")
			return

		if amount < 0:
			yield from client.send_message(message.channel, "Amount can't be negative.")
			return

		name = data[2]
		for i in range(3, len(data)):
			name += " " + data[i]

		if message.mentions:
			for leveluser in level_data:
				if "id" in leveluser and message.mentions[0].id == leveluser["id"]:
					if leveluser["exp"] < amount: leveluser["exp"] = 0
					else: leveluser["exp"] -= amount
			return

		member = discord.utils.get(message.server.members, name=name)
		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					if leveluser["exp"] < amount: leveluser["exp"] = 0
					else: leveluser["exp"] -= amount
			return

		member = discord.utils.get(message.server.members, nick=name)
		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					if leveluser["exp"] < amount: leveluser["exp"] = 0
					else: leveluser["exp"] -= amount
			return

		yield from client.send_message(message.channel, "Member not found.")
	else:
		yield from client.send_message(message.channel, "Syntax incorrect.")


@asyncio.coroutine
def set_exp(message, client):
	# check syntax
	data = message.content.split(" ")
	if data and len(data) >= 3:
		try:
			amount = int(data[1])
		except ValueError:
			yield from client.send_message(message.channel, "Syntax incorrect")
			return

		if amount < 0:
			yield from client.send_message(message.channel, "Amount can't be negative.")
			return

		name = data[2]
		for i in range(3, len(data)):
			name += " " + data[i]

		found = False

		if message.mentions:
			for leveluser in level_data:
				if "id" in leveluser and message.mentions[0].id == leveluser["id"]:
					leveluser["exp"] = amount
					found = True

			if not found:
				level_data.append({"id": message.mentions[0].id, "exp": amount})

			return
		member = discord.utils.get(message.server.members, name=name)
		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					leveluser["exp"] = amount
					found = True

			if not found:
				level_data.append({"id": member.id, "exp": amount})
			return

		member = discord.utils.get(message.server.members, nick=name)

		if member:
			for leveluser in level_data:
				if "id" in leveluser and member.id == leveluser["id"]:
					leveluser["exp"] = amount
					found = True

			if not found:
				level_data.append({"id": member.id, "exp": amount})
			return
		yield from client.send_message(message.channel, "Member not found.")
	else:
		yield from client.send_message(message.channel, "Syntax incorrect.")


@asyncio.coroutine
def top(message, client):
	def getName(id):
		for member in message.server.members:
			if member.id == id:
				if member.nick:
					return member.nick
				return member.name
		return "Not found."

	data = message.content.split(" ")
	if len(data) < 2:
		yield from client.send_message(message.channel, "Syntax incorrect.")
		return
	if len(data) >= 3:
		amount1 = max(int(data[1]), 1)
		amount2 = max(min(int(data[2]), 30), 1)
		newlist = sorted(level_data, key=itemgetter('exp'), reverse=True)

		description = ""

		for i in range(amount1 - 1, amount1 + amount2):
			if i < len(newlist):
				description += str(i + 1) + ". " + getName(newlist[i]['id']) + " - Level " + \
								str(int(newlist[i]['exp'] / LEVEL_EXPERIENCE_NEEDED) + 1) + "\n"

		em = discord.Embed(title="Top from " + str(amount1) + " to " + str(amount1 + amount2), description=description, colour=0x47ef6f)

		yield from client.send_message(message.channel, embed=em)
		return

	amount = int(data[1])
	amount = max(min(amount, 30), 1)
	newlist = sorted(level_data, key=itemgetter('exp'), reverse=True)

	description = ""

	for i in range(0, amount):
		if i < len(newlist):
			description += str(i + 1) + ". " + getName(newlist[i]['id']) + " - Level " + \
							str(int(newlist[i]['exp'] / LEVEL_EXPERIENCE_NEEDED) + 1) + "\n"

	em = discord.Embed(title="Top " + str(amount), description=description, colour=0x47ef6f)

	yield from client.send_message(message.channel, embed=em)