diff options
| -rw-r--r-- | toot_me.py | 189 |
1 files changed, 189 insertions, 0 deletions
diff --git a/toot_me.py b/toot_me.py new file mode 100644 index 0000000..84f9344 --- /dev/null +++ b/toot_me.py @@ -0,0 +1,189 @@ +from mastodon import Mastodon +from mastodon.errors import MastodonAPIError +import requests +import re +import os + +mastodon = Mastodon( + access_token=os.environ.get("MASTODON_SECRET"), + api_base_url="https://mastodon.social" + ) + +url_base = "https://web.navan.dev" +sample_markdown_file = "Content/posts/2022-12-25-blog-to-toot.md" + +tags = [] +toots = [] +image_idx = 0 +markdown_image = r'(?:!\[(.*?)\]\((.*?)\))' +markdown_links = r'(?:\[(.*?)\]\((.*?)\))' + +def get_image(code, language: str = "python", title: str = "Code Snippet"): + params = ( + ('code', code), + ('language', language), + ('title', title), + ) + + response = requests.get('http://localhost:3000/api', params=params) + + return response.content + +class TootContent: + def __init__(self, text: str = ""): + self.text = text + self.images = [] + self.links = [] + self.image_count = len(images) + + def __str__(self): + toot_text = self.text + for link in self.links: + toot_text += " " + link + return toot_text + + def get_text(self): + toot_text = self.text + for link in self.links: + toot_text += " " + link + return toot_text + + def get_length(self): + length = len(self.text) + for link in self.links: + length += 23 + return length + + def add_link(self, link): + if len(self.text) + 23 < 498: + if link[0].lower() != 'h': + link = url_base + link + self.links.append(link) + return True + return False + + def add_image(self, image): + + if len(self.images) == 4: + # will handle in future + print("cannot upload more than 4 images per toot") + exit(1) + # upload image and get id + self.images.append(image) + self.image_count = len(self.images) + + def add_text(self, text): + if len(self.text + text) > 400: + return False + else: + self.text += f" {text}" + return True + + def get_links(self): + print(len(self.links)) + + +in_metadata = False +in_code_block = False + +my_toots = [] +text = "" +images = [] +image_links = [] +extra_links = [] +tags = [] + +code_block = "" +language = "bash" + +current_toot = TootContent() + +metadata_regex = r"---\s*\n(.*?)\n---\s*\n" + + +with open(sample_markdown_file) as f: + markdown_content = f.read() + + +metadata = re.search(metadata_regex, markdown_content, re.DOTALL) +if metadata: + tags_match = re.search(r"tags: ([\w,\s]+)", metadata.group(1)) + if tags_match: + tags = tags_match.group(1).split(",") + + +markdown_content = markdown_content.rsplit("---\n",1)[-1].strip() + +for line in markdown_content.split("\n"): + if current_toot.get_length() < 400: + if line.strip() == '': + continue + if line[0] == '#': + line = line.replace("#","".strip()) + if len(my_toots) == 0: + current_toot.add_text( + f"{line}: a cross-posted blog post \n" + ) + hashtags = "" + for tag in tags: + hashtags += f"#{tag.strip()}," + current_toot.add_text(hashtags[:-1]) + my_toots.append(current_toot) + current_toot = TootContent() + else: + my_toots.append(current_toot) + current_toot = TootContent(text=f"{line}:") + continue + else: + if "```" in line: + in_code_block = not in_code_block + if in_code_block: + language = line.strip().replace("```",'') + continue + else: + with open(f"code-snipped_{image_idx}.png","wb") as f: + f.write(get_image(code_block, language)) + current_toot.add_image(f"code-snipped_{image_idx}.png") + image_idx += 1 + code_block = "" + continue + if in_code_block: + line = line.replace(" ","\t") + code_block += line + "\n" + continue + if len(re.findall(markdown_image,line)) > 0: + for image_link in re.findall(markdown_links, line): + image_link.append(image_link[1]) + # not handled yet + line = re.sub(markdown_image,"",line) + if len(re.findall(markdown_links,line)) > 0: + for link in re.findall(markdown_links, line): + if not (current_toot.add_link(link[1])): + extra_links.append(link[1]) + line = line.replace(f'[{link[0]}]({link[1]})',link[0]) + if not current_toot.add_text(line): + my_toots.append(current_toot) + current_toot = TootContent(line) + else: + my_toots.append(current_toot) + current_toot = TootContent() + +my_toots.append(current_toot) + +in_reply_to_id = None +for toot in my_toots: + image_ids = [] + for image in toot.images: + print(f"uploading image, {image}") + try: + image_id = mastodon.media_post(image) + image_ids.append(image_id.id) + except MastodonAPIError: + print("failed to upload. Continuing...") + if image_ids == []: + image_ids = None + + in_reply_to_id = mastodon.status_post( + toot.get_text(), in_reply_to_id=in_reply_to_id, media_ids=image_ids + ).id + |