예제 #1
0
파일: farm.py 프로젝트: derwells/shalltear
    async def setfarmname(self, ctx, name):
        """Show target's farm details."""
        if len(name) > 32:
            await ctx.send(MSG_FARM_RENAME_NAME_TOO_LONG.format(ctx.author))
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        _farm.name = name
        _account = EconomyAccount.get_economy_account(ctx.author,
                                                      self.bot.db_session)

        if not _account.has_balance(FARM_NAME_CHANGE_PRICE, raw=True):
            await ctx.send(
                MSG_INSUFFICIENT_FUNDS_EXTRA.format(
                    ctx.author, _account.get_balance(),
                    FARM_NAME_CHANGE_PRICE / 10000))
            return

        _account.add_debit(self.bot.db_session,
                           FARM_NAME_CHANGE_PRICE,
                           name="FARM NAME CHANGE",
                           raw=True)

        self.bot.db_session.add(_farm)
        self.bot.db_session.commit()

        await ctx.send(
            MSG_FARM_RENAME_SUCCESS.format(ctx.author, name,
                                           _account.get_balance()))
예제 #2
0
파일: farm.py 프로젝트: derwells/shalltear
    async def farmplots(self, ctx, page_number: int = 1):
        """Show the details of your plots."""
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        _plots = _farm.get_all_plots(self.bot.db_session)

        plot_count = len(_plots)

        paginated_plots = [_plots[i:i + 20] for i in range(0, plot_count, 20)]

        # Make sure page number is in bounds
        page_number = min(page_number, len(paginated_plots))
        page_number = max(page_number, 1)

        plot_str = ""
        plot_count = 1 + (20 * (page_number - 1))
        for _plot in paginated_plots[page_number - 1]:
            plot_str += "Plot #{0:04d}: {1}\n".format(plot_count,
                                                      _plot.get_status_str())
            plot_count += 1

        embed = discord.Embed(
            title="{0.name}#{0.discriminator}'s farm, {1}".format(
                ctx.author, _farm.name),
            color=0xffd700)

        embed.add_field(name=MSG_PLOTS_STATUS.format(page_number,
                                                     len(paginated_plots)),
                        value="```{}```".format(plot_str))

        embed.set_footer(text="Showing 20 plots per page. s!fp [page_number]")

        await ctx.send(embed=embed)
예제 #3
0
파일: farm.py 프로젝트: derwells/shalltear
    async def plotbuy(self, ctx, up_count: int = 1):
        """Buy new N plots. (N <= 1M)"""
        up_count = max(1, up_count)
        up_count = min(1000000, up_count)
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        _account = EconomyAccount.get_economy_account(ctx.author,
                                                      self.bot.db_session)

        plot_count = _farm.get_plot_count()
        to_max_plots = FARM_PLOTS_MAX - plot_count
        if to_max_plots == 0:
            await ctx.send(
                "**{0.mention}, you have reached maximum plot count.**".format(
                    ctx.author))
            return

        up_count = min(up_count, to_max_plots)

        price = _farm.get_next_plot_price(raw=True, up_count=up_count)
        if _account.has_balance(price, raw=True):
            _farm.add_plot(self.bot.db_session, up_count=up_count)
            _account.add_debit(self.bot.db_session,
                               price,
                               name="PLOTBUY",
                               raw=True)
            await ctx.send(
                MSG_PLOT_BUY_SUCCESS.format(ctx.author, _account.get_balance(),
                                            up_count))
        else:
            await ctx.send(
                MSG_INSUFFICIENT_FUNDS_EXTRA.format(ctx.author,
                                                    _account.get_balance(),
                                                    price / 10000))
