matu
/
Heroku
mirror of https://github.com/coddrago/Heroku
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add ability to specify filename in fast uploader

pull/1/head
hikariatama 2022-04-30 17:53:45 +00:00
parent b411d5156e
commit 5d4962389b
2 changed files with 10 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
hikka/fast_uploader.py
View File

@ -373,6 +373,7 @@ async def _internal_transfer_to_telegram(
client: TelegramClient, client: TelegramClient,
response: BinaryIO, response: BinaryIO,
progress_callback: callable, progress_callback: callable,
filename: str = "upload",
) -> Tuple[TypeInputFile, int]: ) -> Tuple[TypeInputFile, int]:
file_id = helpers.generate_random_long() file_id = helpers.generate_random_long()
file_size = os.path.getsize(response.name) file_size = os.path.getsize(response.name)
@ -403,10 +404,12 @@ async def _internal_transfer_to_telegram(
if len(buffer) > 0: if len(buffer) > 0:
await uploader.upload(bytes(buffer)) await uploader.upload(bytes(buffer))
await uploader.finish_upload() await uploader.finish_upload()
if is_large:
return InputFileBig(file_id, part_count, "upload"), file_size return (
else: (InputFileBig(file_id, part_count, filename), file_size)
return InputFile(file_id, part_count, "upload", hash_md5.hexdigest()), file_size if is_large
else (InputFile(file_id, part_count, filename, hash_md5.hexdigest()), file_size)
)
async def download_file( async def download_file(
@ -436,8 +439,9 @@ async def download_file(
async def upload_file( async def upload_file(
file: BinaryIO = None, file: BinaryIO = None,
progress_callback: callable = None, progress_callback: callable = None,
filename: str = "upload",
_client: TelegramClient = None, _client: TelegramClient = None,
) -> TypeInputFile: ) -> TypeInputFile:
res = (await _internal_transfer_to_telegram(_client, file, progress_callback))[0] res = (await _internal_transfer_to_telegram(_client, file, progress_callback, filename))[0]
return res return res

1
hikka/version.py
View File

@ -1 +1,2 @@
"""Represents current userbot version"""
__version__ = (1, 1, 11) __version__ = (1, 1, 11)