Compare commits

...

3 commits
master ... main

5 changed files with 25 additions and 25 deletions

View file

@ -41,7 +41,7 @@ You can use the content of the *mastodon-bot* directory to create your own Masto
# License
Copyright 2003-2021 Davide Alberani <da@erlug.linux.it>
Copyright 2003-2021 Davide Alberani <da@mimante.net>
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.

View file

@ -1,10 +1,10 @@
FROM alpine
LABEL \
maintainer="Davide Alberani <da@erlug.linux.it>"
maintainer="Davide Alberani <da@mimante.net>"
RUN \
apk add --no-cache git fortune python3 py3-cffi py3-six py3-requests py3-cryptography && \
pip3 install Mastodon.py && \
apk add --no-cache git fortune python3 py3-pip py3-cffi py3-six py3-requests py3-cryptography && \
pip3 install --break-system-packages Mastodon.py && \
cd / && \
git clone https://github.com/alberanid/fortunes-spam.git
COPY fortunes-spam-bot.py /fortunes-spam

View file

@ -6,9 +6,6 @@ import sys
import subprocess
from mastodon import Mastodon
API_URL = 'https://botsin.space/'
def getSpam(sections=('spam-o', 'spam-ita-o')):
cmd = ['fortune', '-o', *sections]
process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
@ -26,10 +23,10 @@ def getSpam(sections=('spam-o', 'spam-ita-o')):
return stdout
def serve(token):
def serve(api_url, token):
spam = getSpam()
print('serving:\n%s' % spam)
mastodon = Mastodon(access_token=token, api_base_url=API_URL)
mastodon = Mastodon(access_token=token, api_base_url=api_url)
mastodon.status_post(spam, sensitive=True, spoiler_text='NSFW')
@ -37,4 +34,7 @@ if __name__ == '__main__':
if 'SPAMBOT_TOKEN' not in os.environ:
print("Please specify the Mastodon token in the SPAMBOT_TOKEN environment variable")
sys.exit(1)
serve(token=os.environ['SPAMBOT_TOKEN'])
if 'API_URL' not in os.environ:
print("Please specify the Mastodon server in the API_URL environment variable")
sys.exit(1)
serve(api_url=os.environ['API_URL'], token=os.environ['SPAMBOT_TOKEN'])

View file

@ -1,10 +1,10 @@
FROM alpine
LABEL \
maintainer="Davide Alberani <da@erlug.linux.it>"
maintainer="Davide Alberani <da@mimante.net>"
RUN \
apk add --no-cache git fortune python3 py3-cffi py3-six py3-requests py3-cryptography py3-pip && \
pip3 install python-telegram-bot && \
pip3 install --break-system-packages python-telegram-bot && \
cd / && \
git clone https://github.com/alberanid/fortunes-spam.git
COPY fortunes-spam-bot.py /fortunes-spam

View file

@ -4,13 +4,14 @@
Build it with: docker build -t fortunes-spam-bot .
Run it with something like: docker run -ti --rm -e SPAMBOT_TOKEN=your-telegram-token fortunes-spam-bot
Copyright 2018-2021 Davide Alberani <da@erlug.linux.it> Apache 2.0 license
Copyright 2018-2025 Davide Alberani <da@mimante.net> Apache 2.0 license
"""
import os
import logging
import subprocess
from telegram.ext import Updater, CommandHandler
from telegram import Update
from telegram.ext import Application, CommandHandler
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
@ -29,7 +30,7 @@ def getSpam(section):
try:
stdout = stdout.strip()
stdout = stdout.decode('utf8')
except:
except Exception:
return 'uh-oh: something was wrong with the encoding of the can\'s label; try again'
if process.returncode != 0:
return 'something terrible is happening: exit code: %s, stderr: %s' % (
@ -39,10 +40,10 @@ def getSpam(section):
return stdout
def serve(section, update, context):
async def serve(section, update, context):
spam = getSpam(section)
logging.info('%s wants some spam; serving:\n%s' % (update.effective_user.username, spam))
update.message.reply_text(spam)
await update.message.reply_text(spam)
def en(update, context):
@ -53,18 +54,17 @@ def it(update, context):
return serve('spam-ita-o', update, context)
def about(update, context):
async def about(update, context):
logging.info('%s required more info' % update.effective_user.username)
update.message.reply_text('See https://github.com/alberanid/fortunes-spam')
await update.message.reply_text('See https://github.com/alberanid/fortunes-spam')
if __name__ == '__main__':
if 'SPAMBOT_TOKEN' not in os.environ:
print("Please specify the Telegram token in the SPAMBOT_TOKEN environment variable")
logging.info('start serving delicious spam')
updater = Updater(os.environ['SPAMBOT_TOKEN'])
updater.dispatcher.add_handler(CommandHandler('en', en))
updater.dispatcher.add_handler(CommandHandler('it', it))
updater.dispatcher.add_handler(CommandHandler('about', about))
updater.start_polling()
updater.idle()
application = Application.builder().token(os.environ['SPAMBOT_TOKEN']).build()
application.add_handler(CommandHandler('en', en))
application.add_handler(CommandHandler('it', it))
application.add_handler(CommandHandler('about', about))
application.run_polling(allowed_updates=Update.ALL_TYPES)