Dmemer.py

import discord
import psycopg2
import os
import random
import asyncio
import requests
from urllib.parse import unquote
from itertools import cycle
from discord.ext import commands, tasks


bot = commands.Bot(command_prefix=['n!', 'N!'], case_insensitive = True)


status = cycle(['Looking at the records', 'transferring money', 'Waiting for drama'])


@tasks.loop(seconds=2)
async def change_status():
	await bot.change_presence(activity=discord.Game(next(status)))
	
	
def connectsql():
	DATABASE_URL = os.environ['DATABASE_URL']
	global conn
	conn = psycopg2.connect(DATABASE_URL, sslmode='require')
	global cur
	cur = conn.cursor()
	
ownerid = [708645600165625872, 419742289188093952]


@bot.event
async def on_ready():
	download_questions()
	change_status.start()
	drop.start()
	global antinsfw
	antinsfw = False
	print('Bot Online')


@bot.command()
async def ping(ctx):
	await ctx.send(f"Pong! {round(bot.latency * 1000)} ms")
	
	
@commands.has_permissions(administrator=True)
@bot.command()
async def nonsfw(ctx, option):
	global antinsfw
	if option == 'yes':
		antinsfw = True
		await ctx.send('Beginning to cleanise this chat')
	elif option == 'no':
		antinsfw = False
		await ctx.send('Okay, entering the dark side')
		
		
@commands.has_permissions(administrator=True)
@bot.command()
async def init(ctx):
	connectsql()
	cur.execute("CREATE TABLE data (id BIGINT, username TEXT, amount INTEGER)")
	conn.commit()
	conn.close()
	connectsql()
	cur.execute("CREATE TABLE trivia (id BIGINT, category TEXT, difficulty TEXT, question TEXT, correct TEXT, wrong1 TEXT, wrong2 TEXT, wrong3 TEXT)")
	conn.commit()
	conn.close()
	connectsql()
	message = ''
	for guild in bot.guilds:
		for member in guild.members:
			if member.bot:
				pass
			else:
				targetname = repr(member.name + "#" + member.discriminator)
				cur.execute(f"INSERT INTO data (id, username, amount) VALUES  ({member.id}, {targetname}, 5000)")
				message = message + '\n' + (f"Member {targetname} has been added to the database")
	await ctx.send(message)
	conn.commit()
	conn.close()
	download_questions()
	await ctx.send('Questions added to database')
	
@commands.has_permissions(administrator=True)
@bot.command()
async def wipe(ctx):
	connectsql()
	cur.execute("DROP TABLE IF EXISTS data")
	cur.execute("DROP TABLE IF EXISTS trivia")
	conn.commit()
	conn.close()
	await ctx.send("Table Wiped")
	
@bot.command()
async def rich(ctx):
	connectsql()
	currentdata = ''
	for guild in bot.guilds:
		cur.execute(f"SELECT * FROM data ORDER BY amount DESC")
		rows = cur.fetchall()
		for row in rows:
			if row is None:
				break
			currentdata = currentdata + "\n" + (f"Name: {row[1]}\nBalance: {row[2]}")
			
	await ctx.send(currentdata)
	conn.close()
	
	
@bot.command()
async def bal(ctx, target : discord.Member = None):
	if target is None:
		target = ctx.author
	connectsql()
	embed=discord.Embed(color=0xffff00)
	message = f"Balance of {target}"
	cur.execute(f"SELECT * FROM data WHERE id = {target.id}")
	data = cur.fetchone()
	amount = data[2]
	embed.add_field(name=message, value=amount, inline=True)
	embed.set_footer(text="Bot made by Xx_THK_xX")
	await ctx.send(embed=embed)
	conn.close()
	
