telegramBOT.py

import logging
import scanner
import os

from dotenv import load_dotenv
from datetime import datetime
from telegram.ext import CommandHandler, Application
from os import path

load_dotenv()
telegram_token = os.getenv('TELEGRAM_TOKEN')
if telegram_token is None:
    raise EnvironmentError('TELEGRAM_TOKEN is not set in .env file')

chat_ids = []

if not path.exists("chat_ids.txt"):
    with open("chat_ids.txt", 'x') as f:
        f.write('')
else:
    with open("chat_ids.txt") as f:
        lines = f.readlines()
        chat_ids = [line.rstrip() for line in lines]

async def time_check():
    global starting_hour_int
    now = datetime.now()
    now_hour_int = int(now.strftime("%H"))

    if now_hour_int > starting_hour_int or (starting_hour_int == 23 and now_hour_int < 1):
        starting_hour_int = -1

        return False
    else:
        remain_min = 60 - now.minute
        print('Starting at:', remain_min, 'minutes')

        return True


async def check_address(context):

    if await time_check() is False:

        missing_transactions = scanner.main(verbose=False)

        if missing_transactions is None:
            missing_transactions = []

        if len(missing_transactions) != 0:
            for tx in missing_transactions:
                str_tx = '---NEW TRANSACTION---\n\n'
                for key, value in tx.items():
                    print('\t', key, ' : ', value)
                    str_tx += key + ': ' + str(value)

                    if key != 'total cost ($)':
                        str_tx += '\n'

                if str_tx != '':
                    for chat_id in chat_ids:
                        context.bot.send_message(chat_id=chat_id, text=str_tx)
                print()


async def start(update, context):
    await update.message.reply_text("""Hello ! Welcome to WhaleTrackerBot. \n\nThis bot will check if the biggest bitcoin whale is adding or removing btc from its wallet.
       
* To start receiving new whales transactions click-type: /check 
    
* If you want to stop receiving messages from this bot click-type: /stop
    """)


async def stop(update, context):
    chat_id = update.message.chat_id
    chat_id = str(chat_id)

    if chat_id in chat_ids:
        with open("chat_ids.txt", "r") as f:
            lines = f.readlines()
        with open("chat_ids.txt", "w") as f:
            for line in lines:
                if line.strip("\n") != chat_id:
                    f.write(line)

        chat_ids.remove(chat_id)
        await update.message.reply_text("Stopped! You will no longer receiving new transaction of the whale")
        print(chat_ids)
    else:
        await update.message.reply_text("First you have to start receiving new whales transactions. Click-type: /check")


async def check(update, context):
    chat_id = update.message.chat_id
    chat_id = str(chat_id)

    if chat_id not in chat_ids:
        with open("chat_ids.txt", 'a') as file:
            file.write(chat_id + '\n')

        chat_ids.append(chat_id)
        await update.message.reply_text("Start monitoring...")
        print(chat_ids)
    else:
        await update.message.reply_text("You are already monitoring...")


async def admin_users(update, context):
    users_count = len(chat_ids)

    await update.message.reply_text(str(users_count) + " active user(s)")

def main():
    # logging.basicConfig(filename='logging_telegramBOT.log', level=logging.DEBUG,
                        # format='%(asctime)s %(levelname)s: %(message)s', )
    
    print("Starting Telegram Bot...")
    app = Application.builder().token(telegram_token).build()

    app.add_handler(CommandHandler('start', start))
    app.add_handler(CommandHandler('stop', stop))
    app.add_handler(CommandHandler('check', check))
    app.add_handler(CommandHandler('admin_users', admin_users))

    print('Polling...')
    app.run_polling(poll_interval=3)

    
if __name__ == '__main__':
    main()