mirror of
https://github.com/Brandon-Rozek/website.git
synced 2024-11-09 10:40:34 -05:00
Script to archive a mastodon toot
This commit is contained in:
parent
dab55fd1da
commit
9947478c6c
1 changed files with 63 additions and 0 deletions
63
archive_toot.py
Executable file
63
archive_toot.py
Executable file
|
@ -0,0 +1,63 @@
|
|||
#!/usr/bin/env python
|
||||
"""
|
||||
Saves a toot to reference as a shortcode
|
||||
in a post.
|
||||
"""
|
||||
from http.client import HTTPResponse
|
||||
from urllib.parse import urlparse
|
||||
from urllib import request
|
||||
from typing import Optional
|
||||
import argparse
|
||||
import json
|
||||
import sys
|
||||
|
||||
SAVE_DIR = "static/data/toots"
|
||||
|
||||
# Grab arguments
|
||||
parser = argparse.ArgumentParser(description="Save a toot")
|
||||
parser.add_argument("toot_url", type=str, help="URL to specific toot")
|
||||
args = vars(parser.parse_args())
|
||||
|
||||
parsed_url = urlparse(args['toot_url'])
|
||||
|
||||
# Parse Arguments
|
||||
SERVER = f"{parsed_url.scheme}://{parsed_url.netloc}"
|
||||
TOOT_ID = parsed_url.path.split("/")[-1]
|
||||
|
||||
# Query server
|
||||
url = SERVER + "/api/v1/statuses/" + TOOT_ID
|
||||
response: Optional[HTTPResponse] = None
|
||||
|
||||
try:
|
||||
response = request.urlopen(url)
|
||||
except Exception:
|
||||
print("Unable to grab toots from Mastodon.")
|
||||
if response is None:
|
||||
sys.exit(-1)
|
||||
|
||||
# Parse server response
|
||||
server_data: Optional[dict] = None
|
||||
try:
|
||||
server_data = json.loads(response.read())
|
||||
except Exception:
|
||||
print("Malformed JSON response from server.")
|
||||
|
||||
if server_data is None:
|
||||
sys.exit(-1)
|
||||
|
||||
if not isinstance(server_data, dict):
|
||||
print("Unexpected JSON response, should be of form dict.")
|
||||
sys.exit(-1)
|
||||
|
||||
# Save toot
|
||||
TOOT_REFERENCE = f"{parsed_url.netloc.replace('.', '-')}-{TOOT_ID}"
|
||||
TOOT_SAVE_FILE = f"{SAVE_DIR}/{TOOT_REFERENCE}.json"
|
||||
|
||||
try:
|
||||
with open(TOOT_SAVE_FILE, "w", encoding="UTF-8") as f:
|
||||
json.dump(server_data, f)
|
||||
except:
|
||||
print("Unable to write to save location.")
|
||||
sys.exit(-1)
|
||||
|
||||
print("Saved Toot Reference", TOOT_REFERENCE)
|
Loading…
Reference in a new issue