@bot.command()
@commands.cooldown(1, 10, commands.BucketType.user)
async def rob(ctx, target : discord.Member):
	connectsql()
	attackerid = ctx.author.id
	user = target
	if user.id == attackerid:
		await ctx.send("You cannot rob yourself you dumb fuck")
		return
	cur.execute(f"SELECT * FROM data WHERE id = {user.id}")
	row = cur.fetchone()
	if row is None:
		await ctx.send("Unable to find target")
	targetbal = row[2]
	cur.execute(f"SELECT * FROM data WHERE id = {attackerid}")
	row = cur.fetchone()
	if row is None:
		await ctx.send("Unable to find your profile, are you sure you're enrolled?")
	attackerbal = row[2]
	successmin = 50
	roll1 = random.randint(1,100)
	if roll1 >= successmin:
		stolenpercent = random.randint(3, 35)
		stolen = round(targetbal * (stolenpercent / 100))
		remain = targetbal - stolen
		attackernewbal = attackerbal + stolen
		cur.execute(f"UPDATE data SET amount = {remain} WHERE id = {user.id}")
		cur.execute(f"UPDATE data SET amount = {attackernewbal} WHERE id = {attackerid}")
		await ctx.send(f"Successful Steal! You've swooped {stolen}, or {stolenpercent}% from {user}")
	else:
		roll2 = random.randint(1,100)
		death = 10
		if roll2 <= death:
			cur.execute(f"UPDATE data SET amount = 0 WHERE id = {attackerid}")
			await ctx.send("Oh fuck, you tripped over a banana and hit your head in a pile of shit, and now you're dead")
		else:
			await ctx.send("You Failed the rob, noooob")
	conn.commit()
	conn.close()
	
@rob.error
async def rob_command_error(ctx, error):
	if isinstance(error, commands.CommandOnCooldown):
		await ctx.send(f"You're robbing way too much from these noobs, you can rob them in {round(error.retry_after,2)}s")
		
def download_questions():
	print('Downloading questions from Open Trivia DB...')
	api_url = 'https://opentdb.com/api.php?amount=50&type=multiple&encode=url3986'
	r = requests.get(api_url)
	
	connectsql()
	cur.execute("DELETE FROM trivia")
	conn.commit()
	conn.close()
	
	connectsql()
	api_result = r.json()
	questions = api_result['results']
	id = 0
	for q in questions:
		id = id + 1
		category = repr(unquote(q['category']))
		difficulty = repr(unquote(q['difficulty']))
		question = repr(unquote(q['question']))
		correctans = repr(unquote(q['correct_answer']))
		badans1 = repr(unquote(q['incorrect_answers'][0]))
		badans2 = repr(unquote(q['incorrect_answers'][1]))
		badans3 = repr(unquote(q['incorrect_answers'][2]))
		cur.execute(f"INSERT INTO trivia (id, category, difficulty, question, correct, wrong1, wrong2, wrong3) VALUES (%s, %s, %s, %s, %s, %s, %s, %s)", (id, category, difficulty, question, correctans, badans1, badans2, badans3))
		
	conn.commit()
	conn.close()
	
def getquestion():
	connectsql()
	questionid = random.randint(1,50)
	cur.execute(f"SELECT * FROM trivia WHERE id = {questionid}")
	data = cur.fetchone()
	cat = data[1]
	diff = data[2]
	diff = diff.replace("'", "")
	question = data[3]
	correctans = data[4]
	correctans = correctans.replace("'", "")
	wrongans1 = data[5]
	wrongans2 = data[6]
	wrongans3 = data[7]
	allans = [correctans, wrongans1, wrongans2, wrongans3]
	random.shuffle(allans)
	
	ans1 = "A)" + allans[0]
	ans2 = "B)" + allans[1]
	ans3 = "C)" + allans[2]
	ans4 = "D)" + allans[3]
	
	answers = (f"{ans1} \n {ans2} \n {ans3} \n {ans4}")
	letternum = 0
	for a in allans:
		letternum = letternum + 1
		if correctans in a:
			break
	if letternum == 1:
		letter = "A"
	elif letternum == 2:
		letter = "B"
	elif letternum == 3:
		letter = "C"
	elif letternum == 4:
		letter = "D"
		
	response = "Category:" + cat.replace("'", "") + "\n" + "Difficulty:" + diff.replace("'", "") + "\n" + "Question:" + question.replace("'", "") + "\n " + answers.replace("'", "")
	
	return response, correctans, letter, diff
	
