juju/plugins/tildewash.py

75 lines
4.7 KiB
Python

#!/usr/bin/python3
import pinhook.plugin
import util.tilde
import subprocess
import uuid
from random import randint
from decimal import Decimal,ROUND_DOWN,ROUND_UP
@pinhook.plugin.command('&brainwash', help_text='Alias -- &cleanwash')
@pinhook.plugin.command('&cleanwash', help_text='Clean your dirty money.')
def cleanwash_money(msg):
if util.tilde.ADMIN_ONLY and msg.nick not in msg.ops:
return
if msg.channel != util.tilde.GOOD_CHAN and not util.tilde.DEBUG:
return pinhook.plugin.message("{} is a meanie and gets no washed money. **ju only works in the {} channel.**".format(msg.nick, util.tilde.GOOD_CHAN))
with open(util.tilde.SCORE_FILE, "r") as scorefile:
time = 0
scores = scorefile.readlines()
for score in scores:
name, score_on_file, timestamp = score.strip("\n").split("&^%")
if name == msg.nick:
time = int(timestamp.split('.')[0])
if time == 0 or time <= 1000: return pinhook.plugin.message("You must waste your time with &tilde twice.")
if util.tilde.too_recent(msg.timestamp, time) and not (util.tilde.DEBUG or util.tilde.TIMESKIP): return pinhook.plugin.message("You must wait until being able to use this again.")
nick = msg.nick
if msg.nick in msg.ops and msg.arg: nick = msg.arg
balance = Decimal(subprocess.check_output(['/home/jmjl/bin/coin','Ju','internal_balance',nick],universal_newlines=True))
if balance == Decimal(0): return pinhook.plugin.message("You don't have any tildecoins with us, so you can't cleanwash")
subprocess.check_output(['/home/jmjl/bin/coin','Ju','add_internal_balance',nick,f'-{str(balance)}'],universal_newlines=True).strip()
new_balance = Decimal(randint(59,110))/Decimal(100)*Decimal(balance)
new_balance = new_balance.quantize(Decimal('1.'), rounding=ROUND_DOWN)
percentage_return = new_balance/balance * Decimal(100)
percentage_return = percentage_return.quantize(Decimal('.00'), rounding=ROUND_DOWN)
jackpot_balance = (Decimal(0.10)*Decimal(balance)).quantize(Decimal('1.'), rounding=ROUND_DOWN)
with open(util.tilde.JACKPOT_FILE, "r+") as jackpotfile:
jackpot = int(jackpotfile.readline().strip("\n"))
jackpotfile.seek(0)
jackpotfile.truncate()
jackpot_now = str(jackpot + jackpot_balance)
jackpotfile.write(jackpot_now)
message = f"{msg.nick}: Cleanwashed {balance}TCN, and given the user {new_balance}TCN ({percentage_return}%) and contributed {jackpot_balance}TCN to the balance, jackpot at {jackpot_now}TCN"
util.tilde.admin_update_tilde(nick,amount=new_balance,time=msg.timestamp,comment=message)
return pinhook.plugin.message(message)
@pinhook.plugin.command('&brainwashreturnbeforeplaying', help_text='Alias -- &cleanwashreturnbeforeplaying')
@pinhook.plugin.command('&cleanwashreturnbeforeplaying', help_text='Give you back your money, if it wasn\'t dirty money.')
def cleanwash_return_before(msg):
if util.tilde.ADMIN_ONLY and msg.nick not in msg.ops:
return
if msg.channel != util.tilde.GOOD_CHAN and not util.tilde.DEBUG:
return pinhook.plugin.message("{} is a meanie and gets no washed money. **ju only works in the {} channel.**".format(msg.nick, util.tilde.GOOD_CHAN))
nick = msg.nick
if msg.nick in msg.ops and msg.arg: nick = msg.arg.split(' ')[1]
amount_to_return = None
try:
if msg.arg:
amount_to_return_str = msg.arg.split(' ')[0]
amount_to_return = Decimal(amount_to_return_str)
except Exception: return pinhook.plugin.message(f"{msg.nick}: Error, {msg.arg.split(' ')[0]} is probably not a number")
balance = Decimal(subprocess.check_output(['/home/jmjl/bin/coin','Ju','internal_balance',nick],universal_newlines=True))
if not amount_to_return: amount_to_return = balance
if amount_to_return > balance: return pinhook.plugin.message(f"{msg.nick}: Error, {amount_to_return} is more than what you have with us. (You have {balance}TCN returnable with us)")
sentproc = subprocess.run(['/home/jmjl/bin/coin','Ju','send',nick,str(amount_to_return.quantize(Decimal('.00'),rounding=ROUND_DOWN)),f"Requested by {msg.nick} at {__import__('time').time()}"],capture_output=True,text=True)
if sentproc.returncode == 0: return pinhook.plugin.message(f"{msg.nick}: Succesfully sent {amount_to_return}TCN back")
else:
err_id = str(uuid.uuid4())
return pinhook.plugin.message(f"{msg.nick}: Failed for some reason, consult bot admin. [{err_id}")
print(f"[ERROR]: ju/tildewash: Failed to return {amount_to_return}TCN from {nick} [{err_id}")
print(f"Stdout [{err_id}]")
print(sentproc.stdout)
print(f"Stderr [{err_id}]")
print(sentproc.stderr)
print(f"End [{err_id}]")