We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7aff6b0 commit 3d36cb3Copy full SHA for 3d36cb3
1 file changed
lib/dl_connector_promql/dl_connector_promql/core/us_connection.py
@@ -13,6 +13,7 @@
13
ConnectionBase,
14
)
15
from dl_core.utils import secrepr
16
+from dl_utils.utils import DataKey
17
18
from dl_connector_promql.core.constants import (
19
SOURCE_TYPE_PROMQL,
@@ -34,6 +35,13 @@ class DataModel(ClassicConnectionSQL.DataModel):
34
35
auth_type: PromQLAuthType = attr.ib()
36
auth_header: str | None = attr.ib(repr=secrepr, default=None)
37
38
+ @classmethod
39
+ def get_secret_keys(cls) -> set[DataKey]:
40
+ return {
41
+ *super().get_secret_keys(),
42
+ DataKey(parts=("auth_header",)),
43
+ }
44
+
45
def get_conn_dto(self) -> PromQLConnDTO:
46
return PromQLConnDTO(
47
conn_id=self.uuid,
0 commit comments