rocketchat-API 3.5.0__tar.gz → 3.6.0__tar.gz

{rocketchat_api-3.5.0 → rocketchat_api-3.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocketchat_API
-Version: 3.5.0
+Version: 3.6.0
 Summary: Python API wrapper for Rocket.Chat
 Author-email: "Jorge Alberto Díaz Orozco (Akiel)" <diazorozcoj@gmail.com>
 Maintainer-email: "Jorge Alberto Díaz Orozco (Akiel)" <diazorozcoj@gmail.com>

{rocketchat_api-3.5.0 → rocketchat_api-3.6.0}/rocketchat_API/APISections/rooms.py

@@ -69,11 +69,22 @@ class RocketChatRooms(RocketChatBase):
         """Upload media files to a room."""
         files = {
             "file": (
-                os.path.basename(file),
-                open(file, "rb"),
-                mimetypes.guess_type(file)[0],
+                file
+                if isinstance(file, (list, tuple))
+                else (
+                    os.path.basename(file),
+                    open(file, "rb"),
+                    mimetypes.guess_type(file)[0],
+                )
             ),
         }
         return self.call_api_post(
             "rooms.media/" + room_id, kwargs=kwargs, use_json=False, files=files
         )
+
+    def rooms_media_confirm(self, room_id, file_id, **kwargs):
+        """Confirm a previously uploaded media file in a room."""
+        return self.call_api_post(
+            "rooms.mediaConfirm/" + room_id + "/" + file_id,
+            kwargs=kwargs,
+        )

{rocketchat_api-3.5.0 → rocketchat_api-3.6.0}/rocketchat_API.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: rocketchat_API
-Version: 3.5.0
+Version: 3.6.0
 Summary: Python API wrapper for Rocket.Chat
 Author-email: "Jorge Alberto Díaz Orozco (Akiel)" <diazorozcoj@gmail.com>
 Maintainer-email: "Jorge Alberto Díaz Orozco (Akiel)" <diazorozcoj@gmail.com>

{rocketchat_api-3.5.0 → rocketchat_api-3.6.0}/tests/test_rooms.py

@@ -15,6 +15,47 @@ def test_rooms_media(logged_rocket):
     assert str(rooms_media.get("file").get("url")).endswith("avatar.png")
 
 
+def test_rooms_media_array_file(logged_rocket):
+    with open("tests/assets/avatar.png", "rb") as avatar_file:
+        avatar_bytes = avatar_file.read()
+
+    rooms_media = logged_rocket.rooms_media(
+        "GENERAL",
+        file=("avatar.png", avatar_bytes, "image/png"),
+        description="hey there",
+    )
+    assert str(rooms_media.get("file").get("url")).endswith("avatar.png")
+
+
+def test_rooms_media_array_file_with_list(logged_rocket):
+    with open("tests/assets/avatar.png", "rb") as avatar_file:
+        avatar_bytes = avatar_file.read()
+
+    rooms_media = logged_rocket.rooms_media(
+        "GENERAL",
+        file=["avatar.png", avatar_bytes, "image/png"],  # list instead of tuple
+        description="hey there",
+    )
+
+    assert str(rooms_media.get("file").get("url")).endswith("avatar.png")
+
+
+def test_rooms_media_confirm(logged_rocket):
+    uploaded_media = logged_rocket.rooms_media(
+        "GENERAL", file="tests/assets/avatar.png"
+    )
+    file_id = str(uploaded_media.get("file").get("_id"))
+    assert file_id
+
+    confirmed_media = logged_rocket.rooms_media_confirm(
+        room_id="GENERAL",
+        file_id=file_id,
+        msg="confirmed upload",
+    )
+
+    assert confirmed_media.get("message").get("_id")
+
+
 def test_rooms_get(logged_rocket):
     rooms_get = logged_rocket.rooms_get()
     assert "update" in rooms_get