Skip to content

Commit f1c38d8

Browse files
committed
log async
1 parent aa32918 commit f1c38d8

File tree

2 files changed

+23
-19
lines changed
  • stac_fastapi
    • elasticsearch/stac_fastapi/elasticsearch
    • opensearch/stac_fastapi/opensearch

2 files changed

+23
-19
lines changed

stac_fastapi/elasticsearch/stac_fastapi/elasticsearch/config.py

Lines changed: 11 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -92,14 +92,6 @@ def create_client(self):
9292
return Elasticsearch(**_es_config())
9393

9494

95-
# Warn at import if direct response is enabled
96-
if ElasticsearchSettings().enable_direct_response:
97-
logging.basicConfig(level=logging.WARNING)
98-
logging.warning(
99-
"ENABLE_DIRECT_RESPONSE is True: All FastAPI dependencies (including authentication) are DISABLED for all routes!"
100-
)
101-
102-
10395
class AsyncElasticsearchSettings(ApiSettings, ApiBaseSettings):
10496
"""
10597
API settings.
@@ -118,3 +110,14 @@ class AsyncElasticsearchSettings(ApiSettings, ApiBaseSettings):
118110
def create_client(self):
119111
"""Create async elasticsearch client."""
120112
return AsyncElasticsearch(**_es_config())
113+
114+
115+
# Warn at import if direct response is enabled (applies to either settings class)
116+
if (
117+
ElasticsearchSettings().enable_direct_response
118+
or AsyncElasticsearchSettings().enable_direct_response
119+
):
120+
logging.basicConfig(level=logging.WARNING)
121+
logging.warning(
122+
"ENABLE_DIRECT_RESPONSE is True: All FastAPI dependencies (including authentication) are DISABLED for all routes!"
123+
)

stac_fastapi/opensearch/stac_fastapi/opensearch/config.py

Lines changed: 12 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -90,14 +90,6 @@ def create_client(self):
9090
return OpenSearch(**_es_config())
9191

9292

93-
# Warn at import if direct response is enabled (applies to both settings)
94-
if OpensearchSettings().enable_direct_response:
95-
logging.basicConfig(level=logging.WARNING)
96-
logging.warning(
97-
"ENABLE_DIRECT_RESPONSE is True: All FastAPI dependencies (including authentication) are DISABLED for all routes!"
98-
)
99-
100-
10193
class AsyncOpensearchSettings(ApiSettings, ApiBaseSettings):
10294
"""
10395
API settings.
@@ -110,11 +102,20 @@ class AsyncOpensearchSettings(ApiSettings, ApiBaseSettings):
110102
forbidden_fields: Set[str] = _forbidden_fields
111103
indexed_fields: Set[str] = {"datetime"}
112104
enable_response_models: bool = False
113-
enable_direct_response: bool = (
114-
os.getenv("ENABLE_DIRECT_RESPONSE", "false").lower() == "true"
115-
)
105+
enable_direct_response: bool = get_bool_env("ENABLE_DIRECT_RESPONSE", default=False)
116106

117107
@property
118108
def create_client(self):
119109
"""Create async elasticsearch client."""
120110
return AsyncOpenSearch(**_es_config())
111+
112+
113+
# Warn at import if direct response is enabled (applies to either settings class)
114+
if (
115+
OpensearchSettings().enable_direct_response
116+
or AsyncOpensearchSettings().enable_direct_response
117+
):
118+
logging.basicConfig(level=logging.WARNING)
119+
logging.warning(
120+
"ENABLE_DIRECT_RESPONSE is True: All FastAPI dependencies (including authentication) are DISABLED for all routes!"
121+
)

0 commit comments

Comments
 (0)