@tasks.loop(seconds=60)
async def drop():
	chance = random.randint(1,100)
	if chance >= 99:
		gamechannel = bot.get_channel(724274805381267498)
		death = random.randint(1,100)
		await gamechannel.send("It's time for another trivia question!")
		question, correct, letter, diff = getquestion()
		
		global amount
		if diff == 'easy':
			amount = random.randint(0,2000)
		elif diff == 'medium':
			amount = random.randint(1000,5000)
		elif diff == 'hard':
			amount = random.randint(4000,10000)
			
		global flagged
		flagged = []
		def check(m):
			if (str(m.content).upper() == str(correct).upper() or str(m.content).upper() == letter) and (m.author.id not in flagged and m.author.id != 710363488182206465):
				return True
			else:
				if m.author.id not in flagged:
					flagged.append(m.author.id)
		await gamechannel.send(question)
		try:
			answer = await bot.wait_for('message', check=check, timeout = 10.0)
		except asyncio.TimeoutError:
			await gamechannel.send(f"Oh well, look like no one's smart enough, the answer is actually {correct}")
		else:
			connectsql()
			if death == 1:
				newbal = 0
				cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {answer.author.id}")
				await gamechannel.send("Welp, you answered correctly, but some other person ran over you while you're going home so you died instead, suck to be you")
				conn.commit()
				conn.close()
			else:
				cur.execute(f"SELECT * FROM data WHERE id = {answer.author.id}")
				data = cur.fetchone()
				currentbal = data[2]
				newbal = currentbal + amount
				cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {answer.author.id}")
				await gamechannel.send(f"Smart guy, you got {amount}")
				conn.commit()
				conn.close()
		finally:
			if len(flagged) >=1:
				connectsql()
				punished = []
				for victim in flagged:
					cur.execute(f"SELECT * FROM data WHERE id = {victim}")
					data = cur.fetchone()
					bal = data[2]
					newbal = bal - round(amount / 4)
					user = await bot.fetch_user(victim)
					punished.append(user.mention)
					cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {victim}")
				await gamechannel.send(f"Bad luck to: \n {punished}, you all lose {round(amount/4)}, gg u suck")
				conn.commit()
				conn.close()
			else:
				pass
	else:
		pass
		
@commands.has_permissions(administrator=True)
@bot.command()
async def dropnow(ctx):
	if True:
		gamechannel = bot.get_channel(724274805381267498)
		death = random.randint(1,100)
		await gamechannel.send('Getting Question...')
		question, correct, letter, diff = getquestion()
		
		if diff == 'easy':
			amount = random.randint(0,2000)
		elif diff == 'medium':
			amount = random.randint(1000,5000)
		elif diff == 'hard':
			amount = random.randint(4000,10000)
		global flagged
		flagged = []
		def check(m):
			if (str(m.content).upper() == str(correct).upper() or str(m.content).upper() == letter) and m.author.id not in flagged:
				return True
			else:
				if m.author.id not in flagged and m.author.id != 710363488182206465:
					flagged.append(m.author.id)
					
		await gamechannel.send(question)
		try:
			answer = await bot.wait_for('message', check=check, timeout = 10.0)
		except asyncio.TimeoutError:
			await gamechannel.send(f"Oh well, look like no one's smart enough, the answer is actually {correct}")
		else:
			connectsql()
			if death == 1:
				newbal = 0
				cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {answer.author.id}")
				await gamechannel.send("Welp, you answered correctly, but some other person ran over you while you're going home so you died instead, suck to be you")
				conn.commit()
				conn.close()
			else:
				cur.execute(f"SELECT * FROM data WHERE id = {answer.author.id}")
				data = cur.fetchone()
				currentbal = data[2]
				newbal = currentbal + amount
				cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {answer.author.id}")
				await gamechannel.send(f"Smart guy, you got {amount}")
				conn.commit()
				conn.close()
		finally:
			if len(flagged) >=1:
				connectsql()
				punished = []
				for victim in flagged:
					cur.execute(f"SELECT * FROM data WHERE id = {victim}")
					data = cur.fetchone()
					bal = data[2]
					newbal = bal - round(amount/4)
					user = await bot.fetch_user(victim)
					punished.append(user.mention)
					cur.execute(f"UPDATE data SET amount = {newbal} WHERE id = {victim}")
				await gamechannel.send(f"Bad luck to: \n {punished}, you all lose {round(amount/4)}, gg u suck")
				conn.commit()
				conn.close()
	else:
		pass
		
		
@bot.event
async def on_message(message):
	global antinsfw
	if message.author.id == 285480424904327179 and antinsfw:
		await message.delete()
	else:
		await bot.process_commands(message)
		
		
token = os.environ.get('BOT_TOKEN')
bot.run(token)