116 lines
3.4 KiB
Python
116 lines
3.4 KiB
Python
from telegram.ext import Updater, CommandHandler, MessageHandler, Filters
|
|
from telegram import ParseMode
|
|
from block_io import BlockIo, BlockIoAPIError
|
|
import io
|
|
|
|
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']
|
|
|
|
# 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'])
|
|
|
|
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) > amount:
|
|
address_receiver = get_address(receiver)
|
|
return block_io.withdraw_from_labels(amounts=amount, from_labels=sender, to_labels=receiver)
|
|
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']
|
|
|
|
bot.send_message(chat_id=update.message.chat_id, parse_mode=ParseMode.MARKDOWN, text="Transaction effectuée !\n [tx](https://chain.so/tx/DOGETEST/" + txid + ")")
|
|
|
|
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 = 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" + str(balance) + " DOGE")
|
|
|
|
# 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)
|
|
|
|
updater.start_polling()
|