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
80
81
82
83
84
85
86
87
88
89
|
import mimetypes
import traceback
from os import mkdir
from os.path import exists
import requests
from telegram import Update
from telegram.constants import ParseMode
from telegram.ext import ApplicationBuilder, MessageHandler, CallbackContext, filters
def upload_file(file_path: str, comment: str) -> str:
mime_type, _ = mimetypes.guess_type(file_path)
if mime_type is None:
raise Exception("Unknown MIME type")
response = requests.post(
'https://tnd.quest/upload.php',
files=dict(
file=(file_path.split('/')[-1], open(file_path, 'rb'), mime_type)
),
data=dict(
comment=comment,
visibility=1
),
headers=dict(
accept='application/json'
)
)
if response.status_code == 201:
j = response.json()
return j['data']['urls']['download_url']
else:
raise Exception(f"Failed to send a file ({response.status_code}): {response.text}")
async def download_locally_file(update, file_id):
file = await update.get_bot().get_file(file_id)
file_extension = ""
if file.file_path is not None:
file_extension = f".{file.file_path.split('.')[-1]}".lower()
file_path = f"./temp/{file_id.file_unique_id}{file_extension}"
await file.download_to_drive(file_path)
return file_path
async def download_file(update: Update, _: CallbackContext) -> None:
if not exists("./temp"):
mkdir("./temp")
file_path = None
file_comment = update.message.caption
if len(update.message.photo) > 0:
file_path = await download_locally_file(update, update.message.photo[-1])
elif update.message.video is not None:
file_path = await download_locally_file(update, update.message.video)
elif update.message.video_note is not None:
file_path = await download_locally_file(update, update.message.video_note)
elif update.message.voice is not None:
file_path = await download_locally_file(update, update.message.voice)
if file_path is not None:
try:
url = upload_file(file_path, file_comment)
await update.message.reply_text(f'Saved! Here is your URL: [{url}]({url})', ParseMode.MARKDOWN,
reply_to_message_id=update.message.message_id)
except Exception as e:
traceback.print_tb(e.__traceback__)
await update.message.reply_text(f"Error occurred while uploading your file: {str(e)}")
def run():
print("Starting the bot...")
app = ApplicationBuilder().token("xd").build()
app.add_handler(MessageHandler(
filters.PHOTO | filters.VIDEO | filters.VIDEO_NOTE |
filters.VOICE,
download_file
))
app.run_polling()
if __name__ == "__main__":
run()
|
