pfxposter/pfxposter/pfxposter.py

49 lines
1.3 KiB
Python

#!/usr/bin/env python3
from mastodon import Mastodon
import requests
import atoma
import re
import tempfile
import toml
import os
os_user = os.getenv("USER")
config_file = "/home/{}/.pfxposter".format(os_user)
config = toml.load(config_file)
username = config['pixelfed']['username']
last_updated = config['updated']
mastodon = Mastodon(
access_token = config['mastodon']['access_token'],
api_base_url = config['mastodon']['mastodon_url']
)
def main():
pixelfeed_get = requests.get('https://pixelfed.social/users/{}.atom'.format(username))
pixelfeed = atoma.parse_atom_bytes(pixelfeed_get.content)
latest_post = pixelfeed.entries[0]
last_updated_atom = latest_post.updated
if last_updated == last_updated_atom:
print("Up-to-date")
os._exit(0)
config['updated'] = last_updated_atom
with open(config_file, "w") as f:
toml.dump(config, f)
print("Config file updated")
image_name = latest_post.title.value
image_url = re.search("(?P<url>https?://[^\s]+)", latest_post.summary.value).group("url").rstrip('">').replace("_thumb", "")
tmp = tempfile.NamedTemporaryFile(suffix=".jpg")
get_image = requests.get(image_url)
tmp.write(get_image.content)
mastodon_media = mastodon.media_post(tmp.name)
#mastodon.status_post(image_name, media_ids=mastodon_media['id'])
print("Status posted: ", image_name)
tmp.close()
if __name__ == "__main__":
main()