aljaz/chatgpt-telegram-bot
1
1
Fork 0
mirror of https://github.com/aljazceru/chatgpt-telegram-bot.git synced 2025-12-23 15:44:54 +01:00
Code Issues Packages Projects Releases Wiki Activity

made is_admin not async

Browse Source
This commit is contained in:
ned
2023-03-26 12:46:46 +02:00
parent 1a87a44421
commit 789d6cd0bc
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
bot/telegram_bot.py
View File

@@ -99,7 +99,7 @@ class ChatGPT3TelegramBot:
if budget < float('inf'): if budget < float('inf'):
text_budget += f"You have a remaining budget of ${budget:.2f} this month.\n" text_budget += f"You have a remaining budget of ${budget:.2f} this month.\n"
# add OpenAI account information for admin request # add OpenAI account information for admin request
if await self.is_admin(update): if self.is_admin(update):
grant_balance = self.openai.get_grant_balance() grant_balance = self.openai.get_grant_balance()
if grant_balance > 0.0: if grant_balance > 0.0:
text_budget += f"Your remaining OpenAI grant balance is ${grant_balance:.2f}.\n" text_budget += f"Your remaining OpenAI grant balance is ${grant_balance:.2f}.\n"
@@ -486,7 +486,7 @@ class ChatGPT3TelegramBot:
if self.config['allowed_user_ids'] == '*': if self.config['allowed_user_ids'] == '*':
return True return True
if await self.is_admin(update): if self.is_admin(update):
return True return True
allowed_user_ids = self.config['allowed_user_ids'].split(',') allowed_user_ids = self.config['allowed_user_ids'].split(',')
@@ -504,7 +504,7 @@ class ChatGPT3TelegramBot:
return False return False
async def is_admin(self, update: Update) -> bool: def is_admin(self, update: Update) -> bool:
""" """
Checks if the user is the admin of the bot. Checks if the user is the admin of the bot.
The first user in the user list is the admin. The first user in the user list is the admin.
@@ -526,7 +526,7 @@ class ChatGPT3TelegramBot:
if user_id not in self.usage: if user_id not in self.usage:
self.usage[user_id] = UsageTracker(user_id, update.message.from_user.name) self.usage[user_id] = UsageTracker(user_id, update.message.from_user.name)
if await self.is_admin(update): if self.is_admin(update):
return float('inf') return float('inf')
if self.config['monthly_user_budgets'] == '*': if self.config['monthly_user_budgets'] == '*':
@@ -557,7 +557,7 @@ class ChatGPT3TelegramBot:
if user_id not in self.usage: if user_id not in self.usage:
self.usage[user_id] = UsageTracker(user_id, update.message.from_user.name) self.usage[user_id] = UsageTracker(user_id, update.message.from_user.name)
if await self.is_admin(update): if self.is_admin(update):
return True return True
if self.config['monthly_user_budgets'] == '*': if self.config['monthly_user_budgets'] == '*':