예제 #4
0
파일: farm.py 프로젝트: derwells/shalltear
    async def farmsell(self, ctx, plant_name):
        """Sell all of your target crop."""
        _plant = Plant.get_plant(self.bot.db_session, plant_name)
        if _plant is None:
            await ctx.send(MSG_PLANT_NOT_FOUND.format(ctx.author))
            return

        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        _account = EconomyAccount.get_economy_account(ctx.author,
                                                      self.bot.db_session)

        total_amount = 0
        for i in range(len(_farm.harvests))[::-1]:
            if _farm.harvests[i].plant.id == _plant.id:
                total_amount += _farm.harvests[i].amount
                del _farm.harvests[i]

        if total_amount == 0:
            await ctx.send(MSG_SELL_NONE.format(ctx.author, _plant.name))
            return

        plant_sell_price = _plant.get_sell_price(raw=True)
        raw_credit = total_amount * plant_sell_price
        _account.add_credit(self.bot.db_session,
                            raw_credit,
                            name="S:{0.id}={1}".format(_plant, total_amount),
                            raw=True)

        _plant.decrement_demand(self.bot.db_session, total_amount)
        _farm.decrease_storage(self.bot.db_session, total_amount)

        await ctx.send(
            MSG_SELL_SUCCESS.format(ctx.author, total_amount, _plant,
                                    raw_credit / 10000, _account.get_balance(),
                                    plant_sell_price / 10000))
예제 #5
0
파일: farm.py 프로젝트: derwells/shalltear
    async def showharvests(self, ctx):
        """Show your harvest inventory."""
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)

        harvested_plants = set([i.plant for i in _farm.harvests])
        print(_farm.harvests)

        id_to_plant_name = {
            _plant.id: _plant.name
            for _plant in harvested_plants
        }
        total_harvests = {_plant.id: 0 for _plant in harvested_plants}

        if len(_farm.harvests) == 0:
            await ctx.send(MSG_SHOW_HARVESTS_NONE.format(ctx.author))
            return

        for harvest in _farm.harvests:
            total_harvests[harvest.plant.id] += harvest.amount

        harvest_str = ""

        embed = discord.Embed(
            title="{0.name}#{0.discriminator}'s Harvests".format(ctx.author),
            color=0xffd700)

        for harvest in total_harvests:
            harvest_str += "**{0}**, {1} units\n".format(
                id_to_plant_name[harvest], total_harvests[harvest])

        embed.add_field(name="Crops", value=harvest_str, inline=False)

        # await ctx.send(MSG_SHOW_HARVESTS.format(ctx.author, harvest_str))
        await ctx.send(embed=embed)
예제 #6
0
파일: farm.py 프로젝트: derwells/shalltear
 async def siloprice(self, ctx, up_count: int = 1):
     """Show the price of up to the next 1M silos (storage upgrade)."""
     up_count = max(1, up_count)
     up_count = min(1000000, up_count)
     _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
     result = _farm.get_next_storage_upgrade_price(up_count=up_count)
     await ctx.send(
         MSG_SILO_PRICE_CHECK.format(ctx.author, result, up_count))
예제 #7
0
파일: farm.py 프로젝트: derwells/shalltear
 async def purgeplots(self, ctx, target: discord.Member = None):
     """(Owner) Purge the plants planted in a target's plots."""
     if target == None:
         target = ctx.author
     _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
     for _plot in _farm.plots:
         _plot.plant = None
         _plot.planted_at = None
         self.bot.db_session.add(_plot)
     self.bot.db_session.add(_plot)
     await ctx.send(
         "**{0.mention}'s plots have been purged.**".format(target))
