diff options
| author | Ernst Widerberg <ernst@sunet.se> | 2021-11-04 14:26:35 +0100 |
|---|---|---|
| committer | Ernst Widerberg <ernst@sunet.se> | 2021-11-04 14:26:35 +0100 |
| commit | 8c7e1c2e0c6faa608bf93867d9b409b4fd8b4f71 (patch) | |
| tree | 9c6fa75fc2dc2a1187344348f03a70db764a0c69 | |
| parent | dd0664047ca0843123079fc7e0dfe2f1f5c87e1b (diff) | |
Update feeder for new data format and add asn_country_code
| -rw-r--r-- | tools/feeder.py | 43 |
1 files changed, 19 insertions, 24 deletions
diff --git a/tools/feeder.py b/tools/feeder.py index 224ec1b..cf52c36 100644 --- a/tools/feeder.py +++ b/tools/feeder.py @@ -8,24 +8,6 @@ import json import threading import time -data = { - "ip": "", - "port": 0, - "whois_description": "", - "asn": "", - "asn_country_code": "", - "ptr": "", - "abuse_mail": "", - "domain": "", - "timestamp_in_utc": "", - "producer_unique_keys": { - "subject_cn": "unknown", - "subject_o": "unknown", - "full_name": "", - "end_of_general_support": False, - } -} - def random_string(iters=1): full_string = '' @@ -143,16 +125,29 @@ def generate_blobs(nr_blobs=1): 'port': random_port(), 'whois_description': random_string(2), 'asn': random_asn(), + 'asn_country_code': random_country(), 'ptr': random_string(2), 'abuse_mail': random_email(), 'domain': "sunet.se", 'timestamp_in_utc': get_timestamp(), - 'producer_unique_keys': { - 'subject_cn': 'unknown', - 'subject_o': 'unknown', - 'full_name': random_string(3), - 'end_of_general_support': False, - } + 'user_presentation': { + 'description': 'A presentation of the observation as a whole (optional)', + 'data': { + random_string(): { + 'display_name': random_string(), + 'data': random_string(), + 'description': random_string(), + }, + random_string(): { + 'display_name': random_string(), + 'data': random_string(), + }, + random_string(): { + 'display_name': random_string(), + 'data': random_string(), + }, + } + }, }) return blobs |