Skip to content
Snippets Groups Projects
Commit 764226b2 authored by Erik Reid's avatar Erik Reid
Browse files

fixed overwritten data key

parent 9b0ab276
No related branches found
No related tags found
No related merge requests found
......@@ -63,8 +63,8 @@ if __name__ == "__main__":
with open("config.json") as f:
params = config.load(f)
# update_network_details(params)
update_network_details(params)
network_info = load_network_details(params["redis"])
with open("./router-info.json", "w") as f:
f.write(json.dumps(network_info))
# network_info = load_network_details(params["redis"])
# with open("./router-info.json", "w") as f:
# f.write(json.dumps(network_info))
import json
import logging
from celery import bootsteps, Task
import redis
......@@ -6,11 +7,13 @@ import redis
from inventory_provider.tasks.app import app
from inventory_provider import config
from inventory_provider import juniper, snmp
from inventory_provider import constants
class InventoryTask(Task):
config = None
logger = None
def __init__(self):
pass
......@@ -24,6 +27,8 @@ class InventoryTask(Task):
name=hostname,
key=key,
value=json.dumps(data))
InventoryTask.logger.debug(
"saved %s, key %s" % (hostname, key))
return "OK"
......@@ -31,6 +36,7 @@ class WorkerArgs(bootsteps.Step):
def __init__(self, worker, config_filename, **options):
with open(config_filename) as f:
InventoryTask.config = config.load(f)
InventoryTask.logger = logging.getLogger(constants.TASK_LOGGER_NAME)
def worker_args(parser):
......@@ -74,7 +80,7 @@ def juniper_refresh_interfaces(self, hostname):
def snmp_refresh_interfaces(self, hostname, community):
InventoryTask.save_key(
hostname,
"interfaces",
"snmp-interfaces",
list(snmp.get_router_interfaces(
hostname,
community,
......
This diff is collapsed.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment