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

Update builder to accept any number of roles

Browse Source
This commit is contained in:
bobloy 2020-09-24 17:06:03 -04:00
parent 61d1313411
commit f263f97cc2
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
werewolf/builder.py
View File

@ -9,21 +9,33 @@ import discord
# Import all roles here # Import all roles here
from redbot.core import commands from redbot.core import commands
from .roles.seer import Seer # from .roles.seer import Seer
from .roles.vanillawerewolf import VanillaWerewolf # from .roles.vanillawerewolf import VanillaWerewolf
from .roles.villager import Villager # from .roles.villager import Villager
from werewolf import roles
from redbot.core.utils.menus import menu, prev_page, next_page, close_menu from redbot.core.utils.menus import menu, prev_page, next_page, close_menu
from werewolf.role import Role
log = logging.getLogger("red.fox_v3.werewolf.builder") log = logging.getLogger("red.fox_v3.werewolf.builder")
# All roles in this list for iterating # All roles in this list for iterating
ROLE_LIST = sorted([Villager, Seer, VanillaWerewolf], key=lambda x: x.alignment) ROLE_DICT = {name: cls for name, cls in roles.__dict__.items() if isinstance(cls, type)}
ROLE_LIST = sorted(
[cls for cls in ROLE_DICT.values()],
key=lambda x: x.alignment,
)
log.debug(f"{ROLE_DICT=}")
ALIGNMENT_COLORS = [0x008000, 0xFF0000, 0xC0C0C0] ALIGNMENT_COLORS = [0x008000, 0xFF0000, 0xC0C0C0]
TOWN_ROLES = [(idx, role) for idx, role in enumerate(ROLE_LIST) if role.alignment == 1] # TOWN_ROLES = [(idx, role) for idx, r_tuple in enumerate(ROLE_LIST) if role.alignment == 1]
WW_ROLES = [(idx, role) for idx, role in enumerate(ROLE_LIST) if role.alignment == 2] # WW_ROLES = [(idx, role) for idx, r_tuple in enumerate(ROLE_LIST) if role.alignment == 2]
OTHER_ROLES = [(idx, role) for idx, role in enumerate(ROLE_LIST) if role.alignment not in [0, 1]] # OTHER_ROLES = [
# (idx, role) for idx, r_tuple in enumerate(ROLE_LIST) if role.alignment not in [0, 1]
# ]
ROLE_PAGES = [] ROLE_PAGES = []
PAGE_GROUPS = [0] PAGE_GROUPS = [0]
@ -81,9 +93,7 @@ def setup():
for k, v in ROLE_CATEGORIES.items(): for k, v in ROLE_CATEGORIES.items():
if 0 < k <= 6: if 0 < k <= 6:
ROLE_PAGES.append( ROLE_PAGES.append(
discord.Embed( discord.Embed(title="RANDOM:Town Role", description=f"Town {v}", color=0x008000)
title="RANDOM:Town Role", description=f"Town {v}", color=0x008000
)
) )
CATEGORY_COUNT.append(k) CATEGORY_COUNT.append(k)