alberanid/fortunes-spam
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

get the server from the environment

Browse source
This commit is contained in:
Davide Alberani 2024-12-05 22:02:39 +01:00
parent 8b3275ea0e
commit b9aea57c10
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
mastodon-bot/fortunes-spam-bot.py
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'])