aioqzone 1.8.4.dev3__tar.gz → 1.8.5.dev1__tar.gz

{aioqzone-1.8.4.dev3 → aioqzone-1.8.5.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: aioqzone
-Version: 1.8.4.dev3
+Version: 1.8.5.dev1
 Summary: A Python wrapper for Qzone login and H5 APIs.
 Home-page: https://github.com/aioqzone/aioqzone
 License: AGPL-3.0

{aioqzone-1.8.4.dev3 → aioqzone-1.8.5.dev1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "aioqzone"
-version = "1.8.4.dev3"
+version = "1.8.5.dev1"
 description = "A Python wrapper for Qzone login and H5 APIs."
 authors = ["aioqzone <zzzzss990315@gmail.com>"]
 license = "AGPL-3.0"

{aioqzone-1.8.4.dev3 → aioqzone-1.8.5.dev1}/src/aioqzone/api/h5/model.py

@@ -1,4 +1,5 @@
 import logging
+from os import PathLike
 
 from pydantic import ValidationError
 from tenacity import AsyncRetrying, TryAgain, after_log, stop_after_attempt
@@ -177,7 +178,7 @@ class QzoneH5API:
     async def publish_mood(
         self,
         content: str,
-        photos: t.Optional[t.List[PhotoData]] = None,
+        photos: t.Optional[t.Sequence[t.Union[PhotoData, PicInfo]]] = None,
         sync_weibo=False,
         ugc_right: UgcRight = UgcRight.all,
     ) -> PublishMoodResp:
@@ -189,21 +190,47 @@ class QzoneH5API:
         :param ugc_right: access right, default to "Available to Everyone".
         """
         photos = photos or []
-        return await self.call(PublishMoodApi(params=PublishMoodParams.model_validate(locals())))
+        return await self.call(
+            PublishMoodApi(params=PublishMoodParams.model_validate(locals(), from_attributes=True))
+        )
 
     async def upload_pic(
-        self, picture: bytes, width: int, height: int, quality: int
+        self,
+        picture: t.Union[bytes, str, PathLike, t.IO[bytes]],
+        width: t.Optional[int] = None,
+        height: t.Optional[int] = None,
+        quality: t.Union[int, float] = 70,
     ) -> UploadPicResponse:
-        return await self.call(
-            UploadPicApi(
-                params=UploadPicParams(
-                    picture=picture,
-                    hd_width=width,
-                    hd_height=height,
-                    hd_quality=quality,
-                ),
+        """
+        .. versionchanged:: 1.8.5
+
+            In version <= 1.8.4, user is responsible for compressing a image and this api
+            encode the :obj:`picture` with Base64 and send it to Qzone _ASIS_.
+
+            Since version 1.8.5, we recognize a compressed image by :obj:`width` and :obj:`height`
+            parameters. If :obj:`width` and :obj:`height` is provided, this API will keep the former
+            behavior. If not provided, the image will be compressed with the given quality.
+        """
+        if isinstance(quality, float):
+            if quality < 1:
+                quality *= 100
+            quality = int(quality)
+
+        assert 0 < quality <= 100
+
+        if isinstance(picture, (str, PathLike, t.IO)):
+            params = UploadPicParams.from_image(picture, quality)
+        elif (width is None) or (height is None):
+            params = UploadPicParams.from_bytes(picture, quality)
+        else:
+            params = UploadPicParams(
+                picture=picture,
+                hd_width=width,
+                hd_height=height,
+                hd_quality=quality,
             )
-        )
+
+        return await self.call(UploadPicApi(params=params))
 
     async def preupload_photos(
         self, upload_pics: t.List[UploadPicResponse], cur_num=0, upload_hd=False

{aioqzone-1.8.4.dev3 → aioqzone-1.8.5.dev1}/src/aioqzone/model/api/request.py

@@ -1,9 +1,12 @@
 import typing as t
 from base64 import b64encode
+from io import BytesIO
 from math import floor
+from os import PathLike
 from time import time
 
-from pydantic import BaseModel, Field, field_serializer
+from pydantic import BaseModel, Field, field_serializer, field_validator
+from typing_extensions import Buffer
 
 from aioqzone.utils.time import time_ms
 
@@ -183,6 +186,22 @@ class UploadPicParams(QzoneRequestParams):
     def b64_picture(self, picture: t.ByteString) -> str:
         return b64encode(picture).decode()
 
+    @classmethod
+    def from_image(cls, image_file: t.Union[str, PathLike, t.IO[bytes]], quality=70):
+        import PIL.Image as image
+
+        with image.open(image_file) as f:
+            buf = BytesIO()
+            f.save(buf, "JPEG", quality=quality)
+            return cls(
+                picture=buf.getvalue(), hd_height=f.height, hd_width=f.width, hd_quality=quality
+            )
+
+    @classmethod
+    def from_bytes(cls, image_bytes: Buffer, quality=70):
+        buf = BytesIO(image_bytes)
+        return cls.from_image(buf, quality=quality)
+
 
 class PhotosPreuploadParams(QzoneRequestParams):
     uin_fields = ("uin",)

{aioqzone-1.8.4.dev3 → aioqzone-1.8.5.dev1}/src/aioqzone/model/api/response.py

@@ -33,6 +33,7 @@ __all__ = [
     "UploadPicResponse",
     "PhotosPreuploadResponse",
     "FeedData",
+    "PicInfo",
     "ProfileFeedData",
 ]
 
@@ -273,4 +274,4 @@ class PhotosPreuploadResponse(QzoneResponse):
 
         picinfos = json_loads(m.group(1))
         assert isinstance(picinfos, list)
-        return dict(photos=[PicInfo.from_response_object(info["picinfo"]) for info in picinfos])
+        return dict(photos=[PicInfo.from_response_object(info["picinfo"]) for info in picinfos])  # type: ignore