Telegram bot integrate with Chapa Payment API(Python)

python -m venv env # or use virtualenv like below if u have it
# virtualenv env
source env/bin/activate # activate the environment

pip install python-telegram-bot chapa python-dotenv

TOKEN="YOUR BOT TOKEN FROM BOT FATHER"

import os
import logging # use logging rather than print.

from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
from telegram import InlineKeyboardButton, InlineKeyboardMarkup

# to export the environment variable
from dotenv import load_dotenv
load_dotenv()

# you can obviously edit the logging parameter as you want
# i personally like this pattern. If you don't know what i am saying just ignore it or read more about logging.
logging.basicConfig(
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s', level=logging.INFO
)

TOKEN = os.getenv("TOKEN")


def add_buy_button(update, context):
    '''
    This will add a button to any post in the channel

    Args:
        update: telegram.Update object
        context: telegram.Context object

    Return:
        None
    '''
    bot = context.bot.get_me()
    post = update.channel_post
    url = 'https://t.me/{bot.username}/?start={post.message_id}'

    keyboard = [[
        InlineKeyboardButton('Buy', url=url)
    ]]
    reply_markup = InlineKeyboardMarkup(keyboard)

    context.bot.edit_message_reply_markup(
        chat_id=post.chat.id,
        message_id=post.message_id,
        reply_markup=reply_markup
    )

    return


def main():
    '''
    This is main method to control the bot events and add the handlers
    '''
    updater = Updater(TOKEN)
    dispatcher = updater.dispatcher

    dispatcher.add_handler(MessageHandler(
        Filters.chat_type.channel, add_buy_button)
    )

    updater.start_polling()

    updater.idle()

if __name__ == '__main__':
    main()

TOKEN="..."
CHAPA_SECRET_KEY="...."

# update the telegram import
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters, CallbackContext
from telegram import InlineKeyboardButton, InlineKeyboardMarkup, WebAppInfo

# add this imports
import random
import string
from chapa import Chapa

CHAPA_SECRET_KEY = os.getenv('CHAPA_SECRET_KEY')
chapa = Chapa(CHAPA_SECRET_KEY, response_format='obj')

CHANNEL_ID = "ID we copied earlier (negative value)"

# new method
def create_chapa_checkout(data):
    '''
    Create chapa checkout url and return the result.

    Args:
        data(dict): data to send to chapa api to generate checkout

    Return:
        data(dict): api response
    '''
    response = chapa.initialize(**data)
    if response.status != 'success':
        return None

    return response.data

def start(update, context):
    '''
    Handle any /start message to bot.

    Args:
        update: telegram.Update object
        context: telegram.Context object

    Return:
        None
    '''
    if update.message.text != '/start':
        cmd = context.args[0] # to identify if the user comes from the channel or not.
    else:
        cmd = None

    bot = context.bot
    bot_username = bot.get_me().username
    user = update.effective_user

    # if the user is coming from the channel not from other place
    if cmd and not cmd.startswith('tx_'):
        try:
            # generate transaction number, random email and random domain. hope you will use the correct value here
            random_tx = "".join(random.choices(string.ascii_letters, k=5))
            random_email = "".join(random.choices(string.ascii_letters, k=5))
            random_domain = "".join(random.choices(string.ascii_letters, k=5))

            data = {
                'email': f"{random_email}@{random_domain}.com",
                'amount': 1000,
                'tx_ref': random_tx,
                'first_name': user.first_name,
                'last_name': user.last_name or ' None',
                'callback_url': f'https://t.me/{bot_username}?start=tx_{random_tx}',
                'customization': {
                    'title': 'Amazing Bot',
                    'description': 'This is a test project for chapa python sdk'
                }
            }

            checkout = create_chapa_checkout(data)
            bot.copy_message(
                chat_id=update.effective_user.id,
                from_chat_id=CHANNEL_ID,
                message_id=cmd
            )

            text = 'You want to buy this item? \n Click the button below to continue'

            keyboard = [[
                InlineKeyboardButton(
                    "Chapa Pay", web_app=WebAppInfo(checkout.checkout_url))
            ]]

            reply_markup = InlineKeyboardMarkup(keyboard)

            bot.send_message(
                chat_id=update.effective_user.id,
                text=text,
                reply_markup=reply_markup
            )

            context.user_data[random_tx] = 'PENDING'

            return

        except Exception as e:
            logger.error('Error in Generating Checkout', e)

    elif cmd and cmd.startswith('tx_'):
        if context.user_data.get(cmd[3:]) == 'PENDING':
            response = chapa.verify(cmd[3:])
            if response.status == 'success':
                context.user_data[cmd[3:]] = 'SUCCESS'
                bot.send_message(
                    chat_id=update.effective_user.id,
                    text='Your payment was successful'
                )
                return

            else:
                context.user_data[cmd[3:]] = 'FAILED'
                bot.send_message(
                    chat_id=update.effective_user.id,
                    text='Your payment failed'
                )
                return

    update.message.reply_text('Well come to Chapa Shopping! \n\nuse this channel @chapashoping or https://t.me/chapashoping')