twitwi 0.20.0__py3-none-any.whl → 0.21.0__py3-none-any.whl

twitwi/exceptions.py

@@ -14,4 +14,11 @@ class TwitterPayloadV2IncompleteIncludesError(TwitwiError):
     def __init__(self, kind, key):
         self.kind = kind
         self.key = key
-        super().__init__('{!r} ({})'.format(key, kind))
+        super().__init__("{!r} ({})".format(key, kind))
+
+
+class BlueskyPayloadError(TwitwiError):
+    def __init__(self, source, message):
+        self.source = source
+        self.message = message
+        super().__init__(f"Error while processing Bluesky post {source}:\n{message}")

twitwi/formatters.py

@@ -12,64 +12,68 @@ from twitwi.constants import (
     TWEET_PLURAL_FIELDS,
     USER_FIELDS,
     USER_BOOLEAN_FIELDS,
-    USER_PLURAL_FIELDS
+    USER_PLURAL_FIELDS,
 )
 
 
 def apply_tcat_format(item):
-    result = {v: item[k] for k, v in GAZOU_TO_TCAT['identical_fields'].items() if k in item}
-    result['source'] = "<a href=""{}"" rel=""nofollow"">{}</a>".format(item["source_url"], item["source_name"])
+    result = {
+        v: item[k] for k, v in GAZOU_TO_TCAT["identical_fields"].items() if k in item
+    }
+    result["source"] = "<a href={} rel=nofollow>{}</a>".format(
+        item["source_url"], item["source_name"]
+    )
     return result
 
 
 def make_transform_into_csv_dict(plural_fields, boolean_fields):
-
-    def transform_into_csv_dict(item, item_id=None, plural_separator='|', allow_erroneous_plurals=False):
+    def transform_into_csv_dict(
+        item, item_id=None, plural_separator="|", allow_erroneous_plurals=False
+    ):
         if item_id is not None:
-            item['id'] = item_id
+            item["id"] = item_id
 
-        item['links'] = item.get('proper_links', item.get('links', []))
+        item["links"] = item.get("proper_links", item.get("links", []))
 
         for plural_field in plural_fields:
             plurals = item.get(plural_field, [])
             if allow_erroneous_plurals:
-                plurals = [element if element is not None else '' for element in plurals]
+                plurals = [
+                    element if element is not None else "" for element in plurals
+                ]
             item[plural_field] = plural_separator.join(plurals)
 
         for boolean_field in boolean_fields:
-            item[boolean_field] = int(item[boolean_field]) if boolean_field in item else ''
+            item[boolean_field] = (
+                int(item[boolean_field]) if boolean_field in item else ""
+            )
 
     return transform_into_csv_dict
 
 
 def make_format_as_csv_row(fields, plural_fields, boolean_fields):
-
-    def format_field_for_csv(field, item, item_id=None, plural_separator='|'):
-
-        if field == 'id' and item_id is not None:
+    def format_field_for_csv(field, item, item_id=None, plural_separator="|"):
+        if field == "id" and item_id is not None:
             return item_id
 
         # NOTE: this is hardly the most performant way to proceed by I am lazy...
         if field in plural_fields:
             v = item.get(field, [])
 
-            if field == 'links':
-                v = item.get('proper_links', v)
+            if field == "links":
+                v = item.get("proper_links", v)
 
             return plural_separator.join(v)
 
         if field in boolean_fields:
-            return int(item[field]) if field in item else ''
+            return int(item[field]) if field in item else ""
 
-        return item.get(field, '')
+        return item.get(field, "")
 
-    def format_item_as_csv_row(item, item_id=None, plural_separator='|'):
+    def format_item_as_csv_row(item, item_id=None, plural_separator="|"):
         return [
             format_field_for_csv(
-                field,
-                item,
-                item_id=item_id,
-                plural_separator=plural_separator
+                field, item, item_id=item_id, plural_separator=plural_separator
             )
             for field in fields
         ]
@@ -78,31 +82,25 @@ def make_format_as_csv_row(fields, plural_fields, boolean_fields):
 
 
 transform_tweet_into_csv_dict = make_transform_into_csv_dict(
-    TWEET_PLURAL_FIELDS,
-    TWEET_BOOLEAN_FIELDS
+    TWEET_PLURAL_FIELDS, TWEET_BOOLEAN_FIELDS
 )
 
 format_tweet_as_csv_row = make_format_as_csv_row(
-    TWEET_FIELDS,
-    TWEET_PLURAL_FIELDS,
-    TWEET_BOOLEAN_FIELDS
+    TWEET_FIELDS, TWEET_PLURAL_FIELDS, TWEET_BOOLEAN_FIELDS
 )
 
 transform_user_into_csv_dict = make_transform_into_csv_dict(
-    USER_PLURAL_FIELDS,
-    USER_BOOLEAN_FIELDS
+    USER_PLURAL_FIELDS, USER_BOOLEAN_FIELDS
 )
 
 format_user_as_csv_row = make_format_as_csv_row(
-    USER_FIELDS,
-    USER_PLURAL_FIELDS,
-    USER_BOOLEAN_FIELDS
+    USER_FIELDS, USER_PLURAL_FIELDS, USER_BOOLEAN_FIELDS
 )
 
 __all__ = [
-    'transform_tweet_into_csv_dict',
-    'format_tweet_as_csv_row',
-    'transform_user_into_csv_dict',
-    'format_user_as_csv_row',
-    'apply_tcat_format'
+    "transform_tweet_into_csv_dict",
+    "format_tweet_as_csv_row",
+    "transform_user_into_csv_dict",
+    "format_user_as_csv_row",
+    "apply_tcat_format",
 ]