153 lines
4.9 KiB
Python
153 lines
4.9 KiB
Python
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
|
from telegram import ParseMode
|
|
from block_io import BlockIo, BlockIoAPIError
|
|
import logging
|
|
import os
|
|
|
|
BLOCK_IO_API_KEY = os.environ['BLOCK_IO_API_KEY']
|
|
BLOCK_IO_PIN = os.environ['BLOCK_IO_PIN']
|
|
TELEGRAM_API_KEY = os.environ['TELEGRAM_API_KEY']
|
|
NETWORK = os.environ['NETWORK']
|
|
|
|
# Logging
|
|
|
|
logging.basicConfig(level=logging.ERROR,
|
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
|
|
# Exceptions
|
|
|
|
class NoAccountError(Exception):
|
|
pass
|
|
|
|
class BalanceError(Exception):
|
|
pass
|
|
|
|
class AccountExisting(Exception):
|
|
pass
|
|
|
|
# BlockIO
|
|
|
|
version = 2
|
|
block_io = BlockIo(BLOCK_IO_API_KEY, BLOCK_IO_PIN, version)
|
|
|
|
# Core functions
|
|
|
|
def get_balance(account):
|
|
try:
|
|
response = block_io.get_address_by(label=account)
|
|
except BlockIoAPIError:
|
|
raise NoAccountError(account)
|
|
else:
|
|
return (float(response['data']['available_balance']), float(response['data']['pending_received_balance']))
|
|
|
|
def create_address(account):
|
|
try:
|
|
response = block_io.get_new_address(label=account)
|
|
except BlockIoAPIError:
|
|
raise AccountExisting
|
|
else:
|
|
return response['data']['address']
|
|
|
|
def get_address(account):
|
|
try:
|
|
response = block_io.get_address_by(label=account)
|
|
except BlockIoAPIError:
|
|
raise NoAccountError(account)
|
|
else:
|
|
return response['data']['address']
|
|
|
|
def transaction(sender, receiver, amount):
|
|
try:
|
|
if get_balance(sender)[0] > amount:
|
|
address_receiver = get_address(receiver)
|
|
return block_io.withdraw_from_labels(amounts=amount, from_labels=sender, to_labels=receiver, priority="low")
|
|
else:
|
|
return "Pas assez de doge"
|
|
except NoAccountError as e:
|
|
return "Merci de vous créer un compte @" + str(e)
|
|
except BlockIoAPIError:
|
|
return "Erreur d'API"
|
|
|
|
def address_transaction(account, address, amount):
|
|
try:
|
|
if get_balance(account)[0] > amount:
|
|
return block_io.withdraw_from_labels(amounts=amount, from_labels=account, to_addresses=address, priority="low")
|
|
else:
|
|
return "Pas assez de doge"
|
|
except NoAccountError as e:
|
|
return "Merci de vous créer un compte @" + str(e)
|
|
except BlockIoAPIError:
|
|
return "Erreur d'API"
|
|
|
|
# Telegram functions
|
|
|
|
def start(bot, update):
|
|
bot.send_message(chat_id=update.message.chat_id, text="I'm a bot, please talk to me!")
|
|
|
|
def dogetip(bot, update, args):
|
|
montant = int(args[0])
|
|
unit = args[1]
|
|
destinataire = args[2][1:]
|
|
|
|
if unit == "doge":
|
|
response = transaction(update.message.from_user.username, destinataire, montant)
|
|
|
|
txid = response['data']['txid']
|
|
|
|
message = "🚀 Transaction effectuée 🚀\n\n" \
|
|
+ str(montant) + " " + NETWORK + "\n" \
|
|
+ update.message.from_user.username + " → " + destinataire + "\n\n" \
|
|
+ "[Voir la transaction](https://chain.so/tx/" + NETWORK + "/" + txid + ")"
|
|
|
|
bot.send_message(chat_id=update.message.chat_id, parse_mode=ParseMode.MARKDOWN, text=message)
|
|
|
|
def register(bot, update):
|
|
try:
|
|
address = create_address(update.message.from_user.username)
|
|
except AccountExisting:
|
|
bot.send_message(chat_id=update.message.chat_id, text="Vous avez déjà un compte")
|
|
else:
|
|
bot.send_message(chat_id=update.message.chat_id, text=address)
|
|
|
|
def infos(bot, update):
|
|
try:
|
|
address = get_address(update.message.from_user.username)
|
|
balance, unconfirmed_balance = get_balance(update.message.from_user.username)
|
|
except NoAccountError as e:
|
|
bot.send_message(chat_id=update.message.chat_id, text="Merci de vous créer un compte @" + str(e))
|
|
else:
|
|
bot.send_message(chat_id=update.message.chat_id, text=address + "\n\n" + str(balance) + " " + NETWORK + "\n" + str(unconfirmed_balance) + " " + NETWORK + " unconfirmed")
|
|
|
|
def withdraw(bot, update, args):
|
|
montant = int(args[0])
|
|
unit = args[1]
|
|
address = args[2]
|
|
|
|
if unit == "doge":
|
|
response = address_transaction(update.message.from_user.username, address, montant)
|
|
|
|
txid = response['data']['txid']
|
|
|
|
bot.send_message(chat_id=update.message.chat_id, parse_mode=ParseMode.MARKDOWN, text="Transaction effectuée !\n [tx](https://chain.so/tx/" + NETWORK + "/" + txid + ")")
|
|
|
|
# Telegram initialisation
|
|
|
|
updater = Updater(token=TELEGRAM_API_KEY)
|
|
dispatcher = updater.dispatcher
|
|
|
|
start_handler = CommandHandler('start', start)
|
|
dispatcher.add_handler(start_handler)
|
|
|
|
dogetip_handler = CommandHandler('dogetip', dogetip, pass_args=True)
|
|
dispatcher.add_handler(dogetip_handler)
|
|
|
|
register_handler = CommandHandler('register', register)
|
|
dispatcher.add_handler(register_handler)
|
|
|
|
infos_handler = CommandHandler('infos', infos)
|
|
dispatcher.add_handler(infos_handler)
|
|
|
|
withdraw_handler = CommandHandler('withdraw', withdraw, pass_args=True)
|
|
dispatcher.add_handler(withdraw_handler)
|
|
updater.start_polling()
|