• Как добавить обязательную подписку на канал для тг-бота?

    @Medovochka
    Попробуй этот код:

    spoiler
    import logging
    import os
    
    from dotenv import load_dotenv
    from telegram import Update
    from telegram.ext import CallbackContext, CommandHandler, Filters, MessageHandler, Updater
    
    from openai_helper import OpenAIHelper, default_max_tokens
    from telegram_bot import ChatGPTTelegramBot
    
    # Global variable to store subscribed user IDs
    subscribed_users = set()
    
    # Command handler for the /subscribe command
    def subscribe(update: Update, context: CallbackContext):
        user_id = update.message.from_user.id
        subscribed_users.add(user_id)
        update.message.reply_text('Вы успешно подписались на канал!')
    
    # Command handler for the /unsubscribe command
    def unsubscribe(update: Update, context: CallbackContext):
        user_id = update.message.from_user.id
        subscribed_users.discard(user_id)
        update.message.reply_text('Вы успешно отписались от канала!')
    
    # Message handler to check if the user is subscribed before processing messages
    def process_message(update: Update, context: CallbackContext):
        user_id = update.message.from_user.id
        if user_id in subscribed_users:
            # User is subscribed, process the message
            chat_gpt_bot.process_message(update, context)
        else:
            # User is not subscribed, send a reply asking to subscribe
            update.message.reply_text('Пожалуйста, подпишитесь на канал, чтобы воспользоваться ботом.')
    
    def main():
        # Read .env file
        load_dotenv()
    
        # Setup logging
        logging.basicConfig(
            format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
            level=logging.INFO
        )
    
        # Check if the required environment variables are set
        required_values = ['TELEGRAM_BOT_TOKEN', 'OPENAI_API_KEY']
        missing_values = [value for value in required_values if os.environ.get(value) is None]
        if len(missing_values) > 0:
            logging.error(f'The following environment values are missing in your .env: {", ".join(missing_values)}')
            exit(1)
    
        # Setup configurations
        model = os.environ.get('OPENAI_MODEL', 'gpt-3.5-turbo')
        max_tokens_default = default_max_tokens(model=model)
        openai_config = {
            'api_key': os.environ['OPENAI_API_KEY'],
            'show_usage': os.environ.get('SHOW_USAGE', 'false').lower() == 'true',
            'stream': os.environ.get('STREAM', 'true').lower() == 'true',
            'proxy': os.environ.get('PROXY', None),
            'max_history_size': int(os.environ.get('MAX_HISTORY_SIZE', 15)),
            'max_conversation_age_minutes': int(os.environ.get('MAX_CONVERSATION_AGE_MINUTES', 180)),
            'assistant_prompt': os.environ.get('ASSISTANT_PROMPT', 'You are a helpful assistant.'),
            'max_tokens': int(os.environ.get('MAX_TOKENS', max_tokens_default)),
            'n_choices': int(os.environ.get('N_CHOICES', 1)),
            'temperature': float(os.environ.get('TEMPERATURE', 1.0)),
            'image_size': os.environ.get('IMAGE_SIZE', '512x512'),
            'model': model,
            'presence_penalty': float(os.environ.get('PRESENCE_PENALTY', 0.0)),
            'frequency_penalty': float(os.environ.get('FREQUENCY_PENALTY', 0.0)),
            'bot_language': os.environ.get('BOT_LANGUAGE', 'en'),
        }
    
        # log deprecation warning for old budget variable names
        # old variables are caught in the telegram_config definition for now
        # remove support for old budget names at some point in the future
        if os.environ.get('MONTHLY_USER_BUDGETS') is not None:
            logging.warning('The environment variable MONTHLY_USER_BUDGETS is deprecated. '
                            'Please use USER_BUDGETS with BUDGET_PERIOD instead.')
        if os.environ.get('MONTHLY_GUEST_BUDGET') is not None:
            logging.warning('The environment variable MONTHLY_GUEST_BUDGET is deprecated. '
                            'Please use GUEST_BUDGET with BUDGET_PERIOD instead.')
    
        telegram_config = {
            'token': os.environ['TELEGRAM_BOT_TOKEN'],
            'admin_user_ids': os.environ.get('ADMIN_USER_IDS', '-'),
            'allowed_user_ids': os.environ.get('ALLOWED_TELEGRAM_USER_IDS', '*'),
            'enable_quoting': os.environ.get('ENABLE_QUOTING', 'true').lower() == 'true',
            'enable_image_generation': os.environ.get('ENABLE_IMAGE_GENERATION', 'true').lower() == 'true',
            'enable_transcription': os.environ.get('ENABLE_TRANSCRIPTION', 'true').lower() == 'true',
            'budget_period': os.environ.get('BUDGET_PERIOD', 'monthly').lower(),
            'user_budgets': os.environ.get('USER_BUDGETS', os.environ.get('MONTHLY_USER_BUDGETS', '*')),
            'guest_budget': float(os.environ.get('GUEST_BUDGET', os.environ.get('MONTHLY_GUEST_BUDGET', '100.0'))),
            'stream': os.environ.get('STREAM', 'true').lower() == 'true',
            'proxy': os.environ.get('PROXY', None),
            'voice_reply_transcript': os.environ.get('VOICE_REPLY_WITH_TRANSCRIPT_ONLY', 'false').lower() == 'true',
            'voice_reply_prompts': os.environ.get('VOICE_REPLY_PROMPTS', '').split(';'),
            'ignore_group_transcriptions': os.environ.get('IGNORE_GROUP_TRANSCRIPTIONS', 'true').lower() == 'true',
            'group_trigger_keyword': os.environ.get('GROUP_TRIGGER_KEYWORD', ''),
            'token_price': float(os.environ.get('TOKEN_PRICE', 0.002)),
            'image_prices': [float(i) for i in os.environ.get('IMAGE_PRICES', "0.016,0.018,0.02").split(",")],
            'transcription_price': float(os.environ.get('TOKEN_PRICE', 0.006)),
            'bot_language': os.environ.get('BOT_LANGUAGE', 'en'),
        }
    
        # Setup and run ChatGPT and Telegram bot
        openai_helper = OpenAIHelper(config=openai_config)
        telegram_bot = ChatGPTTelegramBot(config=telegram_config, openai=openai_helper)
    
        # Create an Updater and pass it the bot's token
        updater = Updater(token=telegram_config['token'], use_context=True)
    
        # Get the dispatcher to register handlers
        dispatcher = updater.dispatcher
    
        # Register the command handlers
        dispatcher.add_handler(CommandHandler("subscribe", subscribe))
        dispatcher.add_handler(CommandHandler("unsubscribe", unsubscribe))
    
        # Register the message handler
        dispatcher.add_handler(MessageHandler(Filters.text & ~Filters.command, process_message))
    
        # Start the bot
        updater.start_polling()
        updater.idle()
    
    
    if __name__ == '__main__':
        main()

    Ответ написан
    1 комментарий