adulau/passive-dns-atlas
1
0
Fork 0
mirror of https://github.com/adulau/passive-dns-atlas.git synced 2025-01-04 23:13:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

limit is now an option

Browse source
This commit is contained in:
Alexandre Dulaunoy 2017-04-21 11:13:50 +02:00
parent 0198e4e03d
commit 338b7cd1be
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
bin/get_stats.py
View file

@ -14,11 +14,11 @@ import sys
import os
fieldsSagan = ['Type', 'Name', 'TTL', 'Class', 'Serial', 'Rname', 'MasterServerName', 'MaintainerName', 'Data']
limit = 100
parser = argparse.ArgumentParser(description='passive-dns-atlas statistics extractor')
parser.add_argument("-t","--table", default=False, action='store_true', help="Dump statistics table in ASCII")
parser.add_argument("-c","--csvd3js", default=False, action='store_true', help="Generate D3.js Bubble Chart")
parser.add_argument("-l","--limit", type=int, default=100, help="Limit of values to export per ZRANK - default 100")
parser.add_argument("-o","--outputdir", default="./stats/", help="Output directory")
args = parser.parse_args()
@ -28,7 +28,7 @@ if not os.path.exists(args.outputdir):
r = redis.StrictRedis(host='localhost', port=6379)
for field in fieldsSagan:
c = r.zrange(field.upper(), 0, limit, desc=True, withscores=True)
c = r.zrange(field.upper(), 0, args.limit, desc=True, withscores=True)
if args.table:
table = PrettyTable()