jsdio/voice_assistant
Template
1
0
Fork 0
mirror of https://github.com/ijaric/voice_assistant.git synced 2025-10-30 23:23:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: [#38] socks5 is possible

Browse Source
This commit is contained in:
ksieuk 2023-10-07 01:54:50 +03:00
parent 468a73d380
commit fa601a3eca
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/assistant/lib/app/split_settings/proxy.py
View File

@ -1,3 +1,5 @@
import typing
import pydantic import pydantic
import pydantic_settings import pydantic_settings
@ -11,6 +13,7 @@ class ProxySettings(pydantic_settings.BaseSettings):
env_file_encoding="utf-8", env_file_encoding="utf-8",
extra="ignore", extra="ignore",
) )
protocol: typing.Literal["http", "socks5"] = "http"
user: str | None = None user: str | None = None
password: pydantic.SecretStr | None = None password: pydantic.SecretStr | None = None
host: str | None = None host: str | None = None
@ -21,15 +24,15 @@ class ProxySettings(pydantic_settings.BaseSettings):
def dsn(self) -> str: def dsn(self) -> str:
if self.user and self.password: if self.user and self.password:
password = self.password.get_secret_value() password = self.password.get_secret_value()
return f"http://{self.user}:{password}@{self.host}:{self.port}" return f"{self.protocol}://{self.user}:{password}@{self.host}:{self.port}"
return f"http://{self.host}:{self.port}" return f"{self.protocol}://{self.host}:{self.port}"
@pydantic.computed_field @pydantic.computed_field
@property @property
def dsn_as_safe_url(self) -> str: def dsn_as_safe_url(self) -> str:
if self.user and self.password: if self.user and self.password:
return f"http://{self.user}:{self.password}@{self.host}:{self.port}" return f"{self.protocol}://{self.user}:{self.password}@{self.host}:{self.port}"
return f"http://{self.host}:{self.port}" return f"{self.protocol}://{self.host}:{self.port}"
@pydantic.model_validator(mode="after") @pydantic.model_validator(mode="after")
def check_proxy(self): def check_proxy(self):