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

Convert takes optional masks parameter

Browse Source
This commit is contained in:
bobloy 2021-09-03 15:15:25 -04:00
parent 5ce95038f2
commit 374c521e71
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
conquest/conquest.py
View File

@ -426,9 +426,10 @@ class Conquest(commands.Cog):
await ctx.maybe_send_embed(f"Failed to cleanup masks") await ctx.maybe_send_embed(f"Failed to cleanup masks")
@_mapmaker_masks.command(name="convert") @_mapmaker_masks.command(name="convert")
async def _mapmaker_masks_convert(self, ctx: Context): async def _mapmaker_masks_convert(self, ctx: Context, mask_list: Greedy[int]):
""" """
Converts all mask images into 1-bit black-only masks. Converts the provided mask images into 1-bit black-only masks.
Processes all masks if mask_list is empty
Warning: This may take a while. Warning: This may take a while.
""" """
@ -442,7 +443,7 @@ class Conquest(commands.Cog):
return return
async with ctx.typing(): async with ctx.typing():
result = await self.mm.convert_masks() result = await self.mm.convert_masks(mask_list)
if result: if result:
await ctx.maybe_send_embed(f"All masks converted") await ctx.maybe_send_embed(f"All masks converted")
else: else:

2
conquest/regioner.py
View File

@ -504,7 +504,7 @@ class MapMaker(ConquestMap):
points = set(list(zip(found[1], found[0]))) points = set(list(zip(found[1], found[0])))
return points return points
async def convert_masks(self): async def convert_masks(self, regions):
async for mask_path in AsyncIter(self.masks_path().iterdir()): async for mask_path in AsyncIter(self.masks_path().iterdir()):
# Don't both checking if masks are in self.regions # Don't both checking if masks are in self.regions
img: Image.Image = Image.open(mask_path).convert(MASK_MODE) img: Image.Image = Image.open(mask_path).convert(MASK_MODE)