def put(self):
"""
Import
Endpoint that imports the whitelist from a file. The file should be structured the same as the /whitelist
endpoint, for example:
<br>
<pre>
[
{
"id": "cpe:2.3:h:-:wireless_ip_camera_360:-:*:*:*:*:*:*:*",
"type": "cpe"
},
{
"id": "cpe:2.3:a:.bbsoftware:bb_flashback:*:*:*:*:*:*:*:*",
"type": "cpe"
},
{
"comments": [
" test"
],
"id": "cpe:2.3:a:.joomclan:com_joomclip:*:*:*:*:*:*:*:*",
"type": "cpe"
}
]
</pre>
"""
data = request.data
if len(data) == 0:
wl.abort(400, "No file selected")
return {"message": importWhitelist(data)}
def listImport(self, force=None, path=None):
_list = request.url_rule.split('/')[2]
file = request.files['file']
force = request.form.get('force')
count = wl.countWhitelist() if _list.lower == 'whitelist' else bl.countBlacklist()
if (count == 0) | (not count) | (force == "f"):
if _list.lower == 'whitelist':
wl.dropWhitelist()
wl.importWhitelist(TextIOWrapper(file.stream))
else:
bl.dropBlacklist()
bl.importBlacklist(TextIOWrapper(file.stream))
status = _list[0]+"l_imported"
else:
status = _list[0]+"l_already_filled"
return render_template('admin.html', status=status, **self.adminInfo())
def listImport(self, force=None, path=None):
_list = request.url_rule.split('/')[2]
file = request.files['file']
force = request.form.get('force')
count = wl.countWhitelist() if _list.lower == 'whitelist' else bl.countBlacklist()
if (count == 0) | (not count) | (force == "f"):
if _list.lower == 'whitelist':
wl.dropWhitelist()
wl.importWhitelist(TextIOWrapper(file.stream))
else:
bl.dropBlacklist()
bl.importBlacklist(TextIOWrapper(file.stream))
status = _list[0]+"l_imported"
else:
status = _list[0]+"l_already_filled"
return render_template('admin.html', status=status, **self.adminInfo())
def listImport():
_list = request.url_rule.rule.split("/")[2]
file = request.files["file"]
force = request.form.get("force")
count = countWhitelist() if _list.lower() == "whitelist" else countBlacklist()
if (count == 0) | (not count) | (force == "f"):
if _list.lower() == "whitelist":
dropWhitelist()
importWhitelist(file.stream)
else:
dropBlacklist()
importBlacklist(file.stream)
status = _list[0] + "l_imported"
else:
status = _list[0] + "l_already_filled"
return render_template("admin.html", status=status, **adminInfo())
def api_admin_import_whitelist(self):
return wl.importWhitelist(StringIO(request.data.decode("utf-8")))
def api_admin_import_whitelist(self):
return wl.importWhitelist(StringIO(request.data.decode("utf-8")))
