Spaces:
Paused
Paused
File size: 3,191 Bytes
861d763 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 |
from telethon.events import CallbackQuery
from bot import TelegramBot
from bot.modules.decorators import verify_user,byte_to_gb_mb
from bot.modules.static import *
from bot.modules.telegram import get_message, get_file_properties
from bot.config import Telegram, Server
'''async def upload_file_in_chunks(TelegramBot, chat_id, file_path, chunk_size_kb):
file_id = None
total_parts = math.ceil(file_size / (chunk_size_kb * 1024))
current_part = 1
with open(file_path, "rb") as f:
while True:
data = f.read(chunk_size_kb * 1024)
if not data:
break
# Upload the current chunk
file_part = await client.upload_file_part(file_id, current_part, data)
if not file_id:
file_id = file_part.file_id
current_part += 1
# Show progress if desired
progress = current_part / total_parts * 100
print(f"Uploaded {progress:.2f}%")
# Complete the upload
complete_file = await client.upload_file_completed(file_id, total_parts)
# Send the uploaded file to the chat
await client.send_file(chat_id, complete_file)
'''
@TelegramBot.on(CallbackQuery(pattern=r'^rm_'))
#@verify_user(private=True)
async def delete_file(event: CallbackQuery.Event):
query_data = event.query.data.decode().split('_')
if len(query_data) != 3:
return await event.answer(InvalidQueryText, alert=True)
message = await get_message(int(query_data[1]))
if not message:
return await event.answer(MessageNotExist, alert=True)
await message.delete()
return await event.answer(LinkRevokedText, alert=True)
@TelegramBot.on(CallbackQuery(pattern=r'^up_'))
#@verify_user(private=True)
async def up_file(event:CallbackQuery.Event):
query_data = event.query.data.decode().split('_')
if len(query_data) != 5:
return await event.answer(InvalidQueryText, alert=True)
message_id=int(query_data[1])
event_id=int(query_data[3])
event_peer=int(query_data[4])
message = await get_message(int(query_data[1]))
main_event = await TelegramBot.get_messages(event_peer, ids=event_id)
#print(event_id,event_peer)
#print(main_event)
code = str(query_data[2])
def callback(current, total,event_peer,event_id):
print('Uploaded', current, 'out of', total,'bytes: {:.2%}'.format(current / total))
TelegramBot.edit_message(event_peer, event_id, f'Uploaded, {current}, out of, {total}')
if not message or code != message.raw_text:
return await event.answer(MessageNotExist, alert=True)
else:
dl_link = f'{Server.BASE_URL}/dl/{message_id}?code={code}'
file_name, file_size, mime_type = get_file_properties(main_event)
await main_event.reply(
message= f'**File name:{file_name} \nFile size:{byte_to_gb_mb(file_size)} \nID:{message_id} \nAccess Code:{code} \n*If you want to upload with different name please send name with reply of the message \n reply with n for original file name ***',
parse_mode="Markdown"
)
|