create docstrings

Author: jonhealy1

data_loader/data_loader.py

@@ -1,16 +1,21 @@
-import click
-import os
 import json
+import os
+
+import click
 import requests
 
 DATA_DIR = os.path.join(os.path.dirname(__file__), "setup_data/")
 STAC_API_BASE_URL = "http://localhost:8083"
 
+
 def load_data(filename):
+    """Load json data."""
     with open(os.path.join(DATA_DIR, filename)) as file:
         return json.load(file)
 
+
 def load_collection(collection_id):
+    """Load stac collection into the database."""
     collection = load_data("collection.json")
     collection["id"] = collection_id
     try:
@@ -24,7 +29,9 @@ def load_collection(collection_id):
     except requests.ConnectionError:
         click.secho("failed to connect")
 
+
 def load_items():
+    """Load stac items into the database."""
     feature_collection = load_data("sentinel-s2-l2a-cogs_0_100.json")
     collection = "test-collection"
     load_collection(collection)
@@ -34,8 +41,10 @@ def load_items():
             feature["stac_extensions"] = []
             feature["stac_version"] = "1.0.0"
             feature["collection"] = "test-collection"
-            resp = requests.post(f"{STAC_API_BASE_URL}/collections/{collection}/items", json=feature)
-            
+            resp = requests.post(
+                f"{STAC_API_BASE_URL}/collections/{collection}/items", json=feature
+            )
+
             if resp.status_code == 200:
                 print(f"Status code: {resp.status_code}")
                 print(f"Added item: {feature['id']}")
@@ -45,4 +54,5 @@ def load_items():
         except requests.ConnectionError:
             click.secho("failed to connect")
 
+
 load_items()