예제 #8
0
파일: farm.py 프로젝트: derwells/shalltear
    async def trashplots(self, ctx, scrap_range=None):
        """Discard the plants on your plots. No refunds.
        Scrap range can be:
        - a number, to scrap that single plot
        - a range (1-4, 5-7, etc.) with NO SPACES to scrap those plots, the numbers included
        You can also choose to leave the scrap_range blank, to scrap ALL your plots.
        """
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        plot_count = len(_farm.plots)

        all_plots = False

        if scrap_range is not None:
            try:
                scrap_range = list(map(int, scrap_range.split('-')))
            except ValueError:
                await ctx.send(MSG_CMD_INVALID.format(ctx.author))
                return
        else:
            all_plots = True
            scrap_range = [1, plot_count]

        # Command validation:
        if len(scrap_range) > 2:
            await ctx.send(MSG_CMD_INVALID.format(ctx.author))
            return

        if len(scrap_range) == 1:
            scrap_range.append(scrap_range[0])

        if scrap_range[1] < scrap_range[0]:
            await ctx.send(MSG_CMD_INVALID.format(ctx.author))
            return

        if not (0 <= scrap_range[0] - 1 <
                plot_count) or not (0 <= scrap_range[1] - 1 < plot_count):
            await ctx.send(MSG_DISCARD_OUT_OF_RANGE.format(ctx.author))
            return

        for i in range(scrap_range[0] - 1, scrap_range[1]):
            _farm.plots[i].plant = None
            _farm.plots[i].planted_at = None
            self.bot.db_session.add(_farm.plots[i])
        self.bot.db_session.commit()

        if all_plots:
            await ctx.send(MSG_DISCARD_ALL.format(ctx.author))
        elif scrap_range[0] == scrap_range[1]:
            await ctx.send(MSG_DISCARD_SINGLE.format(ctx.author, scrap_range))
        else:
            await ctx.send(MSG_DISCARD_RANGE.format(ctx.author, scrap_range))
예제 #9
0
파일: farm.py 프로젝트: derwells/shalltear
    async def farmplant(self, ctx, plant_name, plant_count=1):
        """Plant a crop on a number of your available plots."""
        _plant = Plant.get_plant(self.bot.db_session, plant_name)
        if _plant is None:
            await ctx.send(MSG_PLANT_NOT_FOUND.format(ctx.author))
            return
        _account = EconomyAccount.get_economy_account(ctx.author,
                                                      self.bot.db_session)

        total_price = _plant.buy_price * plant_count

        if not _account.has_balance(total_price, raw=True):
            await ctx.send(
                MSG_INSUFFICIENT_FUNDS.format(ctx.author,
                                              _account.get_balance()))
            return
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        _plots = _farm.get_available_plots(self.bot.db_session)
        if len(_plots) == None:
            await ctx.send(MSG_PLOT_NOT_FOUND.format(ctx.author))
            return
        if len(_plots) < plant_count:
            await ctx.send(
                MSG_PLANT_NO_PLOTS.format(ctx.author, len(_plots),
                                          plant_count))
            return

        _account.add_debit(
            self.bot.db_session,
            total_price,
            name="B:{0.id}={0.buy_price}".format(_plant),
            raw=True,
        )

        for _plot in _plots[:plant_count]:
            _plot.plant_to_plot(_plant,
                                self.bot.db_session,
                                commit_on_execution=False)
        self.bot.db_session.commit()

        await ctx.send(
            MSG_PLOT_PLANT.format(ctx.author, _plant, plant_count,
                                  total_price / 10000, _account.get_balance()))
예제 #10
0
파일: farm.py 프로젝트: derwells/shalltear
    async def plotprice(self, ctx, up_count: int = 1):
        """Show the price of the next N plots. (N <= 1M)"""
        up_count = max(1, up_count)
        up_count = min(1000000, up_count)
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)

        plot_count = _farm.get_plot_count()
        to_max_plots = FARM_PLOTS_MAX - plot_count
        if to_max_plots == 0:
            await ctx.send(
                "**{0.mention}, you have reached maximum plot count.**".format(
                    ctx.author))
            return

        up_count = min(up_count, to_max_plots)

        result = _farm.get_next_plot_price(up_count=up_count)
        await ctx.send(
            MSG_PLOT_PRICE_CHECK.format(ctx.author, result, up_count))
예제 #11
0
파일: farm.py 프로젝트: derwells/shalltear
    async def farm(self, ctx, target: discord.Member = None):
        """Show target's farm details."""
        if target is None:
            target = ctx.author
        _farm = ORMFarm.get_farm(target, self.bot.db_session)

        farm_name = _farm.get_name(self.bot.db_session)

        embed = discord.Embed(
            title="{0.name}#{0.discriminator}'s farm, {1}".format(
                target, farm_name),
            color=0xffd700)

        embed.add_field(name="Plots",
                        value="{0} / {1}".format(len(_farm.plots),
                                                 FARM_PLOTS_MAX))
        embed.add_field(
            name="Storage",
            value="{0.current_harvest} / {0.harvest_capacity}".format(_farm))

        await ctx.send(embed=embed)
