Premier commit

This commit is contained in:
nyanloutre 2017-10-05 09:17:02 +02:00
commit c7310c1c60
Signed by: nyanloutre
GPG Key ID: F85D93686A3A9063
4 changed files with 133 additions and 0 deletions

8
Dockerfile Normal file
View File

@ -0,0 +1,8 @@
FROM python:alpine
RUN apk add --no-cache gcc g++ make libffi-dev openssl-dev && \
pip install python-telegram-bot requests block-io
COPY run.py /
CMD ["python", "run.py"]

7
Makefile Normal file
View File

@ -0,0 +1,7 @@
.PHONY: build
build:
docker build -t dogetipbot .
.PHONY: run
run: build
docker run --rm --name dogetipbot --env-file api_keys.env dogetipbot

3
api_keys.env Normal file
View File

@ -0,0 +1,3 @@
BLOCK_IO_API_KEY=123455678
BLOCK_IO_PIN=12345678
TELEGRAM_API_KEY=12345678

115
run.py Normal file
View File

@ -0,0 +1,115 @@
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()