Fixed handling of unknown field error

master
Geoffrey Frogeye 2019-12-27 01:10:21 +01:00
родитель 54a9c78534
Коммит 4fca68c6f0
Подписано: geoffrey
Идентификатор ключа GPG: D8A7ECA00A8CD3DD
1 изменённых файлов: 5 добавлений и 4 удалений

Просмотреть файл

@ -53,10 +53,9 @@ class Writer(multiprocessing.Process):
try:
for source in select(self.db, value):
write(self.db, name, updated, source=source)
except (ValueError, IndexError, KeyError):
except (ValueError, IndexError):
# ValueError: non-number in IP
# IndexError: IP too big, missing field
# KeyError: Unknown type field
# IndexError: IP too big
self.log.exception("Cannot execute: %s", record)
def end(self) -> None:
@ -149,7 +148,9 @@ class Rapid7Parser(Parser):
data['name'],
data['value']
)
except IndexError:
except (IndexError, KeyError):
# IndexError: missing field
# KeyError: Unknown type field
self.log.exception("Cannot parse: %s", line)
self.register(record)