Initial commit

site/social/tasks.py

@@ -0,0 +1,59 @@
+from django.tasks import task
+
+from social.connectors.linkedin import LinkedInConnector
+from social.connectors.meta import MetaConnector
+from social.models import Platform, SocialPost, SocialPostTarget
+
+
+def _connector_for(platform: str):
+    if platform in {Platform.FACEBOOK, Platform.INSTAGRAM}:
+        return MetaConnector()
+    if platform == Platform.LINKEDIN:
+        return LinkedInConnector()
+    raise ValueError(f"Unknown platform: {platform}")
+
+
+@task
+def publish_social_post(post_id: str) -> None:
+    try:
+        post = SocialPost.objects.prefetch_related("targets__account").get(pk=post_id)
+    except SocialPost.DoesNotExist:
+        return
+
+    post.status = SocialPost.Status.PUBLISHING
+    post.save(update_fields=["status", "updated_at"])
+
+    any_ok = False
+    for target in post.targets.all():
+        try:
+            remote_id = _connector_for(target.platform).publish(post, target)
+            target.status = SocialPostTarget.Status.PUBLISHED
+            target.remote_id = remote_id
+            target.error = ""
+            target.save(update_fields=["status", "remote_id", "error", "updated_at"])
+            any_ok = True
+        except Exception as exc:  # noqa: BLE001
+            target.status = SocialPostTarget.Status.FAILED
+            target.error = str(exc)[:2000]
+            target.save(update_fields=["status", "error", "updated_at"])
+
+    post.status = (
+        SocialPost.Status.PUBLISHED if any_ok else SocialPost.Status.FAILED
+    )
+    if not any_ok:
+        post.error = "All targets failed"
+    post.save(update_fields=["status", "error", "updated_at"])
+
+
+@task
+def publish_social_target(target_id: str) -> None:
+    try:
+        target = SocialPostTarget.objects.select_related("post", "account").get(
+            pk=target_id
+        )
+    except SocialPostTarget.DoesNotExist:
+        return
+    remote_id = _connector_for(target.platform).publish(target.post, target)
+    target.status = SocialPostTarget.Status.PUBLISHED
+    target.remote_id = remote_id
+    target.save(update_fields=["status", "remote_id", "updated_at"])