WhatTheMike/Fox-V3
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

V3 rework complete

Browse Source
This commit is contained in:
bobloy 2018-09-24 16:03:53 -04:00
parent 662b9bc221
commit 2f6a518ab6
1 changed files with 287 additions and 280 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

477
planttycoon/planttycoon.py
View File

@ -5,7 +5,7 @@ import time
from random import choice from random import choice
import discord import discord
from redbot.core import commands, Config from redbot.core import commands, Config, bank
from redbot.core.bot import Red from redbot.core.bot import Red
@ -16,9 +16,9 @@ class Gardener:
self.user = user self.user = user
self.config = config self.config = config
self.badges = [] self.badges = []
self.points = [] self.points = 0
self.products = [] self.products = {}
self.current = [] self.current = {}
async def _load_config(self): async def _load_config(self):
self.badges = await self.config.user(self.user).badges() self.badges = await self.config.user(self.user).badges()
@ -30,14 +30,52 @@ class Gardener:
await self.config.user(self.user).badges.set(self.badges) await self.config.user(self.user).badges.set(self.badges)
await self.config.user(self.user).points.set(self.points) await self.config.user(self.user).points.set(self.points)
await self.config.user(self.user).products.set(self.products) await self.config.user(self.user).products.set(self.products)
await self.config.user(self.user).current.set(self.user) await self.config.user(self.user).current.set(self.current)
async def _die_in(gardener, degradation):
#
# Calculating how much time in minutes remains until the plant's health hits 0
#
return int(gardener.current['health'] / degradation.degradation)
async def _grow_time(gardener):
#
# Calculating the remaining grow time for a plant
#
now = int(time.time())
then = gardener.current['timestamp']
return (gardener.current['time'] - (now - then)) / 60
async def _send_message(channel, message):
"""Sendsa message"""
em = discord.Embed(description=message, color=discord.Color.green())
await channel.send(embed=em)
async def _withdraw_points(gardener: Gardener, amount):
#
# Substract points from the gardener
#
points = gardener.points
if (points - amount) < 0:
return False
else:
gardener.points -= amount
return True
class PlantTycoon: class PlantTycoon:
"""Grow your own plants! Be sure to take proper care of it.""" """Grow your own plants! Be sure to take proper care of it."""
def __init__(self, bot: Red): def __init__(self, bot: Red):
self.bot = bot self.bot = bot
# #
# Loading all data # Loading all data
# #
@ -45,9 +83,9 @@ class PlantTycoon:
default_user = { default_user = {
'badges': [], 'badges': [],
'points': [], 'points': 0,
'products': [], 'products': {},
'current': [] 'current': {}
} }
self.config.register_user(**default_user) self.config.register_user(**default_user)
@ -841,9 +879,9 @@ class PlantTycoon:
# Loading bank # Loading bank
# #
self.bank = bot.get_cog('Economy').bank # self.bank = bot.get_cog('Economy').bank
async def _gardener(self, user: discord.User): async def _gardener(self, user: discord.User) -> Gardener:
# #
# This function returns an individual gardener namedtuple # This function returns an individual gardener namedtuple
@ -853,17 +891,7 @@ class PlantTycoon:
await g._load_config() await g._load_config()
return g return g
async def _grow_time(self, gardener): async def _degradation(self, gardener: Gardener):
#
# Calculating the remaining grow time for a plant
#
now = int(time.time())
then = gardener.current['timestamp']
return (gardener.current['time'] - (now - then)) / 60
async def _degradation(self, gardener):
# #
# Calculating the rate of degradation per check_completion() cycle. # Calculating the rate of degradation per check_completion() cycle.
@ -871,60 +899,33 @@ class PlantTycoon:
modifiers = sum( modifiers = sum(
[self.products[product]['modifier'] for product in gardener.products if gardener.products[product] > 0]) [self.products[product]['modifier'] for product in gardener.products if gardener.products[product] > 0])
degradation = (100 / (gardener.current['time'] / 60) * ( degradation = (100 / (gardener.current['time'] / 60) * (
self.defaults['degradation']['base_degradation'] + gardener.current['degradation'])) + modifiers self.defaults['degradation']['base_degradation'] + gardener.current['degradation'])) + modifiers
d = collections.namedtuple('degradation', 'degradation time modifiers') d = collections.namedtuple('degradation', 'degradation time modifiers')
return d(degradation=degradation, time=gardener.current['time'], modifiers=modifiers) return d(degradation=degradation, time=gardener.current['time'], modifiers=modifiers)
async def _die_in(self, gardener, degradation): # async def _get_member(self, user_id):
# #
# Calculating how much time in minutes remains until the plant's health hits 0 # #
# # Return a member object
# #
# #
# return discord.User(id=user_id) # I made it a string just to be sure
return int(gardener.current['health'] / degradation.degradation)
async def _withdraw_points(self, id, amount):
# #
# Substract points from the gardener # async def _send_notification(self, user_id, message):
# #
# #
points = self.gardeners[id]['points'] # # Sends a Direct Message to the gardener
if (points - amount) < 0: # #
return False
else:
self.gardeners[id]['points'] -= amount
return True
async def _get_member(self, user_id):
#
# Return a member object
# #
# member = await self._get_member(user_id)
# em = discord.Embed(description=message, color=discord.Color.green())
# await self.bot.send_message(member, embed=em)
return discord.User(id=str(id)) # I made it a string just to be sure async def _add_health(self, channel, gardener: Gardener, product, product_category):
async def _send_notification(self, user_id, message):
#
# Sends a Direct Message to the gardener
#
member = await self._get_member(user_id)
em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.send_message(member, embed=em)
async def _send_message(self, channel, message):
#
# Sends a message
#
em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.send_message(channel, embed=em)
async def _add_health(self, channel, id, product, product_category):
# #
# The function to add health # The function to add health
@ -933,28 +934,29 @@ class PlantTycoon:
product = product.lower() product = product.lower()
product_category = product_category.lower() product_category = product_category.lower()
if product in self.products and self.products[product]['category'] == product_category: if product in self.products and self.products[product]['category'] == product_category:
if product in self.gardeners[id]['products']: if product in gardener.products:
if self.gardeners[id]['products'][product] > 0: if gardener.products[product] > 0:
self.gardeners[id]['current']['health'] += self.products[product]['health'] gardener.current['health'] += self.products[product]['health']
self.gardeners[id]['products'][product] -= 1 gardener.products[product] -= 1
if self.gardeners[id]['products'][product] == 0: if gardener.products[product] == 0:
del [self.gardeners[id]['products'][product.lower()]] del gardener.products[product.lower()]
if product_category == "water": if product_category == "water":
emoji = ":sweat_drops:" emoji = ":sweat_drops:"
elif product_category == "fertilizer": elif product_category == "fertilizer":
emoji = ":poop:" emoji = ":poop:"
elif product_category == "tool": # elif product_category == "tool":
else:
emoji = ":scissors:" emoji = ":scissors:"
message = 'Your plant got some health back! {}'.format(emoji) message = 'Your plant got some health back! {}'.format(emoji)
if self.gardeners[id]['current']['health'] > self.gardeners[id]['current']['threshold']: if gardener.current['health'] > gardener.current['threshold']:
self.gardeners[id]['current']['health'] -= self.products[product]['damage'] gardener.current['health'] -= self.products[product]['damage']
if product_category == 'tool': if product_category == 'tool':
damage_msg = 'You used {} too many times!'.format(product) damage_msg = 'You used {} too many times!'.format(product)
else: else:
damage_msg = 'You gave too much of {}.'.format(product) damage_msg = 'You gave too much of {}.'.format(product)
message = '{} Your plant lost some health. :wilted_rose:'.format(damage_msg) message = '{} Your plant lost some health. :wilted_rose:'.format(damage_msg)
self.gardeners[id]['points'] += self.defaults['points']['add_health'] gardener.points += self.defaults['points']['add_health']
await self._save_gardeners() await gardener._save_gardener()
else: else:
message = 'You have no {}. Go buy some!'.format(product) message = 'You have no {}. Go buy some!'.format(product)
else: else:
@ -969,51 +971,56 @@ class PlantTycoon:
emcolor = discord.Color.blue() emcolor = discord.Color.blue()
elif product_category == "fertilizer": elif product_category == "fertilizer":
emcolor = discord.Color.dark_gold() emcolor = discord.Color.dark_gold()
elif product_category == "tool": # elif product_category == "tool":
else:
emcolor = discord.Color.dark_grey() emcolor = discord.Color.dark_grey()
em = discord.Embed(description=message, color=emcolor)
await self.bot.say(embed=em)
@commands.group(pass_context=True, name='gardening') em = discord.Embed(description=message, color=emcolor)
async def _gardening(self, context): await channel.send(embed=em)
@commands.group(name='gardening', autohelp=False)
async def _gardening(self, ctx: commands.Context):
"""Gardening commands.""" """Gardening commands."""
if context.invoked_subcommand is None: if ctx.invoked_subcommand is None:
prefix = context.prefix prefix = ctx.prefix
title = '**Welcome to Plant Tycoon.**\n' title = '**Welcome to Plant Tycoon.**\n'
description = 'Grow your own plant. Be sure to take proper care of yours. If it successfully grows, you get a reward.\n' description = ''''Grow your own plant. Be sure to take proper care of yours.\n
description += 'As you nurture your plant, you gain Thneeds which can be exchanged for credits.\n\n' If it successfully grows, you get a reward.\n
description += '**Commands**\n\n' As you nurture your plant, you gain Thneeds which can be exchanged for credits.\n\n
description += '``{0}gardening seed``: Plant a seed inside the earth.\n' **Commands**\n\n
description += '``{0}gardening profile``: Check your gardening profile.\n' ``{0}gardening seed``: Plant a seed inside the earth.\n
description += '``{0}gardening plants``: Look at the list of the available plants.\n' ``{0}gardening profile``: Check your gardening profile.\n
description += '``{0}gardening plant``: Look at the details of a plant.\n' ``{0}gardening plants``: Look at the list of the available plants.\n
description += '``{0}gardening state``: Check the state of your plant.\n' ``{0}gardening plant``: Look at the details of a plant.\n
description += '``{0}gardening buy``: Buy gardening supplies.\n' ``{0}gardening state``: Check the state of your plant.\n
description += '``{0}gardening convert``: Exchange Thneeds for credits.\n' ``{0}gardening buy``: Buy gardening supplies.\n
description += '``{0}shovel``: Shovel your plant out.\n' ``{0}gardening convert``: Exchange Thneeds for credits.\n
description += '``{0}water``: Water your plant.\n' ``{0}shovel``: Shovel your plant out.\n
description += '``{0}fertilize``: Fertilize the soil.\n' ``{0}water``: Water your plant.\n
description += '``{0}prune``: Prune your plant.\n' ``{0}fertilize``: Fertilize the soil.\n
``{0}prune``: Prune your plant.\n'''
em = discord.Embed(title=title, description=description.format(prefix), color=discord.Color.green()) em = discord.Embed(title=title, description=description.format(prefix), color=discord.Color.green())
em.set_thumbnail(url='https://image.prntscr.com/image/AW7GuFIBSeyEgkR2W3SeiQ.png') em.set_thumbnail(url='https://image.prntscr.com/image/AW7GuFIBSeyEgkR2W3SeiQ.png')
em.set_footer( em.set_footer(
text='This cog was made by SnappyDragon18 and PaddoInWonderland. Inspired by The Lorax (2012).') text='This cog was made by SnappyDragon18 and PaddoInWonderland. Inspired by The Lorax (2012).')
await self.bot.say(embed=em) await ctx.send(embed=em)
@_gardening.command(pass_context=True, name='seed') @_gardening.command(name='seed')
async def _seed(self, context): async def _seed(self, ctx: commands.Context):
"""Plant a seed inside the earth.""" """Plant a seed inside the earth."""
author = context.message.author author = ctx.author
# server = context.message.server # server = context.message.server
if author.id not in self.gardeners: # if author.id not in self.gardeners:
self.gardeners[author.id] = {} # self.gardeners[author.id] = {}
self.gardeners[author.id]['current'] = False # self.gardeners[author.id]['current'] = False
self.gardeners[author.id]['points'] = 0 # self.gardeners[author.id]['points'] = 0
self.gardeners[author.id]['badges'] = [] # self.gardeners[author.id]['badges'] = []
self.gardeners[author.id]['products'] = {} # self.gardeners[author.id]['products'] = {}
if not self.gardeners[author.id]['current']: gardener = await self._gardener(author)
if not gardener.current:
d = datetime.date.today() d = datetime.date.today()
month = d.month month = d.month
@ -1055,33 +1062,33 @@ class PlantTycoon:
'Once it blooms, something nice might come from it. ' \ 'Once it blooms, something nice might come from it. ' \
'If it dies, however, you will get nothing.'.format(plant['article'], plant['name'], 'If it dies, however, you will get nothing.'.format(plant['article'], plant['name'],
plant['rarity']) plant['rarity'])
if 'water' not in self.gardeners[author.id]['products']: if 'water' not in gardener.products:
self.gardeners[author.id]['products']['water'] = 0 gardener.products['water'] = 0
self.gardeners[author.id]['products']['water'] += 5 gardener.products['water'] += 5
self.gardeners[author.id]['current'] = plant gardener.current = plant
await self._save_gardeners() await gardener._save_gardener()
em = discord.Embed(description=message, color=discord.Color.green()) em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.say(embed=em)
else: else:
plant = self.gardeners[author.id]['current'] plant = gardener.current
message = 'You\'re already growing {} **{}**, silly.'.format(plant['article'], plant['name']) message = 'You\'re already growing {} **{}**, silly.'.format(plant['article'], plant['name'])
em = discord.Embed(description=message, color=discord.Color.green()) em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.say(embed=em)
@_gardening.command(pass_context=True, name='profile') await ctx.send(embed=em)
async def _profile(self, context, *, member: discord.Member = None):
@_gardening.command(name='profile')
async def _profile(self, ctx: commands.Context, *, member: discord.Member = None):
"""Check your gardening profile.""" """Check your gardening profile."""
if member: if member:
author = member author = member
else: else:
author = context.message.author author = ctx.author
if author.id in self.gardeners:
gardener = await self._gardener(author) gardener = await self._gardener(author)
em = discord.Embed(color=discord.Color.green(), description='\a\n') em = discord.Embed(color=discord.Color.green()) # , description='\a\n')
avatar = author.avatar_url if author.avatar else author.default_avatar_url avatar = author.avatar_url if author.avatar else author.default_avatar_url
em.set_author(name='Gardening profile of {}'.format(author.name), icon_url=avatar) em.set_author(name='Gardening profile of {}'.format(author.name), icon_url=avatar)
em.add_field(name='**Thneeds**', value=gardener.points) em.add_field(name='**Thneeds**', value=str(gardener.points))
if not gardener.current: if not gardener.current:
em.add_field(name='**Currently growing**', value='None') em.add_field(name='**Currently growing**', value='None')
else: else:
@ -1106,21 +1113,19 @@ class PlantTycoon:
em.add_field(name='**Products**', value=products) em.add_field(name='**Products**', value=products)
if gardener.current: if gardener.current:
degradation = await self._degradation(gardener) degradation = await self._degradation(gardener)
die_in = await self._die_in(gardener, degradation) die_in = await _die_in(gardener, degradation)
to_grow = await self._grow_time(gardener) to_grow = await _grow_time(gardener)
em.set_footer( em.set_footer(
text='Total degradation: {0:.2f}% / {1} min (100 / ({2} / 60) * (BaseDegr {3:.2f} + PlantDegr {4:.2f})) + ModDegr {5:.2f}) Your plant will die in {6} minutes and {7:.1f} minutes to go for flowering.'.format( text='Total degradation: {0:.2f}% / {1} min (100 / ({2} / 60) * (BaseDegr {3:.2f} + PlantDegr {4:.2f}))'
' + ModDegr {5:.2f}) Your plant will die in {6} minutes '
'and {7:.1f} minutes to go for flowering.'.format(
degradation.degradation, self.defaults['timers']['degradation'], degradation.time, degradation.degradation, self.defaults['timers']['degradation'], degradation.time,
self.defaults['degradation']['base_degradation'], gardener.current['degradation'], self.defaults['degradation']['base_degradation'], gardener.current['degradation'],
degradation.modifiers, die_in, to_grow)) degradation.modifiers, die_in, to_grow))
await self.bot.say(embed=em) await ctx.send(embed=em)
else:
message = 'Who?'
em = discord.Embed(description=message, color=discord.Color.red())
await self.bot.say(embed=em)
@_gardening.command(pass_context=True, name='plants') @_gardening.command(name='plants')
async def _plants(self, context): async def _plants(self, ctx):
"""Look at the list of the available plants.""" """Look at the list of the available plants."""
tick = '' tick = ''
tock = '' tock = ''
@ -1135,10 +1140,10 @@ class PlantTycoon:
em = discord.Embed(title='All plants that are growable', color=discord.Color.green()) em = discord.Embed(title='All plants that are growable', color=discord.Color.green())
em.add_field(name='\a', value=tick) em.add_field(name='\a', value=tick)
em.add_field(name='\a', value=tock) em.add_field(name='\a', value=tock)
await self.bot.say(embed=em) await ctx.send(embed=em)
@_gardening.command(pass_context=True, name='plant') @_gardening.command(name='plant')
async def _plant(self, context, *plant): async def _plant(self, ctx: commands.Context, *plant):
"""Look at the details of a plant.""" """Look at the details of a plant."""
plant = ' '.join(plant) plant = ' '.join(plant)
t = False t = False
@ -1148,8 +1153,7 @@ class PlantTycoon:
t = True t = True
break break
if t: if t:
em = discord.Embed(title='Plant statistics of {}'.format(plant['name']), color=discord.Color.green(), em = discord.Embed(title='Plant statistics of {}'.format(plant['name']), color=discord.Color.green())
description='\a\n')
em.set_thumbnail(url=plant['image']) em.set_thumbnail(url=plant['image'])
em.add_field(name='**Name**', value=plant['name']) em.add_field(name='**Name**', value=plant['name'])
em.add_field(name='**Rarity**', value=plant['rarity'].capitalize()) em.add_field(name='**Rarity**', value=plant['rarity'].capitalize())
@ -1160,35 +1164,41 @@ class PlantTycoon:
else: else:
message = 'What plant?' message = 'What plant?'
em = discord.Embed(description=message, color=discord.Color.red()) em = discord.Embed(description=message, color=discord.Color.red())
await self.bot.say(embed=em) await ctx.send_help()
await ctx.send(embed=em)
@_gardening.command(pass_context=True, name='state') @_gardening.command(name='state')
async def _state(self, context): async def _state(self, ctx):
"""Check the state of your plant.""" """Check the state of your plant."""
author = context.message.author author = ctx.author
gardener = await self._gardener(author) gardener = await self._gardener(author)
if author.id not in self.gardeners or not gardener.current: if not gardener.current:
message = 'You\'re currently not growing a plant.' message = 'You\'re currently not growing a plant.'
emcolor = discord.Color.red() em_color = discord.Color.red()
else: else:
plant = gardener.current plant = gardener.current
degradation = await self._degradation(gardener) degradation = await self._degradation(gardener)
die_in = await self._die_in(gardener, degradation) die_in = await _die_in(gardener, degradation)
to_grow = await self._grow_time(gardener) to_grow = await _grow_time(gardener)
message = 'You\'re growing {0} **{1}**. ' \ message = 'You\'re growing {0} **{1}**. ' \
'Its health is **{2:.2f}%** and still has to grow for **{3:.1f}** minutes. ' \ 'Its health is **{2:.2f}%** and still has to grow for **{3:.1f}** minutes. ' \
'It is losing **{4:.2f}%** per minute and will die in **{5:.1f}** minutes.'.format( 'It is losing **{4:.2f}%** per minute and will die in **{5:.1f}** minutes.'.format(
plant['article'], plant['name'], plant['health'], to_grow, degradation.degradation, die_in) plant['article'],
emcolor = discord.Color.green() plant['name'],
em = discord.Embed(description=message, color=emcolor) plant['health'],
await self.bot.say(embed=em) to_grow,
degradation.degradation,
die_in)
em_color = discord.Color.green()
em = discord.Embed(description=message, color=em_color)
await ctx.send(embed=em)
@_gardening.command(pass_context=True, name='buy') @_gardening.command(name='buy')
async def _buy(self, context, product=None, amount: int = 1): async def _buy(self, ctx, product=None, amount: int = 1):
"""Buy gardening supplies.""" """Buy gardening supplies."""
author = context.message.author author = ctx.author
if product is None: if product is None:
em = discord.Embed(title='All gardening supplies that you can buy:', description='\a\n', em = discord.Embed(title='All gardening supplies that you can buy:',
color=discord.Color.green()) color=discord.Color.green())
for product in self.products: for product in self.products:
em.add_field(name='**{}**'.format(product.capitalize()), em.add_field(name='**{}**'.format(product.capitalize()),
@ -1196,107 +1206,106 @@ class PlantTycoon:
self.products[product]['cost'], self.products[product]['health'], self.products[product]['cost'], self.products[product]['health'],
self.products[product]['damage'], self.products[product]['uses'], self.products[product]['damage'], self.products[product]['uses'],
self.products[product]['category'])) self.products[product]['category']))
await self.bot.say(embed=em) await ctx.send(embed=em)
else: else:
if amount <=0: if amount <= 0:
message = "Invalid amount! Must be greater than 1" message = "Invalid amount! Must be greater than 1"
else: else:
if author.id not in self.gardeners: gardener = await self._gardener(author)
message = 'You\'re currently not growing a plant.'
else:
if product.lower() in self.products and amount > 0: if product.lower() in self.products and amount > 0:
cost = self.products[product.lower()]['cost'] * amount cost = self.products[product.lower()]['cost'] * amount
withdraw_points = await self._withdraw_points(author.id, cost) withdraw_points = await _withdraw_points(gardener, cost)
if withdraw_points: if withdraw_points:
if product.lower() not in self.gardeners[author.id]['products']: if product.lower() not in gardener.products:
self.gardeners[author.id]['products'][product.lower()] = 0 gardener.products[product.lower()] = 0
self.gardeners[author.id]['products'][product.lower()] += amount gardener.products[product.lower()] += amount
self.gardeners[author.id]['products'][product.lower()] += amount * \ gardener.products[product.lower()] += amount * self.products[product.lower()]['uses']
self.products[product.lower()]['uses'] await gardener._save_gardener()
await self._save_gardeners()
message = 'You bought {}.'.format(product.lower()) message = 'You bought {}.'.format(product.lower())
else: else:
message = 'You don\'t have enough Thneeds. You have {}, but need {}.'.format( message = 'You don\'t have enough Thneeds. You have {}, but need {}.'.format(
self.gardeners[author.id]['points'], self.products[product.lower()]['cost'] * amount) gardener.points, self.products[product.lower()]['cost'] * amount)
else: else:
message = 'I don\'t have this product.' message = 'I don\'t have this product.'
em = discord.Embed(description=message, color=discord.Color.green()) em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.say(embed=em) await ctx.send(embed=em)
@_gardening.command(pass_context=True, name='convert') @_gardening.command(name='convert')
async def _convert(self, context, amount: int): async def _convert(self, ctx: commands.Context, amount: int):
"""Exchange Thneeds for credits.""" """Exchange Thneeds for credits."""
author = context.message.author author = ctx.author
if self.bank.account_exists(author): gardener = await self._gardener(author)
withdraw_points = await self._withdraw_points(author.id, amount)
plural = ""; withdraw_points = await _withdraw_points(gardener, amount)
plural = ""
if amount > 0: if amount > 0:
plural = "s"; plural = "s"
if withdraw_points: if withdraw_points:
self.bank.deposit_credits(author, amount) await bank.deposit_credits(author, amount)
message = '{} Thneed{} successfully exchanged for credits.'.format(amount, plural) message = '{} Thneed{} successfully exchanged for credits.'.format(amount, plural)
else: else:
message = 'You don\'t have enough Thneed{}. You have {}, but need {}.'.format(plural, message = 'You don\'t have enough Thneed{}. ' \
self.gardeners[author.id][ 'You have {}, but need {}.'.format(plural, gardener.points, amount)
'points'], amount)
else:
message = 'Account not found.'
em = discord.Embed(description=message, color=discord.Color.green())
await self.bot.say(embed=em)
@commands.command(pass_context=True, name='shovel') em = discord.Embed(description=message, color=discord.Color.green())
async def _shovel(self, context): await ctx.send(embed=em)
@commands.command(name='shovel')
async def _shovel(self, ctx: commands.Context):
"""Shovel your plant out.""" """Shovel your plant out."""
author = context.message.author author = ctx.author
if author.id not in self.gardeners or not self.gardeners[author.id]['current']: gardener = await self._gardener(author)
if not gardener.current:
message = 'You\'re currently not growing a plant.' message = 'You\'re currently not growing a plant.'
else: else:
self.gardeners[author.id]['current'] = False gardener.current = {}
message = 'You sucessfuly shovelled your plant out.' message = 'You sucessfuly shovelled your plant out.'
if self.gardeners[author.id]['points'] < 0: if gardener.points < 0:
self.gardeners[author.id]['points'] = 0 gardener.points = 0
await self._save_gardeners() await gardener._save_gardener()
em = discord.Embed(description=message, color=discord.Color.dark_grey())
await self.bot.say(embed=em)
@commands.command(pass_context=True, name='water') em = discord.Embed(description=message, color=discord.Color.dark_grey())
async def _water(self, context): await ctx.send(embed=em)
@commands.command(name='water')
async def _water(self, ctx):
"""Water your plant.""" """Water your plant."""
author = context.message.author author = ctx.author
channel = context.message.channel channel = ctx.channel
gardener = await self._gardener(author)
product = 'water' product = 'water'
product_category = 'water' product_category = 'water'
if author.id not in self.gardeners or not self.gardeners[author.id]['current']: if not gardener.current:
message = 'You\'re currently not growing a plant.' message = 'You\'re currently not growing a plant.'
await self._send_message(channel, message) await _send_message(channel, message)
else: else:
await self._add_health(channel, author.id, product, product_category) await self._add_health(channel, gardener, product, product_category)
@commands.command(pass_context=True, name='fertilize') @commands.command(name='fertilize')
async def _fertilize(self, context, fertilizer): async def _fertilize(self, ctx, fertilizer):
"""Fertilize the soil.""" """Fertilize the soil."""
author = context.message.author gardener = await self._gardener(ctx.author)
channel = context.message.channel channel = ctx.channel
product = fertilizer product = fertilizer
product_category = 'fertilizer' product_category = 'fertilizer'
if author.id not in self.gardeners or not self.gardeners[author.id]['current']: if not gardener.current:
message = 'You\'re currently not growing a plant.' message = 'You\'re currently not growing a plant.'
await self._send_message(channel, message) await _send_message(channel, message)
else: else:
await self._add_health(channel, author.id, product, product_category) await self._add_health(channel, gardener, product, product_category)
@commands.command(pass_context=True, name='prune') @commands.command(name='prune')
async def _prune(self, context): async def _prune(self, ctx):
"""Prune your plant.""" """Prune your plant."""
author = context.message.author gardener = await self._gardener(ctx.author)
channel = context.message.channel channel = ctx.channel
product = 'pruner' product = 'pruner'
product_category = 'tool' product_category = 'tool'
if author.id not in self.gardeners or not self.gardeners[author.id]['current']: if not gardener.current:
message = 'You\'re currently not growing a plant.' message = 'You\'re currently not growing a plant.'
await self._send_message(channel, message) await _send_message(channel, message)
else: else:
await self._add_health(channel, author.id, product, product_category) await self._add_health(channel, gardener, product, product_category)
async def check_degradation(self): async def check_degradation(self):
while 'PlantTycoon' in self.bot.cogs: while 'PlantTycoon' in self.bot.cogs:
@ -1305,54 +1314,52 @@ class PlantTycoon:
gardener = await self._gardener(user) gardener = await self._gardener(user)
if gardener.current: if gardener.current:
degradation = await self._degradation(gardener) degradation = await self._degradation(gardener)
self.gardeners[id]['current']['health'] -= degradation.degradation gardener.current['health'] -= degradation.degradation
self.gardeners[id]['points'] += self.defaults['points']['growing'] gardener.points += self.defaults['points']['growing']
await self._save_gardeners() await gardener._save_gardener()
await asyncio.sleep(self.defaults['timers']['degradation'] * 60) await asyncio.sleep(self.defaults['timers']['degradation'] * 60)
async def check_completion(self): async def check_completion(self):
while 'PlantTycoon' in self.bot.cogs: while 'PlantTycoon' in self.bot.cogs:
now = int(time.time()) now = int(time.time())
delete = False message = None
for id in self.gardeners: users = await self.config.all_users()
gardener = await self._gardener(id) for user in users:
gardener = await self._gardener(user)
if gardener.current: if gardener.current:
then = gardener.current['timestamp'] then = gardener.current['timestamp']
health = gardener.current['health'] health = gardener.current['health']
grow_time = gardener.current['time'] grow_time = gardener.current['time']
badge = gardener.current['badge'] badge = gardener.current['badge']
reward = gardener.current['reward'] reward = gardener.current['reward']
if delete:
delete = False
if (now - then) > grow_time: if (now - then) > grow_time:
self.gardeners[id]['points'] += reward gardener.points += reward
if badge not in self.gardeners[id]['badges']: if badge not in gardener.badges:
self.gardeners[id]['badges'].append(badge) gardener.badges.append(badge)
message = 'Your plant made it! You are rewarded with the **{}** badge and you have recieved **{}** Thneeds.'.format( message = 'Your plant made it! ' \
'You are rewarded with the **{}** badge and you have recieved **{}** Thneeds.'.format(
badge, reward) badge, reward)
delete = True
if health < 0: if health < 0:
message = 'Your plant died!' message = 'Your plant died!'
delete = True if message:
if delete: await user.send(message)
await self.bot.send_message(discord.User(id=str(id)), message) gardener.current = {}
self.gardeners[id]['current'] = False await gardener._save_gardener()
await self._save_gardeners()
await asyncio.sleep(self.defaults['timers']['completion'] * 60) await asyncio.sleep(self.defaults['timers']['completion'] * 60)
async def send_notification(self): async def send_notification(self):
while 'PlantTycoon' in self.bot.cogs: while 'PlantTycoon' in self.bot.cogs:
for id in self.gardeners: users = await self.config.all_users()
gardener = await self._gardener(id) for user in users:
gardener = await self._gardener(user)
if gardener.current: if gardener.current:
health = gardener.current['health'] health = gardener.current['health']
if health < self.defaults['notification']['max_health']: if health < self.defaults['notification']['max_health']:
message = choice(self.notifications['messages']) message = choice(self.notifications['messages'])
await self.bot.send_notification(gardener, message) await user.send(message)
await asyncio.sleep(self.defaults['timers']['notification'] * 60) await asyncio.sleep(self.defaults['timers']['notification'] * 60)
def __unload(self): def __unload(self):
self.completion_task.cancel() self.completion_task.cancel()
self.degradation_task.cancel() self.degradation_task.cancel()
self.notification_task.cancel() self.notification_task.cancel()
self._save_gardeners()