Use logging module for informational prints

This commit is contained in:
Adam Goldsmith 2022-03-14 12:26:09 -04:00
parent bdb9387d91
commit a018aeb1c3
1 changed files with 25 additions and 5 deletions

View File

@ -4,6 +4,7 @@ import configargparse
from datetime import datetime from datetime import datetime
import json import json
import re import re
import logging
from pathlib import Path from pathlib import Path
import discord import discord
@ -56,11 +57,11 @@ class ArkhamDBUpdater(commands.Bot):
json.dump(list(self.channel_list), f) json.dump(list(self.channel_list), f)
async def on_ready(self) -> None: async def on_ready(self) -> None:
print(f"Logged in as {self.user} (ID: {self.user.id})") logging.info(f"Logged in as {self.user} (ID: {self.user.id})")
print("Enabled on servers:") logging.info("Enabled on servers:")
async for guild in self.fetch_guilds(limit=150): async for guild in self.fetch_guilds(limit=150):
print(" -", guild.name) logging.info(f" - {guild.name}")
print("------") logging.info("------")
self.arkhamdb_monitor.start() self.arkhamdb_monitor.start()
@ -109,7 +110,7 @@ class ArkhamDBUpdater(commands.Bot):
return status return status
async def update_channel_latest_decks(self, channel: discord.TextChannel) -> None: async def update_channel_latest_decks(self, channel: discord.TextChannel) -> None:
print(f"Running update in channel {channel.guild} - {channel.name}") logging.info(f"Running update in channel {channel.guild} - {channel.name}")
async with channel.typing(): async with channel.typing():
deck_ids = await self.gather_deck_ids(channel) deck_ids = await self.gather_deck_ids(channel)
latest_decks = await self.arkhamdb_client.get_latest_decks(deck_ids) latest_decks = await self.arkhamdb_client.get_latest_decks(deck_ids)
@ -200,8 +201,27 @@ def main():
help="Discord API token", help="Discord API token",
env_var="DISCORD_TOKEN", env_var="DISCORD_TOKEN",
) )
parser.add(
"-d",
"--debug",
help="Print lots of debugging statements",
action="store_const",
dest="loglevel",
const=logging.DEBUG,
default=logging.WARNING,
)
parser.add_argument(
"-v",
"--verbose",
help="Be verbose",
action="store_const",
dest="loglevel",
const=logging.INFO,
)
args = parser.parse_args() args = parser.parse_args()
logging.basicConfig(level=args.loglevel)
bot = ArkhamDBUpdater( bot = ArkhamDBUpdater(
channel_list_file=args.channel_list, command_prefix="!arkhamdb " channel_list_file=args.channel_list, command_prefix="!arkhamdb "
) )