From e9c2b96a2528bcd921d4b37dae1d19bbe6b2d87a Mon Sep 17 00:00:00 2001
From: Erik Reid <erik.reid@geant.org>
Date: Fri, 4 Sep 2020 17:42:19 +0200
Subject: [PATCH] pep8

---
 inventory_provider/routes/common.py   | 5 ++++-
 inventory_provider/routes/poller.py   | 7 ++++---
 test/per_router/test_poller_routes.py | 4 ++--
 test/test_general_poller_routes.py    | 5 +++--
 4 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/inventory_provider/routes/common.py b/inventory_provider/routes/common.py
index ecabd233..275b9765 100644
--- a/inventory_provider/routes/common.py
+++ b/inventory_provider/routes/common.py
@@ -160,7 +160,10 @@ def _redis_client_proc(key_queue, value_queue, config_params, doc_type):
 
 
 def _load_redis_docs(
-        config_params, key_pattern, num_threads=10, doc_type=_DECODE_TYPE_JSON):
+        config_params,
+        key_pattern,
+        num_threads=10,
+        doc_type=_DECODE_TYPE_JSON):
     """
     load all docs from redis and decode as `doc_type`
 
diff --git a/inventory_provider/routes/poller.py b/inventory_provider/routes/poller.py
index 583f2a04..b6f1f66f 100644
--- a/inventory_provider/routes/poller.py
+++ b/inventory_provider/routes/poller.py
@@ -2,7 +2,7 @@ import json
 import logging
 import re
 
-from flask import Blueprint, Response, jsonify, current_app
+from flask import Blueprint, Response, current_app
 from inventory_provider import juniper
 from inventory_provider.routes import common
 
@@ -73,7 +73,7 @@ def _load_services(hostname=None):
 
         m = re.match(r'^opsdb:interface_services:([^:]+):(.+)', doc['key'])
         if not m:
-            logger.warning(f'can''t parse redis service key {doc["key"]}')
+            logger.warning(f'can\'t parse redis service key {doc["key"]}')
             # there are some weird records (dtn*, dp1*)
             continue
 
@@ -84,6 +84,7 @@ def _load_services(hostname=None):
 
     return result
 
+
 def _load_interfaces(hostname):
     key_pattern = f'netconf:{hostname}*' if hostname else 'netconf:*'
     for doc in common.load_xml_docs(
@@ -111,7 +112,7 @@ def _load_poller_interfaces(hostname=None):
 
     snmp_indexes = _load_snmp_indexes(hostname)
     bundles = _load_interface_bundles(hostname)
-    services= _load_services(hostname)
+    services = _load_services(hostname)
 
     for ifc in _load_interfaces(hostname):
 
diff --git a/test/per_router/test_poller_routes.py b/test/per_router/test_poller_routes.py
index 8d380553..6879805e 100644
--- a/test/per_router/test_poller_routes.py
+++ b/test/per_router/test_poller_routes.py
@@ -32,7 +32,7 @@ INTERFACE_LIST_SCHEMA = {
                     'minimum': 1
                 },
                 'bundle': {
-                    'type' : 'array',
+                    'type': 'array',
                     'items': {'type': 'string'}
                 },
                 'bundle-parents': {
@@ -59,7 +59,7 @@ INTERFACE_LIST_SCHEMA = {
 
 def test_router_interfaces(router, client):
     rv = client.post(
-        "/poller/interfaces/" + router,
+        f'/poller/interfaces/{router}',
         headers=DEFAULT_REQUEST_HEADERS)
 
     assert rv.status_code == 200
diff --git a/test/test_general_poller_routes.py b/test/test_general_poller_routes.py
index e14491ab..ada06035 100644
--- a/test/test_general_poller_routes.py
+++ b/test/test_general_poller_routes.py
@@ -59,7 +59,7 @@ INTERFACE_LIST_SCHEMA = {
                     'minimum': 1
                 },
                 'bundle': {
-                    'type' : 'array',
+                    'type': 'array',
                     'items': {'type': 'string'}
                 },
                 'bundle-parents': {
@@ -83,6 +83,7 @@ INTERFACE_LIST_SCHEMA = {
     'items': {'$ref': '#/definitions/interface'}
 }
 
+
 @pytest.mark.parametrize('category', ['mdvpn', 'lhcone', 'MDVpn', 'LHCONE'])
 def test_service_category(client, mocked_worker_module, category):
     worker._build_service_category_interface_list()
@@ -107,7 +108,7 @@ def test_service_category_not_found(client, mocked_worker_module, category):
 
 def test_get_all_interfaces(client):
     rv = client.get(
-        f'/poller/interfaces',
+        '/poller/interfaces',
         headers=DEFAULT_REQUEST_HEADERS)
     assert rv.status_code == 200
     assert rv.is_json
-- 
GitLab