functions.py

import datetime
import time
import requests
from pymongo import MongoClient
from telegram import ReplyKeyboardMarkup
from telegram._bot import Bot
from binance import Client

time_fun = datetime.datetime
bscscan_api_key = 'ADNU4CKRS8PNSHKZYSF1PQVIYEPYDNIB23'

admin_list = ["pugalkmc", "sarankmc", "groovemark", "Riyanmark799", "Dustin_05"]

api_key = '0rILNUx04etsxLKcM5ydIDepAkaGdQ9O6GyNuVvYkbSgQ56RVXDZH7Xz3dZKVBA6'
api_secret = 'CgNIsKOVjun0WFAcuCSWSlWi9mD7fue0xOkR8r3OctFGa5occMkDpALdR5Gt1anB'
client = Client(api_key, api_secret)

mongo_client = MongoClient("mongodb+srv://marketersgroove:pugalsaran143@cluster0.hnxcrow.mongodb.net/")
db = mongo_client["groovedb"]
peoples_col = db["peoples"]
tasks_col = db["tasks"]

BOT_TOKEN = "6109952194:AAHVz5BIJUNamBUVioTSjJA4a9C3qezTeWs"
bot = Bot(token=BOT_TOKEN)


async def menu_button(update, context):
    message = update.message
    if message.chat.type == 'group' or message.chat.type == 'supergroup':
        return
    chat_id = update.message.chat_id
    username = update.message.from_user.username
    reply_keyboard = []
    if username in admin_list:
        reply_keyboard = [["settings", "My task"], ["Admin Mode"]]
    else:
        reply_keyboard = [["settings", "My task"]]
    await context.bot.send_message(chat_id=chat_id, text="Choose options",
                                   reply_markup=ReplyKeyboardMarkup(reply_keyboard, resize_keyboard=True,
                                                                    one_time_keyboard=True),
                                   reply_to_message_id=update.message.message_id)


async def settings(update, context):
    message = update.message
    if message.chat.type == 'group' or message.chat.type == 'supergroup':
        return
    chat_id = update.message.chat_id
    text = update.message.text
    reply_keyboard = [["Twitter", "Discord", "Binance"], ["TUSD Address", 'UPI ID', 'main menu']]
    await context.bot.send_message(chat_id=chat_id, text="Choose options",
                                   reply_markup=ReplyKeyboardMarkup(reply_keyboard, resize_keyboard=True,
                                                                    one_time_keyboard=True),
                                   reply_to_message_id=update.message.message_id)


async def distribute_payment(update, context, dict_payment, total_cost):
    chat_id = update.message.chat_id
    tusd_balance = client.get_asset_balance(asset='TUSD')
    balance = float(tusd_balance['free'])
    if balance <= total_cost:
        await bot.send_message(chat_id=chat_id, text=f"Available balance on binance: {balance} TUSD\n"
                                                     f"Distribution cancelled")
        return
    for user in dict_payment:
        if 'binance' in user:
            await binance_pay(user['chat_id'], user['address'], user['amount'], chat_id)
            time.sleep(1)
    await bot.send_message(chat_id=chat_id, text="Distribution completed!")


async def binance_pay(chat_id, address, amount, dev_id=None):
    response = client.withdraw(
        coin='TUSD',
        address=address,
        amount=amount,
        network='BSC')

    transaction_details = client.get_withdraw_history(asset='TUSD', txId=response['id'])
    transaction_hash = transaction_details['txId']
    hash_count = 0
    while hash_count < 5:
        hash_count = check_transaction_confirmation(transaction_hash)
        time.sleep(1)

    transaction_url = f"https://bscscan.com/tx/{transaction_hash}"
    await bot.send_message(chat_id=chat_id, text="Your reward for the work has been sent to your binance account\n\n"
                                                 "Assest: TUSD\n"
                                                 f"Amount: {amount}\n"
                                                 f"Hash confirmation: {hash_count}\n"
                                                 f"Check transaction: <a href={transaction_url}>DETAIL</a>")


def check_transaction_confirmation(transaction_hash):
    api_url = f'https://api.bscscan.com/api?module=transaction&action=gettxreceiptstatus&txhash={transaction_hash}&apikey={bscscan_api_key}'
    response = requests.get(api_url)
    transaction_data = response.json()
    return transaction_data['result']['status']