geoffrey/eulaurarien
1
0
Fork 0
Code Issues 7 Pull requests Releases 1 Wiki Activity

Reworked how paths work

Browse source 
Get those tuples out of my eyes
This commit is contained in:
Geoffrey Frogeye 2019-12-15 22:21:05 +01:00
parent 7af2074c7a
commit aec8d3f8de
Signed by: geoffrey
GPG key ID: D8A7ECA00A8CD3DD
4 changed files with 354 additions and 114 deletions
Download patch file Download diff file Expand all files Collapse all files

36
feed_dns.py
View file

@ -49,9 +49,12 @@ class Writer(multiprocessing.Process):
select, write, updated, name, value = record
self.db.enter_step('feed_switch')
for source in select(self.db, value):
# write(self.db, name, updated, source=source)
write(self.db, name, updated)
try:
for source in select(self.db, value):
# write(self.db, name, updated, source=source)
write(self.db, name, updated)
except ValueError:
self.log.exception("Cannot execute: %s", record)
self.db.enter_step('block_wait')
@ -98,19 +101,22 @@ class Rapid7Parser(Parser):
self.prof.enter_step('parse_rapid7')
split = line.split('"')
for k in range(1, 14, 4):
key = split[k]
val = split[k+2]
data[key] = val
try:
for k in range(1, 14, 4):
key = split[k]
val = split[k+2]
data[key] = val
select, writer = FUNCTION_MAP[data['type']]
record = (
select,
writer,
int(data['timestamp']),
data['name'],
data['value']
)
select, writer = FUNCTION_MAP[data['type']]
record = (
select,
writer,
int(data['timestamp']),
data['name'],
data['value']
)
except IndexError:
self.log.exception("Cannot parse: %s", line)
self.register(record)