Add premium field to Media

Author: New-dev0

swibots/api/chat/controllers/media_controller.py

@@ -254,6 +254,7 @@ async def upload_media(
         task_count: int = int(os.getenv("UPLOAD_TASKS", 0)),
         min_file_size: int = None,
         for_document: bool = False,
+        premium: bool = False,
         retries: int = 10,
     ) -> Media:
         """Upload media to switch
@@ -356,6 +357,7 @@ async def upload_media(
             "sourceUri": file_response["fileId"],
             "checksum": file_response["contentSha1"],
             "ownerId": self.client.app.user.id,
+            "premium": premium
         }
         return self.client.build_object(Media, media)
 

swibots/api/chat/controllers/message_controller.py

@@ -169,6 +169,7 @@ async def send_media(
         inline_markup: Optional[InlineMarkup] = None,
         part_size: Optional[int] = None,
         task_count: Optional[int] = None,
+        premium: Optional[bool] = False,
         **kwargs,
     ) -> Message | Task:
         async def _upload_media(media):
@@ -188,8 +189,10 @@ async def _upload_media(media):
                     thumb=thumb,
                     part_size=part_size,
                     task_count=task_count,
+                    premium=premium,
                     for_document=kwargs.get("is_document")
                     or (kwargs.get("media_type", 0) == MediaType.DOCUMENT.value),
+                    
                 )
             elif not media:
                 raise ValueError("'media' or 'document' must be provided!")

swibots/api/chat/methods/upload_media.py

@@ -33,6 +33,7 @@ async def upload_media(
         callback_args: Optional[tuple] = None,
         part_size: Optional[int] = None,
         task_count: Optional[int] = None,
+        premium: Optional[bool] = False,
         **kwargs,
     ) -> Media:
         """upload a file to get `Media` object.
@@ -59,6 +60,7 @@ async def upload_media(
             callback_args=callback_args,
             part_size=part_size,
             task_count=task_count,
+            premium=premium,
             **kwargs,
         )
 
@@ -103,6 +105,7 @@ async def send_media(
         progress: Optional[callable] = None,
         progress_args: Optional[tuple] = None,
         inline_markup: Optional[InlineMarkup] = None,
+        premium: Optional[bool] = False,
         **kwargs,
     ):
         """Send a media to Chat!
@@ -145,6 +148,7 @@ async def send_media(
             progress_args=progress_args,
             reply_to_message_id=reply_to_message_id,
             inline_markup=inline_markup,
+            premium=premium,
             **kwargs,
         )
 

swibots/api/chat/models/message.py

@@ -293,7 +293,7 @@ def from_json(self, data: Optional[JSONDict]) -> "Message":
             self.embed_message = EmbeddedMedia.build_from_json(
                 data.get("embedMessage"), self.app
             )
-            self.is_embed_message = data.get("isEmbedMessage") or bool(
+            self.is_embed_message = data.get("isEmbedMessage") == "true" or bool(
                 self.embed_message
             )
             self.user_session_id = data.get("userSessionId")

swibots/api/common/models/media.py

@@ -23,6 +23,7 @@ def __init__(
         url: Optional[bool] = None,
         owner_id: Optional[int] = None,
         owner_type: Optional[str] = None,
+        premium: Optional[bool] = False
     ):
         super().__init__(app)
         self.id = id
@@ -39,6 +40,7 @@ def __init__(
         self.owner_id = owner_id
         self.url = url
         self.owner_type = owner_type
+        self.premium = premium
 
     @property
     def is_sticker(self) -> bool:
@@ -57,6 +59,7 @@ def to_form_data(self):
             "mediaInfo.thumbnailUrl": self.thumbnail_url,
             "mediaInfo.sourceUri": self.source_id,
             "mediaInfo.typeName": self.type_name,
+            "mediaInfo.premium": self.premium
         }
 
     def to_json(self) -> JSONDict:
@@ -78,6 +81,7 @@ def to_json(self) -> JSONDict:
                 "ownerType": self.owner_type,
                 "downloadUrl": self.url,
                 "url": self.url,
+                "premium": self.premium
             }.items()
             if y
         }
@@ -99,7 +103,7 @@ def from_json(self, data: Optional[JSONDict] = None) -> "Media":
             self.owner_type = data.get("ownerType")
             if isinstance(self.owner_id, str) and self.owner_id.isdigit():
                 self.owner_id = int(self.owner_id)
-
+            self.premium = data.get("premium", False)
         return self
 
     def to_update_request(self):
@@ -114,6 +118,7 @@ def to_update_request(self):
             "fileName": self.file_name,
             "fileSize": self.file_size,
             "url": self.url,
+            "premium": self.premium
         }
 
     async def edit(