metaso-sdk 0.1.4__py3-none-any.whl → 0.1.6__py3-none-any.whl

metaso_sdk/__init__.py

@@ -5,19 +5,22 @@ The official Python SDK for https://metaso.cn
 
 from __future__ import annotations
 
-from .model import File, Query, Status, Topic
+from .model import File, Book, Query, Status, Topic
 from .search import search
 from .subject import create_topic, delete_file, delete_topic, update_progress, upload_directory, upload_file
+from .bookshelf import upload_book
 
 __all__: list[str] = [
     "Status",
     "Query",
     "Topic",
     "File",
+    "Book",
     "search",
     "create_topic",
     "delete_topic",
     "upload_file",
+    "upload_book",
     "update_progress",
     "delete_file",
     "upload_directory",

metaso_sdk/bookshelf.py

@@ -0,0 +1,23 @@
+from typing import Optional
+
+from .client import client
+from .model import Status, Book
+
+
+def upload_book(fileobj_or_url) -> Optional[Book]:
+    """上传文件或 URL 到书架。
+
+    :param fileobj_or_url: 待上传的文件对象或则 URL。
+    :return: 如果文件上传成功，返回 Book 对象；否则返回 None。
+    """
+    if isinstance(fileobj_or_url, str):
+        params = {"data": {"url": fileobj_or_url}}
+    else:
+        params = {"files": {"file": fileobj_or_url}}
+    resp = client.put("/book", **params)
+    resp.raise_for_status()
+    json = resp.json()
+    status = Status.model_validate(json)
+    if status.errCode == 0:
+        book = Book.model_validate(json["data"])
+        return book

metaso_sdk/model.py

@@ -35,3 +35,16 @@ class File(BaseModel):
     previewUrl: Optional[str] = None
     originalUrl: Optional[str] = None
     progress: int
+
+
+class Book(BaseModel):
+    id: Optional[str]
+    thumbImg: Optional[str]
+    hasPpt: bool
+    title: str
+    url: str
+    fileId: str
+    lastPage: int
+    totalPage: int
+    progress: int
+    size: int

metaso_sdk/search.py

@@ -32,4 +32,5 @@ def search(query: Query, *, stream: bool = False, topic: Topic = None):
         return _gen()
 
     resp = client.post("/search/v2", json=query.model_dump())
+    resp.raise_for_status()
     return resp.json()["data"]

metaso_sdk/subject.py

@@ -1,10 +1,10 @@
 from pathlib import Path
-from typing import List, Optional
+from typing import List, Optional, Union
 
 from streamable import Stream
 
 from .client import client
-from .model import File, Status, Topic
+from .model import File, Book, Status, Topic
 
 
 def create_topic(topic: Topic) -> Optional[Topic]:
@@ -14,6 +14,7 @@ def create_topic(topic: Topic) -> Optional[Topic]:
     :return: 专题对象。
     """
     resp = client.put("/topic", json=topic.model_dump())
+    resp.raise_for_status()
     json = resp.json()
     status = Status.model_validate(json)
     if status.errCode == 0:
@@ -28,6 +29,7 @@ def delete_topic(topic: Topic) -> bool:
     :return: 如果成功删除Topic返回True，否则返回False。
     """
     resp = client.post("/topic/trash", json={"ids": [topic.id]})
+    resp.raise_for_status()
     status = Status.model_validate(resp.json())
     return status.errCode == 0
 
@@ -40,6 +42,7 @@ def upload_file(topic: Topic, file) -> Optional[File]:
     :return: 如果文件上传成功，返回文件对象；否则返回None。
     """
     resp = client.put(f"/file/{topic.dirRootId}", files={"file": file})
+    resp.raise_for_status()
     json = resp.json()
     status = Status.model_validate(json)
     if status.errCode == 0:
@@ -47,27 +50,31 @@ def upload_file(topic: Topic, file) -> Optional[File]:
         return file
 
 
-def update_progress(file: File) -> File:
-    """更新文件处理进度。
+def update_progress(file_or_book: Union[File, Book]) -> Union[File, Book]:
+    """更新处理进度。
 
-    :param file: 需要更新进度的文件对象。
-    :return: 更新后的文件对象。
+    :param file_or_book: 待更新进度的 File 或者 Book 对象。
+    :return: 更新 progress 后的对象。
     """
-    resp = client.get(f"/file/{file.id}/progress")
+    file_id = file_or_book.id if isinstance(file_or_book, File) else file_or_book.fileId
+    resp = client.get(f"/file/{file_id}/progress")
+    resp.raise_for_status()
     json = resp.json()
     status = Status.model_validate(json)
     if status.errCode == 0:
-        file.progress = json["data"]
-    return file
+        file_or_book.progress = json["data"]
+    return file_or_book
 
 
-def delete_file(file: File) -> bool:
-    """删除指定的文件。
+def delete_file(file_or_book: Union[File, Book]) -> bool:
+    """删除文件。
 
-    :param file: 要删除的File对象。
-    :return: 如果成功删除File返回True，否则返回False。
+    :param file_or_book: 待删除的 File 或者 Book 对象。
+    :return: 如果删除成功，返回True。否则返回False。
     """
-    resp = client.post("/file/trash", json={"ids": [file.id]})
+    file_id = file_or_book.id if isinstance(file_or_book, File) else file_or_book.fileId
+    resp = client.post("/file/trash", json={"ids": [file_id]})
+    resp.raise_for_status()
     status = Status.model_validate(resp.json())
     return status.errCode == 0
 

{metaso_sdk-0.1.4.dist-info → metaso_sdk-0.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: metaso-sdk
-Version: 0.1.4
+Version: 0.1.6
 Summary: The official Python SDK for https://metaso.cn
 Author-Email: Zhao Xiaohong <zhaoxiaohong@metasota.ai>
 License: MIT

metaso_sdk-0.1.6.dist-info/RECORD

@@ -0,0 +1,15 @@
+metaso_sdk-0.1.6.dist-info/METADATA,sha256=hC_yNcOnjofg2HJ7to-ePGlmDht6TpabzFetAvrMBHc,3270
+metaso_sdk-0.1.6.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
+metaso_sdk-0.1.6.dist-info/entry_points.txt,sha256=VqvzYeWOYZKCpogIumetYL8h5UZsI93oLI15kJjbwj4,63
+metaso_sdk-0.1.6.dist-info/licenses/LICENSE,sha256=7nWd1jm_pniRaOJtzgZnzECn_eeGLoPMvPTjCyRpj1A,1065
+metaso_sdk/__init__.py,sha256=ZXHdIXVXKtjC1QgE6v7AYBXAkOgzOivIzwcVRCEvvvM,582
+metaso_sdk/__main__.py,sha256=WCRS7Ehbf26-NWDSnt-Zmy0eP63paf67q9eYV4V_Q3g,345
+metaso_sdk/bookshelf.py,sha256=wkcSR5NQFweYslT-c1fSjPerfgsCi--Jd0vWVgBKpgI,721
+metaso_sdk/cli.py,sha256=Z7I54ojBpWxQ_Pce7-XM7gucneDIpC21yAqRGkYrm9I,1734
+metaso_sdk/client.py,sha256=j2hC1NBbxsRFHC4hOSD06YrfV5JLffgrA-H5iGNBPq4,204
+metaso_sdk/debug.py,sha256=K3ypLgs78a8bhiAHE_5fZZAL-U622fuKPyEdL7M1UW8,2831
+metaso_sdk/model.py,sha256=_2cSadnrM-YV03e67f8wlLoAbMF3jfIsMvgNqDA5zzg,935
+metaso_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+metaso_sdk/search.py,sha256=gdruxkw7_MAwjMICv4b8Xvqr5Q3P4_BFSydd756vfKw,1110
+metaso_sdk/subject.py,sha256=07jvVeU3kTMPZ1odnAzmzw8DVIT6idZoIIthVR5E4cg,3395
+metaso_sdk-0.1.6.dist-info/RECORD,,

metaso_sdk-0.1.4.dist-info/RECORD

@@ -1,14 +0,0 @@
-metaso_sdk-0.1.4.dist-info/METADATA,sha256=Ci3lD-LlXIN7M48tAwsDkqfBqq-p96JzfzR47YYTFu8,3270
-metaso_sdk-0.1.4.dist-info/WHEEL,sha256=tSfRZzRHthuv7vxpI4aehrdN9scLjk-dCJkPLzkHxGg,90
-metaso_sdk-0.1.4.dist-info/entry_points.txt,sha256=VqvzYeWOYZKCpogIumetYL8h5UZsI93oLI15kJjbwj4,63
-metaso_sdk-0.1.4.dist-info/licenses/LICENSE,sha256=7nWd1jm_pniRaOJtzgZnzECn_eeGLoPMvPTjCyRpj1A,1065
-metaso_sdk/__init__.py,sha256=wP088gdhNJt1byFsSYest-ucQJdwosorwaG7FqCyqwE,510
-metaso_sdk/__main__.py,sha256=WCRS7Ehbf26-NWDSnt-Zmy0eP63paf67q9eYV4V_Q3g,345
-metaso_sdk/cli.py,sha256=Z7I54ojBpWxQ_Pce7-XM7gucneDIpC21yAqRGkYrm9I,1734
-metaso_sdk/client.py,sha256=j2hC1NBbxsRFHC4hOSD06YrfV5JLffgrA-H5iGNBPq4,204
-metaso_sdk/debug.py,sha256=K3ypLgs78a8bhiAHE_5fZZAL-U622fuKPyEdL7M1UW8,2831
-metaso_sdk/model.py,sha256=Y-8Gn_9K4lbtqhiGVs796hB3UXMfgUALAFOjwJXiwDQ,730
-metaso_sdk/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-metaso_sdk/search.py,sha256=3iA9lIncpAcbPHPby0XnTYaYvhM2983bEOBa40Nu_kY,1082
-metaso_sdk/subject.py,sha256=qJBSn5LimFJ1df4sEWnp4or1rQlAGq-nWdPjac_heq4,2966
-metaso_sdk-0.1.4.dist-info/RECORD,,