예제 #12
0
파일: farm.py 프로젝트: derwells/shalltear
 async def silobuy(self, ctx, up_count: int = 1):
     """Buy new N silos (increases your storage by 100). (N <= 1M)"""
     up_count = max(1, up_count)
     up_count = min(1000000, up_count)
     _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
     _account = EconomyAccount.get_economy_account(ctx.author,
                                                   self.bot.db_session)
     price = _farm.get_next_storage_upgrade_price(raw=True,
                                                  up_count=up_count)
     if _account.has_balance(price, raw=True):
         _farm.upgrade_storage(self.bot.db_session, up_count)
         _account.add_debit(self.bot.db_session,
                            price,
                            name="SILOBUY",
                            raw=True)
         await ctx.send(
             MSG_SILO_BUY_SUCCESS.format(ctx.author, _account.get_balance(),
                                         up_count))
     else:
         await ctx.send(
             MSG_INSUFFICIENT_FUNDS_EXTRA.format(ctx.author,
                                                 _account.get_balance(),
                                                 price / 10000))
예제 #13
0
파일: farm.py 프로젝트: derwells/shalltear
    async def farmharvest(self, ctx, harvest_range=None):
        """Harvest all your harvestable crops."""
        _farm = ORMFarm.get_farm(ctx.author, self.bot.db_session)
        plot_count = len(_farm.plots)

        all_plots = False

        if harvest_range is not None:
            try:
                harvest_range = list(map(int, harvest_range.split('-')))
            except ValueError:
                await ctx.send(MSG_CMD_INVALID.format(ctx.author))
                return
        else:
            all_plots = True
            harvest_range = [1, plot_count]

        # Command validation:
        if len(harvest_range) > 2:
            await ctx.send(MSG_CMD_INVALID.format(ctx.author))
            return

        if len(harvest_range) == 1:
            harvest_range.append(harvest_range[0])

        if harvest_range[1] < harvest_range[0]:
            await ctx.send(MSG_CMD_INVALID.format(ctx.author))
            return

        if not (0 <= harvest_range[0] - 1 <
                plot_count) or not (0 <= harvest_range[1] - 1 < plot_count):
            await ctx.send(MSG_DISCARD_OUT_OF_RANGE.format(ctx.author))
            return

        storage_needed = 0
        for i in range(harvest_range[0] - 1, harvest_range[1]):
            storage_needed += _farm.plots[i].get_harvest_amount()

        if not _farm.has_storage(storage_needed):
            await ctx.send(
                MSG_HARVEST_NOT_ENOUGH_CAPACITY.format(
                    ctx.author,
                    max(_farm.harvest_capacity - _farm.current_harvest, 0),
                    storage_needed))
            return

        # Actual harvesting

        await ctx.send(
            "Harvesting **{0.name}#{0.discriminator}**'s crops...\n".format(
                ctx.author))

        harvest_stats = {}
        has_harvest = False

        for i in range(harvest_range[0] - 1, harvest_range[1]):
            _harvest = _farm.plots[i].harvest(self.bot.db_session,
                                              commit_on_execution=False)
            if _harvest is None:
                continue

            has_harvest = True
            if _harvest.plant.name not in harvest_stats:
                harvest_stats[_harvest.plant.name] = 0
            harvest_stats[_harvest.plant.name] += _harvest.amount

        self.bot.db_session.commit()

        if has_harvest:
            harvest_str = ""
            for plant_name in harvest_stats:
                harvest_str += "**{0}**, {1} units\n".format(
                    plant_name, harvest_stats[plant_name])
            await ctx.send(MSG_HARVEST_SUCCESS.format(ctx.author, harvest_str))
        else:
            await ctx.send(MSG_HARVEST_NONE.format(ctx.author))