Add default post formats for each integration (#26)

bot/domain.py → bot/domain/__init__.py

@@ -1,19 +1,15 @@
+# TODO: Refactor
 import datetime
 import io
 import typing
 from dataclasses import dataclass
 
 from bot import constants
 from bot.common import utils
+from bot.domain import post_format
 
 
-DEFAULT_POST_FORMAT = """🔗 URL: {url}
-🧑🏻‍🎨 Author: {author}
-📅 Created: {created}
-👀 Views: {views}
-👍🏻 Likes: {likes} 👎🏻 Dislikes: {dislikes}
-📕 Description: {description}\n
-"""
+DEFAULT_POST_FORMAT = post_format.DEFAULT_POST_FORMAT
 
 DEFAULT_COMMENT_FORMAT = """🧑🏻‍🎨 Author: {author}
 📅 Created: {created}

bot/domain/post_format.py

@@ -0,0 +1,91 @@
+import typing
+
+from bot import constants
+
+
+DEFAULT_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👀 Views: {views}
+👍🏻 Likes: {likes} 👎🏻 Dislikes: {dislikes}
+📕 Description: {description}\n
+"""
+
+FOUR_CHAN_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+📕 Description: {description}\n
+"""
+
+INSTAGRAM_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👍🏻 Likes: {likes}\n
+"""
+
+REDDIT_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👍🏻 Likes: {likes} 👎🏻 Dislikes: {dislikes}
+📕 Description: {description}\n
+"""
+
+THREADS_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👍🏻 Likes: {likes}
+📕 Description: {description}\n
+"""
+
+TIKTOK_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👀 Views: {views}
+👍🏻 Likes: {likes}\n
+"""
+
+TWENTY4_UR_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+📕 Description: {description}\n
+"""
+
+TWITCH_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👀 Views: {views}
+📕 Description: {description}\n
+"""
+
+TWITTER_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👀 Views: {views}
+👍🏻 Likes: {likes}
+📕 Description: {description}\n
+"""
+
+YOUTUBE_POST_FORMAT = """🔗 URL: {url}
+🧑🏻‍🎨 Author: {author}
+📅 Created: {created}
+👀 Views: {views}
+👍🏻 Likes: {likes} 👎🏻 Dislikes: {dislikes}\n
+"""
+
+
+DEFAULT_INTEGRATION_POST_FMT_MAPPING = {
+    constants.Integration.FACEBOOK: DEFAULT_POST_FORMAT,
+    constants.Integration.FOUR_CHAN: FOUR_CHAN_POST_FORMAT,
+    constants.Integration.INSTAGRAM: INSTAGRAM_POST_FORMAT,
+    constants.Integration.REDDIT: REDDIT_POST_FORMAT,
+    constants.Integration.THREADS: THREADS_POST_FORMAT,
+    constants.Integration.TIKTOK: TIKTOK_POST_FORMAT,
+    constants.Integration.TWENTY4_UR: TWENTY4_UR_POST_FORMAT,
+    constants.Integration.TWITCH: TWITCH_POST_FORMAT,
+    constants.Integration.TWITTER: TWITTER_POST_FORMAT,
+    constants.Integration.YOUTUBE: YOUTUBE_POST_FORMAT,
+}
+
+
+def get_or_default(integration: constants.Integration, default: typing.Optional[str]) -> str:
+    return DEFAULT_INTEGRATION_POST_FMT_MAPPING.get(integration, default or DEFAULT_POST_FORMAT)

bot/integrations/base.py

@@ -46,6 +46,10 @@ class BaseClientSingleton:
     _INSTANCE: typing.Optional[typing.Union[BaseClient, int]] = None
     _CONFIG_CLASS = BaseClientConfig
 
+    @classmethod
+    def should_handle(cls, url: str) -> bool:
+        return any(domain in url for domain in cls.DOMAINS)
+
     @classmethod
     def _create_instance(cls) -> None:
         raise NotImplementedError()

bot/integrations/registry.py

@@ -29,7 +29,7 @@
 
 def should_handle(url: str) -> bool:
     for klass in CLASSES:
-        if any(domain in url for domain in klass.DOMAINS):
+        if klass.should_handle(url):
             return klass.get_instance() is not None
 
     return False

bot/integrations/twitch/client.py

@@ -32,6 +32,10 @@ def _create_instance(cls) -> None:
 
         cls._INSTANCE = TwitchClient()
 
+    @classmethod
+    def should_handle(cls, url: str) -> bool:
+        return super().should_handle(url) and '/clip/' in url
+
 
 class TwitchClient(base.BaseClient):
     INTEGRATION = constants.Integration.TWITCH

bot/service.py

@@ -8,6 +8,7 @@
 from bot import logger
 from bot import models
 from bot import repository
+from bot.domain import post_format
 from bot.integrations import registry
 
 
@@ -130,7 +131,12 @@ async def get_post(  # noqa: C901
         logger.debug('Post already in DB, not downloading again...', url=url)
 
     # Set formatting
-    post.set_format(server.integrations[integration].post_format)
+    post.set_format(
+        post_format.get_or_default(
+            integration=integration,
+            default=server.integrations[integration].post_format,
+        )
+    )
 
     repository.save_server_post(
         server_vendor=server_vendor,