128 lines
5.0 KiB
Python
128 lines
5.0 KiB
Python
# Copyright (C) 2020-2021 by DevsExpo@Github, < https://github.com/DevsExpo >.
|
|
#
|
|
# This file is part of < https://github.com/DevsExpo/FridayUserBot > project,
|
|
# and is released under the "GNU v3.0 License Agreement".
|
|
# Please see < https://github.com/DevsExpo/blob/master/LICENSE >
|
|
#
|
|
# All rights reserved.
|
|
|
|
import json
|
|
import os
|
|
import time
|
|
import requests
|
|
from pyrogram import Client, filters, enums
|
|
from pyrogram.types import Message
|
|
|
|
from utils.misc import modules_help, prefix
|
|
from utils.scripts import edit_or_reply, format_exc, time_formatter, humanbytes, edit_or_send_as_file, get_text, progress
|
|
from utils.config import vt_key as vak
|
|
|
|
|
|
@Client.on_message(filters.command("vt", prefix) & filters.me)
|
|
async def scan_my_file(client, message):
|
|
ms_ = await edit_or_reply(message, "`Please Wait! Scanning This File`")
|
|
if not message.reply_to_message:
|
|
return await ms_.edit("`Please Reply To File To Scan For Viruses`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if not message.reply_to_message.document:
|
|
return await ms_.edit("`Please Reply To File To Scan For Viruses`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if vak == None:
|
|
return await ms_.edit("`You Need To Set VIRUSTOTAL_API_KEY For Functing Of This Plugin.`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if int(message.reply_to_message.document.file_size) > 32000000:
|
|
return await ms_.edit("**File Too Large, Use `.vtl` instead**", parse_mode=enums.ParseMode.MARKDOWN)
|
|
c_time = time.time()
|
|
downloaded_file_name = await message.reply_to_message.download(
|
|
progress=progress,
|
|
progress_args=(ms_, c_time, '`Downloading This File!`'),
|
|
)
|
|
|
|
url = "https://www.virustotal.com/vtapi/v2/file/scan"
|
|
params = {"apikey": vak}
|
|
files = {"file": (downloaded_file_name, open(downloaded_file_name, "rb"))}
|
|
response = requests.post(url, files=files, params=params)
|
|
try:
|
|
# result = response.text
|
|
# print(result)
|
|
r_json = response.json()
|
|
md5 = r_json["md5"]
|
|
# print(md5)
|
|
except Exception as e:
|
|
return await ms_.edit(format_exc(e))
|
|
await ms_.edit(f"<b><u>Scanned {message.reply_to_message.document.file_name}</b></u>. <b>You Can Visit :</b> <a href=\"https://www.virustotal.com/gui/file/{md5}\">Here</a> <b>In 5-10 Min To See File Report</b>")
|
|
os.remove(downloaded_file_name)
|
|
|
|
@Client.on_message(filters.command("vtl", prefix) & filters.me)
|
|
async def scan_my_large_file(client, message):
|
|
ms_ = await edit_or_reply(message, "`Please Wait! Scanning This File`")
|
|
if not message.reply_to_message:
|
|
return await ms_.edit("`Please Reply To File To Scan For Viruses`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if not message.reply_to_message.document:
|
|
return await ms_.edit("`Please Reply To File To Scan For Viruses`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if vak == None:
|
|
return await ms_.edit("`You Need To Set VIRUSTOTAL_API_KEY For Functing Of This Plugin.`", parse_mode=enums.ParseMode.MARKDOWN)
|
|
if int(message.reply_to_message.document.file_size) > 650000000:
|
|
return await ms_.edit("**File Too Large, exceeded Max capacity of 650MB**", parse_mode=enums.ParseMode.MARKDOWN)
|
|
c_time = time.time()
|
|
downloaded_file_name = await message.reply_to_message.download(
|
|
progress=progress,
|
|
progress_args=(ms_, c_time, '`Downloading This File!`'),
|
|
)
|
|
|
|
url1 = "https://www.virustotal.com/api/v3/files/upload_url"
|
|
|
|
headers = {
|
|
"accept": "application/json",
|
|
"x-apikey": vak
|
|
}
|
|
|
|
rponse = requests.get(url1, headers=headers)
|
|
try:
|
|
# result = rponse.text
|
|
# print(result)
|
|
r_json = rponse.json()
|
|
upl_data = r_json["data"]
|
|
# print(upl_data)
|
|
except Exception as e:
|
|
return await ms_.edit(format_exc(e))
|
|
|
|
# print(rponse.text)
|
|
|
|
url = upl_data
|
|
|
|
files = {"file": (downloaded_file_name, open(downloaded_file_name, "rb"))}
|
|
headers = {
|
|
"accept": "application/json",
|
|
"x-apikey": vak
|
|
}
|
|
response = requests.post(url, files=files, headers=headers)
|
|
# print(response.text)
|
|
|
|
r_json = response.json()
|
|
analysis_url = r_json["data"]["links"]["self"]
|
|
|
|
url = analysis_url
|
|
|
|
headers = {
|
|
"accept": "application/json",
|
|
"x-apikey": vak
|
|
}
|
|
|
|
response_result = requests.get(url, headers=headers)
|
|
|
|
# print(response_result.text)
|
|
|
|
try:
|
|
# result = response.text
|
|
# print(result)
|
|
r_json = response_result.json()
|
|
md5 = r_json["meta"]["file_info"]["md5"]
|
|
# print(md5)
|
|
except Exception as e:
|
|
return await ms_.edit(format_exc(e))
|
|
await ms_.edit(f"<b><u>Scanned {message.reply_to_message.document.file_name}</b></u>. <b>You Can Visit :</b> <a href=\"https://www.virustotal.com/gui/file/{md5}\">Here</a> <b>In 5-10 Min To See File Report</b>")
|
|
os.remove(downloaded_file_name)
|
|
|
|
modules_help["virustotal"] = {
|
|
"vt [reply to file]*": "Scan for viruses on Virus Total (for lower file size <32MB)",
|
|
"vtl [reply to file]*": "Scan for viruses on Virus Total (for lower file size >=32MB)",
|
|
}
|