python调用virustota接口api实现上传文件返回查毒结果
virustotal是一個可以檢測文件與url,以及接口ip病毒的網站,如果要訪問,可能要翻墻,這一就不詳解了。在網上查看相關的知識的話,都是python2的代碼,好多已經不能用了。因此在這里我就分享一下python3調用virustotal的api接口實現(xiàn)本地文件上傳到該網站并返回查毒結果的功能。
這里一般是exe文件,如果有其他格式文件,就需要其他操作,這里先不講解。
下面是實現(xiàn)代碼,這里的apikey保密,大家可以自己去網站注冊,私有的api可能要聯(lián)系網站。
代碼如下:
#www.virustotal.com/vtapi/v2/import requests import jsondef getFileScanId(url,apikey,a,b):# /file/scan# /文件/掃描# 上傳并掃描文件# 限制為32MBparams = {'apikey': apikey}files = {'file': (a, open(b, 'rb'))}response = requests.post(url, files=files, params=params)my_scan_id = str(response.json()['scan_id'])return my_scan_iddef getFieReportResult(url,apikey,my_scan_id):#/file/report# /文件/報告# 檢索文件掃描報告#該resource參數(shù)可以是要獲取最新的病毒報告文件的MD5,SHA-1或SHA-256。#還可以指定/ file / scan端點scan_id返回的值。#如果allinfo參數(shù)設置為true除了返回防病毒結果之外的其他信息。get_params = {'apikey': apikey, 'resource': my_scan_id,'allinfo': '1'}response2 = requests.get(url, params=get_params)jsondata = json.loads(response2.text)with open("jsonResult.json","w") as f:json.dump(jsondata, f, indent=4)return jsondatadef getResult(json):result = {}for k,v in json["scans"].items():result[k] = v['detected']print(result)print("一共有{0}條殺毒數(shù)據(jù)".format(len(result)))with open("result.txt","w") as g:g.write(str(result))def main():file_name = input("請輸入文件名:")a = str(file_name)file_src = input("請輸入文件路徑:")b = str(file_src)url1 = 'https://www.virustotal.com/vtapi/v2/file/scan'url2 = "https://www.virustotal.com/vtapi/v2/file/report"#需要提供密鑰,否者會出現(xiàn)403錯誤apikey = "保密"#獲得文件scan_idscan_id = getFileScanId(url1,apikey,a,b)#獲得返回的json結果并寫入result文件#getFieReportResult(url2, apikey, scan_id)json = getFieReportResult(url2,apikey,scan_id)getResult(json)if __name__ == '__main__':main()結果如下:
如果是False,則是安全的;如果出現(xiàn)了True,這可能有病毒。
沒有病毒的文件:
有病毒的文件:
但有時可能會出現(xiàn):
這里就是沒有返回結果,為什么會這樣,看一下官方的api文檔的一些規(guī)則就知道了:
這段英文的意思是:
Public API限制為每分鐘4個請求。
Public API不得用于商業(yè)產品或服務。
Private API返回更多威脅數(shù)據(jù)并公開更多端點。
Private API由SLA管理,可確保數(shù)據(jù)準備就緒。
原來我們的公共api只能夠這樣訪問了,不過這些就夠用了。如果大家想要一些高級功能的話,就可以聯(lián)系官方,獲得私有apikey,就可以獲得更多功能了。
總結
以上是生活随笔為你收集整理的python调用virustota接口api实现上传文件返回查毒结果的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: excel 画散点图 怎么设置图片的分辨
- 下一篇: 教你炒股票25:吻,MACD、背弛、中枢