cevheri commited on
Commit
66944b7
·
1 Parent(s): 8588b90

refactor: refactor authentication method

Browse files
Files changed (1) hide show
  1. bot_telegram.py +5 -7
bot_telegram.py CHANGED
@@ -11,12 +11,12 @@ import io
11
  logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
12
 
13
  # Load environment variables from Hugging Face Secrets
 
14
  BOT_TOKEN = os.getenv("BOT_TOKEN")
15
  BASE_URL = os.getenv("BASE_URL")
16
  HF_TOKEN = os.getenv("HF_TOKEN")
17
- API_USERNAME = os.getenv("API_USERNAME")
18
- API_PASSWORD = os.getenv("API_PASSWORD")
19
- BOT_SECRET_PASSWORD = os.getenv("BOT_SECRET_PASSWORD")
20
 
21
  # Set the secret password for authentication
22
  SECRET_PASSWORD = BOT_SECRET_PASSWORD
@@ -29,7 +29,7 @@ AWAITING_PASSWORD = set()
29
  class TelegramBot:
30
  """A Telegram bot with password-based authentication."""
31
 
32
- def __init__(self, bot_token, base_url, username, password):
33
  """Initialize the bot with Telegram API token, API credentials, and authentication."""
34
  self.bot_token = bot_token
35
  self.base_url = base_url
@@ -254,9 +254,7 @@ class TelegramBot:
254
  if __name__ == "__main__":
255
  bot = TelegramBot(
256
  bot_token=BOT_TOKEN,
257
- base_url=BASE_URL,
258
- username=API_USERNAME,
259
- password=API_PASSWORD
260
  )
261
  bot.run()
262
 
 
11
  logging.basicConfig(level=logging.INFO, format="%(asctime)s - %(levelname)s - %(message)s")
12
 
13
  # Load environment variables from Hugging Face Secrets
14
+ BOT_SECRET_PASSWORD = os.getenv("BOT_SECRET_PASSWORD")
15
  BOT_TOKEN = os.getenv("BOT_TOKEN")
16
  BASE_URL = os.getenv("BASE_URL")
17
  HF_TOKEN = os.getenv("HF_TOKEN")
18
+ # API_USERNAME = os.getenv("API_USERNAME")
19
+ # API_PASSWORD = os.getenv("API_PASSWORD")
 
20
 
21
  # Set the secret password for authentication
22
  SECRET_PASSWORD = BOT_SECRET_PASSWORD
 
29
  class TelegramBot:
30
  """A Telegram bot with password-based authentication."""
31
 
32
+ def __init__(self, bot_token, base_url):
33
  """Initialize the bot with Telegram API token, API credentials, and authentication."""
34
  self.bot_token = bot_token
35
  self.base_url = base_url
 
254
  if __name__ == "__main__":
255
  bot = TelegramBot(
256
  bot_token=BOT_TOKEN,
257
+ base_url=BASE_URL
 
 
258
  )
259
  bot.run()
260