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'):
text_budget += f"You have a remaining budget of ${budget:.2f} this month.\n"
# 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()
if grant_balance > 0.0:
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'] == '*':
return True
if await self.is_admin(update):
if self.is_admin(update):
return True
allowed_user_ids = self.config['allowed_user_ids'].split(',')
@@ -504,7 +504,7 @@ class ChatGPT3TelegramBot:
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.
The first user in the user list is the admin.
@@ -526,7 +526,7 @@ class ChatGPT3TelegramBot:
if user_id not in self.usage:
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')
if self.config['monthly_user_budgets'] == '*':
@@ -557,7 +557,7 @@ class ChatGPT3TelegramBot:
if user_id not in self.usage:
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
if self.config['monthly_user_budgets'] == '*':