Skip to content

Commit f60f413

Browse files
committed
formatting code
1 parent b383eda commit f60f413

File tree

4 files changed

+28
-9
lines changed

4 files changed

+28
-9
lines changed

src/guildmaster/cli.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,16 @@
22
import fire
33
from guildmaster.tasks.role_manager import DiscordRoleManager
44

5+
56
class GuildMasterCommands:
67
"""
78
A CLI tool for managing Discord server tasks and settings.
89
"""
10+
911
def __init__(self):
1012
self.role = DiscordRoleManager()
1113

14+
1215
def main():
1316
try:
1417
fire.Fire(name="guildmaster", component=GuildMasterCommands)
@@ -18,5 +21,6 @@ def main():
1821
print(f"An error occurred: {e}")
1922
sys.exit(1)
2023

24+
2125
if __name__ == "__main__":
22-
main()
26+
main()

src/guildmaster/client/discord_client.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@
44

55
TOKEN = os.getenv("DISCORD_TOKEN")
66

7+
78
class DiscordClient:
89
def __init__(self):
910
intents = discord.Intents.default()
@@ -14,4 +15,4 @@ def run(self):
1415
self.bot.run(TOKEN)
1516

1617
def close(self):
17-
self.bot.close()
18+
self.bot.close()

src/guildmaster/config/logger.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
import logging
22

3+
34
class Logger:
45
@staticmethod
56
def setup_logger(name: str, level: int = logging.INFO) -> logging.Logger:
@@ -13,7 +14,9 @@ def setup_logger(name: str, level: int = logging.INFO) -> logging.Logger:
1314
Returns:
1415
logging.Logger: Configured logger instance.
1516
"""
16-
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
17+
formatter = logging.Formatter(
18+
"%(asctime)s - %(name)s - %(levelname)s - %(message)s"
19+
)
1720

1821
console_handler = logging.StreamHandler()
1922
console_handler.setFormatter(formatter)
@@ -22,4 +25,4 @@ def setup_logger(name: str, level: int = logging.INFO) -> logging.Logger:
2225
logger.setLevel(level)
2326
logger.addHandler(console_handler)
2427

25-
return logger
28+
return logger

src/guildmaster/tasks/role_manager.py

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,15 @@
55
from guildmaster.config.logger import Logger
66
from guildmaster.client.discord_client import DiscordClient
77

8+
89
class DiscordRoleManager:
910
"""
1011
A class for managing a role to every member in a Discord server.
1112
"""
13+
1214
def __init__(self):
1315
self.logger = Logger.setup_logger(__name__)
14-
16+
1517
def assign_to_all(self, guild: int, role: str, delay: float = 3.0):
1618
"""
1719
Assign a role to every member in a Discord server.
@@ -42,20 +44,29 @@ async def on_ready():
4244
await client.bot.close()
4345
return
4446

45-
self.logger.debug("Assigning role '%s' to all members of '%s'...", role, guild_obj.name)
47+
self.logger.debug(
48+
"Assigning role '%s' to all members of '%s'...", role, guild_obj.name
49+
)
4650

4751
# Iterate through each member and add the role if they don't have it
4852
members = guild_obj.members
4953
async for member in tqdm(members, desc="Assigning roles", unit="member"):
5054
if role_obj not in member.roles:
5155
try:
5256
await member.add_roles(role_obj)
53-
self.logger.debug("Added role to %s#%s", member.name, member.discriminator)
57+
self.logger.debug(
58+
"Added role to %s#%s", member.name, member.discriminator
59+
)
5460
await asyncio.sleep(delay)
5561
except Exception as e:
56-
self.logger.error("Failed to add role to %s#%s: %s", member.name, member.discriminator, e)
62+
self.logger.error(
63+
"Failed to add role to %s#%s: %s",
64+
member.name,
65+
member.discriminator,
66+
e,
67+
)
5768

5869
self.logger.debug("Finished processing all members.")
5970
await client.bot.close()
6071

61-
client.run()
72+
client.run()

0 commit comments

Comments
 (0)