|
1 | | -import os |
2 | 1 | from dataclasses import dataclass |
3 | 2 |
|
4 | | -from dotenv import load_dotenv |
5 | 3 | from pydantic import BaseModel |
6 | | - |
7 | | -load_dotenv() |
| 4 | +import yaml |
8 | 5 |
|
9 | 6 |
|
10 | 7 | class LogConfig(BaseModel): |
@@ -35,22 +32,21 @@ class LogConfig(BaseModel): |
35 | 32 | LOGGER_NAME: {"handlers": ["default"], "level": LOG_LEVEL}, |
36 | 33 | } |
37 | 34 |
|
| 35 | +CONFIG = yaml.safe_load(open(".config.yml")) |
38 | 36 |
|
39 | 37 | @dataclass |
40 | 38 | class DBConfig(object): |
41 | 39 | """Database configuration to be set for the server""" |
42 | | - |
43 | | - DB_USER = os.getenv("DB_USER") |
44 | | - DB_PASSWORD = os.getenv("DB_PASSWORD") |
45 | | - DB_HOST = os.getenv("DB_HOST") |
46 | | - DB_PORT = os.getenv("DB_PORT") |
47 | | - DB_NAME = os.getenv("DB_NAME") |
| 40 | + DB_USER = CONFIG.get("groceror").get("db").get("DB_USER") |
| 41 | + DB_PASSWORD = CONFIG.get("groceror").get("db").get("DB_PASSWORD") |
| 42 | + DB_HOST = CONFIG.get("groceror").get("db").get("DB_HOST") |
| 43 | + DB_PORT = CONFIG.get("groceror").get("db").get("DB_PORT") |
| 44 | + DB_NAME = CONFIG.get("groceror").get("db").get("DB_USER") |
48 | 45 | DB_URL = f"postgresql://{DB_USER}:{DB_PASSWORD}@{DB_HOST}:{DB_PORT}/{DB_NAME}" |
49 | 46 |
|
50 | 47 |
|
51 | 48 | @dataclass |
52 | 49 | class JWTConfig(object): |
53 | 50 | """JWT related configuration""" |
54 | | - |
55 | | - JWT_ALGORITHM = os.getenv("JWT_ALGORITHM") |
56 | | - JWT_SECRET_KEY = os.getenv("JWT_SECRET_KEY") |
| 51 | + JWT_ALGORITHM = CONFIG.get("groceror").get("JWT_ALGORITHM") |
| 52 | + JWT_SECRET_KEY = CONFIG.get("groceror").get("JWT_SECRET_KEY") |
0 commit comments