Coverage for apps/outers/settings/one_datastore_setting.py: 93%

1from typing import Optional, Any 

2 

3from apps.outers.settings.base_setting import BaseSetting 

4 

5 

6class OneDatastoreSetting(BaseSetting): 

7 DS_ONE_HOST: str 

8 DS_ONE_PORT: str 

9 DS_ONE_USER: str 

10 DS_ONE_PASSWORD: str 

11 DS_ONE_DATABASE: str 

12 URL: Optional[str] = None 

13 

14 def __init__(self, *args: Any, **kwargs: Any) -> None: 

15 super().__init__(*args, **kwargs) 

16 if self.DS_ONE_PASSWORD == "": 

17 self.URL = f"cockroachdb+asyncpg://{self.DS_ONE_USER}@{self.DS_ONE_HOST}:{self.DS_ONE_PORT}/{self.DS_ONE_DATABASE}" 

18 else: 

19 self.URL = f"cockroachdb+asyncpg://{self.DS_ONE_USER}:{self.DS_ONE_PASSWORD}@{self.DS_ONE_HOST}:{self.DS_ONE_PORT}/{self.DS_ONE_DATABASE}"