Premier commit
This commit is contained in:
commit
c7310c1c60
8
Dockerfile
Normal file
8
Dockerfile
Normal 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
7
Makefile
Normal 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
3
api_keys.env
Normal file
@ -0,0 +1,3 @@
|
||||
BLOCK_IO_API_KEY=123455678
|
||||
BLOCK_IO_PIN=12345678
|
||||
TELEGRAM_API_KEY=12345678
|
115
run.py
Normal file
115
run.py
Normal 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()
|
Loading…
Reference in New Issue
Block a user