#10 add logging instead of print()

daily_dragon.py

@@ -1,7 +1,11 @@
+import logging
+
 from openai import OpenAI
 
 import prompts
 
+logger = logging.getLogger(__name__)
+
 LANGUAGES = {'Chinese', 'Japanese'}
 
 
@@ -16,15 +20,15 @@ def __init__(self):
     def get_daily_word(self):
         prompt = prompts.get_daily_word_prompt()
         prompt = prompt.format(language=self.language)
-        print(f"Language: {self.language}")
+        logger.info(f"Language: {self.language}")
         completion = self.openai_client.chat.completions.create(
             model="gpt-3.5-turbo",
             messages=[
                 {"role": "system", "content": f"You are a teacher of {self.language}."},
                 {"role": "user", "content": f"${prompt}"}
             ]
         )
-        print(completion)
+        logger.info(completion)
         return completion.choices[0].message.content
 
     def set_language(self, language: str):

daily_dragon_bot.py

@@ -17,6 +17,7 @@
 if TELEGRAM_TOKEN is None or TELEGRAM_TOKEN == '':
     raise ValueError('TELEGRAM_TOKEN not found in .env file')
 
+logger = logging.getLogger(__name__)
 daily_dragon = DailyDragon()
 
 
@@ -25,7 +26,7 @@
 
 async def random_word(update: Update, context: ContextTypes.DEFAULT_TYPE):
     username = update.effective_user.username
-    print(f'User {username} requested daily word')
+    logger.info(f'User {username} requested daily word')
     if username == RAINY_BABE:
         daily_dragon.set_language('Japanese')
     daily_word_response = daily_dragon.get_daily_word()