lucaspalomodevelop/core
1
0
Fork 0
mirror of https://github.com/lucaspalomodevelop/core.git synced 2026-03-16 09:34:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Network insight, support custom configuration in flowd_aggregate_metadata.py for https://github.com/opnsense/core/issues/2964

Browse Source
This commit is contained in:
Ad Schellevis 2018-11-27 10:43:16 +01:00
parent 4e69e5923b
commit 8fdaf264de
1 changed files with 14 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/opnsense/scripts/netflow/flowd_aggregate_metadata.py
View File

@ -1,7 +1,7 @@
#!/usr/local/bin/python2.7
"""
Copyright (c) 2016 Ad Schellevis <ad@opnsense.org>
Copyright (c) 2016-2018 Ad Schellevis <ad@opnsense.org>
All rights reserved.
Redistribution and use in source and binary forms, with or without
@ -28,16 +28,24 @@
--------------------------------------------------------------------------------------
fetch flowd aggregate metadata
"""
import sys
import ujson
import datetime
import argparse
from lib import load_config
from lib.aggregate import AggMetadata
import lib.aggregates
result = dict()
parser = argparse.ArgumentParser()
parser.add_argument('--config', '--config', help='configuration yaml', default=None)
parser.add_argument('format', help='output format [text (default)|json]')
cmd_args = parser.parse_args()
configuration = load_config(cmd_args.config)
# load global metadata
metadata = AggMetadata()
metadata = AggMetadata(database_dir=configuration.database_dir)
result['last_sync'] = metadata.last_sync()
# fetch aggregators
result['aggregators'] = dict()
@ -47,12 +55,13 @@ for agg_class in lib.aggregates.get_aggregators():
result['aggregators'][agg_class.__name__]['resolutions'].append(resolution)
# output result
if len(sys.argv) > 1 and 'json' in sys.argv:
if cmd_args.format.find('json') > -1:
# json format
print(ujson.dumps(result))
else:
# plain text format
print ('Flowd aggregations [last sync %s]' %
print (
'Flowd aggregations [last sync %s]' %
datetime.datetime.fromtimestamp(result['last_sync']).strftime('%Y-%m-%d %H:%M:%S')
)
print ('\nAggregators:')