Pre-commit issues

src/pe_reports/data/db_query.py

@@ -1,18 +1,18 @@
 """Query the PE PostgreSQL database."""
 
 # Standard Python Libraries
+import json
 import logging
 import re
 import sys
 
 # Third-Party Libraries
-import requests
-import json
 import numpy as np
 import pandas as pd
 import psycopg2
 from psycopg2 import OperationalError
 from psycopg2.extensions import AsIs
+import requests
 
 from .config import config
 
@@ -22,8 +22,9 @@
 LOGGER = logging.getLogger(__name__)
 
 CONN_PARAMS_DIC = config()
-PE_API_KEY = config(section='peapi').get('api_key') 
-PE_API_URL = config(section='peapi').get('api_url')
+PE_API_KEY = config(section="peapi").get("api_key")
+PE_API_URL = config(section="peapi").get("api_url")
+
 
 def sanitize_string(string):
     """Remove special characters from string."""
@@ -35,9 +36,6 @@ def sanitize_uid(string):
     return re.sub(r"[^a-zA-Z0-9\-\s]", "", string)
 
 
-
-
-
 def show_psycopg2_exception(err):
     """Handle errors for PostgreSQL issues."""
     err_type, err_obj, traceback = sys.exc_info()
@@ -62,11 +60,12 @@ def close(conn):
     conn.close()
     return
 
+
 def get_orgs():
     """Query organizations table."""
     headers = {
         "Content-Type": "application/json",
-        "access_token": f'{PE_API_KEY}',
+        "access_token": f"{PE_API_KEY}",
     }
     try:
         response = requests.post(PE_API_URL, headers=headers).json()

src/pe_source/data/pe_db/db_query_source.py

@@ -1,18 +1,17 @@
 """Query the PE PostgreSQL database."""
 
 # Standard Python Libraries
-from datetime import datetime
+import json
 import re
-import requests
 import sys
-import json
 import time
 
 # Third-Party Libraries
 import pandas as pd
 import psycopg2
 from psycopg2 import OperationalError
 import psycopg2.extras as extras
+import requests
 
 # cisagov Libraries
 from pe_reports import app
@@ -64,7 +63,7 @@ def get_orgs():
         "access_token": f"{PE_API_KEY}",
     }
     try:
-        response = requests.post(PE_API_URL +  "get_orgs/", headers=headers).json()
+        response = requests.post(PE_API_URL + "get_orgs/", headers=headers).json()
         return response
     except requests.exceptions.HTTPError as errh:
         print(errh)
@@ -93,7 +92,7 @@ def get_ips(org_uid):
 
 
 def get_data_source_uid(source):
-    """Query organizations table."""
+    """Query data_source table and update the last viewed time."""
     urlOrgs = PE_API_URL
     headers = {
         "Content-Type": "application/json",
@@ -105,7 +104,7 @@ def get_data_source_uid(source):
         ).json()
         # Change last viewed
         uid = response[0]["data_source_uid"]
-        r = requests.put(urlOrgs + "update_last_viewed/" + uid, headers=headers)
+        requests.put(urlOrgs + "update_last_viewed/" + uid, headers=headers)
         LOGGER.info("Updated last viewed for %s", source)
         return response
     except requests.exceptions.HTTPError as errh: