From 696b21422b333dafb5b4a1c522c163218007ce59 Mon Sep 17 00:00:00 2001 From: James Jung Date: Thu, 13 Oct 2022 02:14:36 +0900 Subject: [PATCH] feat: [OD15-13] Set local DB_HOST to localhost --- app/core/config.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/app/core/config.py b/app/core/config.py index 7165998..11126e0 100644 --- a/app/core/config.py +++ b/app/core/config.py @@ -4,12 +4,16 @@ from pydantic import BaseSettings from dotenv import dotenv_values -env = dotenv_values('.env') +if dotenv_values('.env'): + env = dotenv_values('.env') + env['DB_HOST'] = '127.0.0.1' +else: + env = os.environ DB_URI = 'postgresql://' + \ - f'{os.environ.get("DB_USERNAME")}:{os.environ.get("DB_PASSWORD")}' + \ - f'@{os.environ.get("DB_HOST")}:{os.environ.get("DB_PORT")}/{os.environ.get("DB_NAME")}' + f'{env.get("DB_USERNAME")}:{env.get("DB_PASSWORD")}' + \ + f'@{env.get("DB_HOST")}:{env.get("DB_PORT")}/{env.get("DB_NAME")}' class Settings(BaseSettings):