optimize imports and pep8
This commit is contained in:
parent
9ac6d396a8
commit
deb96a73b6
@ -3,15 +3,13 @@ from collections import defaultdict
|
||||
from random import choice
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
|
||||
# Import all roles here
|
||||
from redbot.core import RedContext
|
||||
from redbot.core.utils.menus import menu, prev_page, next_page, close_menu
|
||||
|
||||
from werewolf.roles.seer import Seer
|
||||
from werewolf.roles.vanillawerewolf import VanillaWerewolf
|
||||
from werewolf.roles.villager import Villager
|
||||
from redbot.core.utils.menus import menu, prev_page, next_page, close_menu
|
||||
|
||||
# All roles in this list for iterating
|
||||
|
||||
@ -120,8 +118,6 @@ async def parse_code(code, game):
|
||||
digits += 1
|
||||
continue
|
||||
|
||||
|
||||
|
||||
try:
|
||||
idx = int(built)
|
||||
except ValueError:
|
||||
@ -146,7 +142,6 @@ async def parse_code(code, game):
|
||||
|
||||
built = ""
|
||||
|
||||
|
||||
return decode
|
||||
|
||||
|
||||
@ -321,7 +316,7 @@ class GameBuilder:
|
||||
pass
|
||||
|
||||
if page >= len(ROLE_LIST):
|
||||
self.rand_roles.append(CATEGORY_COUNT[page-len(ROLE_LIST)])
|
||||
self.rand_roles.append(CATEGORY_COUNT[page - len(ROLE_LIST)])
|
||||
else:
|
||||
self.code.append(page)
|
||||
|
||||
|
@ -2,7 +2,6 @@ import asyncio
|
||||
import random
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from redbot.core import RedContext
|
||||
|
||||
from werewolf.builder import parse_code
|
||||
|
@ -30,4 +30,4 @@ class Player:
|
||||
try:
|
||||
await self.member.send(message) # Lets do embeds later
|
||||
except discord.Forbidden:
|
||||
await self.role.game.village_channel.send("Couldn't DM {}, uh oh".format(self.mention))
|
||||
await self.role.game.village_channel.send("Couldn't DM {}, uh oh".format(self.mention))
|
||||
|
@ -16,7 +16,6 @@ class Seer(Role):
|
||||
description = "A mystic in search of answers in a chaotic town.\n" \
|
||||
"Calls upon the cosmos to discern those of Lycan blood"
|
||||
|
||||
|
||||
def __init__(self, game):
|
||||
super().__init__(game)
|
||||
# self.game = game
|
||||
|
@ -1,12 +1,11 @@
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from redbot.core import Config, checks, RedContext
|
||||
|
||||
from redbot.core.bot import Red
|
||||
from redbot.core.utils.menus import menu, DEFAULT_CONTROLS
|
||||
|
||||
from werewolf.builder import GameBuilder, role_from_name, role_from_alignment, role_from_category, role_from_id
|
||||
from werewolf.game import Game
|
||||
from redbot.core.utils.menus import menu, DEFAULT_CONTROLS
|
||||
|
||||
|
||||
class Werewolf:
|
||||
@ -74,7 +73,7 @@ class Werewolf:
|
||||
|
||||
@commands.guild_only()
|
||||
@wwset.command(name="role")
|
||||
async def wwset_role(self, ctx: RedContext, role: discord.Role=None):
|
||||
async def wwset_role(self, ctx: RedContext, role: discord.Role = None):
|
||||
"""
|
||||
Assign the game role
|
||||
This role should not be manually assigned
|
||||
@ -105,7 +104,7 @@ class Werewolf:
|
||||
|
||||
@commands.guild_only()
|
||||
@wwset.command(name="channel")
|
||||
async def wwset_channel(self, ctx: RedContext, channel: discord.TextChannel=None):
|
||||
async def wwset_channel(self, ctx: RedContext, channel: discord.TextChannel = None):
|
||||
"""
|
||||
Assign the village channel
|
||||
"""
|
||||
@ -118,7 +117,7 @@ class Werewolf:
|
||||
|
||||
@commands.guild_only()
|
||||
@wwset.command(name="logchannel")
|
||||
async def wwset_log_channel(self, ctx: RedContext, channel: discord.TextChannel=None):
|
||||
async def wwset_log_channel(self, ctx: RedContext, channel: discord.TextChannel = None):
|
||||
"""
|
||||
Assign the log channel
|
||||
"""
|
||||
@ -388,4 +387,3 @@ class Werewolf:
|
||||
return False, None, None, None, None
|
||||
|
||||
return True, role, category, channel, log_channel
|
||||
|
||||
|
Loading…
x
Reference in New Issue
Block a user