botapi-py 1.4.9__tar.gz → 1.4.10__tar.gz

{botapi_py-1.4.9 → botapi_py-1.4.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: botapi-py
-Version: 1.4.9
+Version: 1.4.10
 Summary: A simple and flexible Python library to conveniently build fast Telegram bots using the Bot API.
 Requires-Dist: httpx>=0.28.1
 Requires-Dist: pydantic>=2.12.0

{botapi_py-1.4.9 → botapi_py-1.4.10}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "botapi-py"
-version = "1.4.9"
+version = "1.4.10"
 description = "A simple and flexible Python library to conveniently build fast Telegram bots using the Bot API."
 # readme = "README.md"
 # authors = [

{botapi_py-1.4.9 → botapi_py-1.4.10}/src/botapi/core.py

@@ -49,7 +49,19 @@ class BaseBot:
 			url += 'test/'
 		url += method
 		return url
-
+		
+	def _compose_api_file_url(self, file_path: str) -> str:
+		'''
+		Composes the URL for file downloads from the Telegram Bot API.
+		'''
+		# https://api.telegram.org/file/bot<token>/<file_path>
+		# https://api.telegram.org/file/bot<token>/test/<file_path>
+		url = f'{self.api_url}/file/bot{self.token}/'
+		if self.test_server:
+			url += 'test/'
+		url += file_path
+		return url
+		
 	def _convert_data(self, data: dict[str, Any]) -> tuple[dict[str, Any], MultipartFiles]:
 		new_data: dict[str, Any] = {}
 		files: MultipartFiles = {} # (filename, file_content, mime_type)

{botapi_py-1.4.9 → botapi_py-1.4.10}/src/botapi/shortcuts.py

@@ -1,6 +1,6 @@
 from __future__ import annotations
 
-from typing import Optional, Union, List, cast
+from typing import Optional, Union, List, cast, overload
 
 from . import enums
 from . import types
@@ -243,3 +243,51 @@ class BotShortcuts:
 			case _:
 				raise ValueError(f'Unknown media type: {file_type}')
 				
+	@overload
+	async def download_file(
+		self,
+		file: types.File,
+		destination_path: str
+	) -> None: ...
+	
+	@overload
+	async def download_file(
+		self,
+		file: types.File,
+		destination_path: None = None
+	) -> bytes: ...
+	
+	async def download_file(
+		self,
+		file: types.File,
+		destination_path: Optional[str] = None
+	) -> bytes | None:
+		'''
+		Convenience method to download a file from Telegram servers.
+		
+		Call this method with a `File` object obtained via `get_file`.
+		
+		:param file: File object to download.
+		:param destination_path: Optional path to save the downloaded file. If not provided, the file content will be returned as bytes.
+		:return: The file content as bytes if destination_path is not provided, otherwise None.
+		'''
+		self = cast(methods.Bot, self)
+		
+		file_path = file.file_path
+		if not file_path:
+			raise ValueError('The file does not have a valid file_path to download.')
+			
+		# https://api.telegram.org/file/bot<token>/<file_path>
+		# https://api.telegram.org/file/bot<token>/test/<file_path>
+		url = self._compose_api_file_url(file_path=file_path) # pyright: ignore[reportPrivateUsage]
+		
+		response = await self.session.get(url)
+		content = response.content
+		
+		if destination_path:
+			with open(destination_path, 'wb') as f:
+				f.write(content)
+			return None
+			
+		return content
+