diff --git a/huntflow_api_client/entities/file.py b/huntflow_api_client/entities/file.py index 482e6dd..479739c 100644 --- a/huntflow_api_client/entities/file.py +++ b/huntflow_api_client/entities/file.py @@ -1,4 +1,4 @@ -from typing import BinaryIO, Optional, Union +from typing import BinaryIO, Dict, Optional, Tuple, Union from huntflow_api_client.entities.base import BaseEntity from huntflow_api_client.models.request.file import UploadFileHeaders @@ -12,6 +12,7 @@ async def upload( headers: UploadFileHeaders, file: Union[bytes, BinaryIO], preset: Optional[str] = None, + filename: Optional[str] = None, ) -> UploadResponse: """ API method reference https://api.huntflow.ai/v2/docs#post-/accounts/-account_id-/upload @@ -20,16 +21,24 @@ async def upload( :param file: File :param preset: Preset :param headers: Headers + :param filename: Filename :return: Additional data """ data = {} if preset: data["preset"] = preset + + files: Dict[str, Union[Union[bytes, BinaryIO], Tuple[str, Union[bytes, BinaryIO]]]] = {} + if filename: + files["file"] = (filename, file) + else: + files["file"] = file + response = await self._api.request( "POST", f"/accounts/{account_id}/upload", - files={"file": file}, + files=files, data=data, headers=headers.jsonable_dict(exclude_none=True, by_alias=True), ) diff --git a/pyproject.toml b/pyproject.toml index e018b42..cb7f5ab 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,7 +1,7 @@ [project] name = "huntflow-api-client" -version = "2.10.0" +version = "2.11.0" description = "Huntflow API Client for Python" authors = [ {name = "Developers huntflow", email = "developer@huntflow.ru"},