Skip to content

Commit 3c1b58b

Browse files
committed
Fix Kafka.SSLProtocol aliases shown by VSCode
1 parent 8942454 commit 3c1b58b

File tree

1 file changed

+19
-10
lines changed

1 file changed

+19
-10
lines changed

onetl/connection/db_connection/kafka/kafka_ssl_protocol.py

Lines changed: 19 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,7 @@
1010
except (ImportError, AttributeError):
1111
from pydantic import Field, SecretStr, validator # type: ignore[no-redef, assignment]
1212

13+
from onetl._util.alias import avoid_alias
1314
from onetl._util.file import is_file_readable
1415
from onetl._util.spark import stringify
1516
from onetl.impl import GenericOptions, LocalPath
@@ -148,17 +149,25 @@ class KafkaSSLProtocol(KafkaProtocol, GenericOptions):
148149
)
149150
"""
150151

151-
keystore_type: Optional[str] = Field(default=None, alias="ssl.keystore.type")
152-
keystore_location: Optional[LocalPath] = Field(default=None, alias="ssl.keystore.location")
153-
keystore_password: Optional[SecretStr] = Field(default=None, alias="ssl.keystore.password")
154-
keystore_certificate_chain: Optional[str] = Field(default=None, alias="ssl.keystore.certificate.chain", repr=False)
155-
keystore_key: Optional[SecretStr] = Field(default=None, alias="ssl.keystore.key")
152+
keystore_type: Optional[str] = Field(default=None, alias=avoid_alias("ssl.keystore.type"))
153+
keystore_location: Optional[LocalPath] = Field(default=None, alias=avoid_alias("ssl.keystore.location"))
154+
keystore_password: Optional[SecretStr] = Field(default=None, alias=avoid_alias("ssl.keystore.password"))
155+
keystore_certificate_chain: Optional[str] = Field(
156+
default=None,
157+
alias=avoid_alias("ssl.keystore.certificate.chain"),
158+
repr=False,
159+
)
160+
keystore_key: Optional[SecretStr] = Field(default=None, alias=avoid_alias("ssl.keystore.key"))
156161
# https://knowledge.informatica.com/s/article/145442?language=en_US
157-
key_password: Optional[SecretStr] = Field(default=None, alias="ssl.key.password")
158-
truststore_type: str = Field(alias="ssl.truststore.type")
159-
truststore_location: Optional[LocalPath] = Field(default=None, alias="ssl.truststore.location")
160-
truststore_password: Optional[SecretStr] = Field(default=None, alias="ssl.truststore.password")
161-
truststore_certificates: Optional[str] = Field(default=None, alias="ssl.truststore.certificates", repr=False)
162+
key_password: Optional[SecretStr] = Field(default=None, alias=avoid_alias("ssl.key.password"))
163+
truststore_type: str = Field(alias=avoid_alias("ssl.truststore.type"))
164+
truststore_location: Optional[LocalPath] = Field(default=None, alias=avoid_alias("ssl.truststore.location"))
165+
truststore_password: Optional[SecretStr] = Field(default=None, alias=avoid_alias("ssl.truststore.password"))
166+
truststore_certificates: Optional[str] = Field(
167+
default=None,
168+
alias=avoid_alias("ssl.truststore.certificates"),
169+
repr=False,
170+
)
162171

163172
class Config:
164173
known_options = frozenset(("ssl.*",))

0 commit comments

Comments
 (0)