pre-commit

Author: jonhealy1

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/config.py

@@ -18,27 +18,30 @@ def _es_config() -> Dict[str, Any]:
     # Initialize the configuration dictionary
     config = {
         "hosts": hosts,
-        "headers": {"accept": "application/vnd.elasticsearch+json; compatible-with=7"}
+        "headers": {"accept": "application/vnd.elasticsearch+json; compatible-with=7"},
     }
 
     # Explicitly exclude SSL settings when not using SSL
     if not use_ssl:
         return config
 
     # Include SSL settings if using https
-    config["ssl_version"] = ssl.TLSVersion.TLSv1_2
-    config["verify_certs"] = os.getenv("ES_VERIFY_CERTS", "true").lower() != "false"
+    config["ssl_version"] = ssl.TLSVersion.TLSv1_2  # type: ignore
+    config["verify_certs"] = os.getenv("ES_VERIFY_CERTS", "true").lower() != "false"  # type: ignore
 
     # Include CA Certificates if verifying certs
     if config["verify_certs"]:
-        config["ca_certs"] = os.getenv("CURL_CA_BUNDLE", "/etc/ssl/certs/ca-certificates.crt")
+        config["ca_certs"] = os.getenv(
+            "CURL_CA_BUNDLE", "/etc/ssl/certs/ca-certificates.crt"
+        )
 
     # Handle authentication
     if (u := os.getenv("ES_USER")) and (p := os.getenv("ES_PASS")):
         config["http_auth"] = (u, p)
 
     return config
 
+
 _forbidden_fields: Set[str] = {"type"}
 
 

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/database_logic.py

@@ -170,12 +170,19 @@ def indices(collection_ids: Optional[List[str]]) -> str:
 
 
 async def create_collection_index() -> None:
+    """
+    Create the index for a Collection.
+
+    Returns:
+        None
+
+    """
     client = AsyncElasticsearchSettings().create_client
 
     await client.options(ignore_status=400).indices.create(
         index=f"{COLLECTIONS_INDEX}-000001",
         aliases={COLLECTIONS_INDEX: {}},
-        mappings=ES_COLLECTIONS_MAPPINGS
+        mappings=ES_COLLECTIONS_MAPPINGS,
     )
     await client.close()