shaileshjadhavSS commited on
Commit
8537c2d
·
1 Parent(s): bd7635f

Updated DB variables

Browse files
Files changed (2) hide show
  1. settings/base.py +6 -14
  2. user_auth/user_manager.py +2 -3
settings/base.py CHANGED
@@ -2,20 +2,12 @@ import os
2
  import logging
3
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
4
 
5
- DATABASES = {
6
- "PSQL":{
7
- "DB_NAME": os.environ["DB_NAME"],
8
- "DB_USER": os.environ["DB_USER"],
9
- "DB_PASSWORD": os.environ["DB_PASSWORD"],
10
- "DB_HOST": os.environ["DB_HOST"],
11
- "DB_PORT": "5432",
12
- }
13
- }
14
- DB_NAME = os.environ.get("DBNAME")
15
- DB_USER = os.environ.get("DBUSER")
16
- DB_PASSWORD = os.environ.get("DBPASSWORD")
17
- DB_HOST = os.environ.get("DBHOST")
18
- DB_PORT = os.environ.get("DBPORT")
19
 
20
  def setup_logger():
21
  logger = logging.getLogger('app_logger')
 
2
  import logging
3
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
4
 
5
+
6
+ DB_NAME = os.environ.get("DB_NAME")
7
+ DB_USER = os.environ.get("DB_USER")
8
+ DB_PASSWORD = os.environ.get("DB_PASSWORD")
9
+ DB_HOST = os.environ.get("DB_HOST")
10
+ DB_PORT = os.environ.get("DB_PORT")
 
 
 
 
 
 
 
 
11
 
12
  def setup_logger():
13
  logger = logging.getLogger('app_logger')
user_auth/user_manager.py CHANGED
@@ -1,15 +1,14 @@
1
  import psycopg2
2
  from user_auth.user import User
3
- from settings.base import setup_logger, DATABASES
4
 
5
 
6
  logger = setup_logger()
7
 
8
  class UserManager:
9
- PSQL = DATABASES["PSQL"]
10
  def __init__(self):
11
  self.conn = psycopg2.connect(
12
- dbname=self.PSQL["DB_NAME"], user=self.PSQL["DB_USER"], password=self.PSQL["DB_PASSWORD"], host=self.PSQL["DB_HOST"], port=self.PSQL["DB_PORT"]
13
  )
14
  self.cursor = self.conn.cursor()
15
 
 
1
  import psycopg2
2
  from user_auth.user import User
3
+ from settings.base import setup_logger, DB_HOST, DB_NAME, DB_PASSWORD, DB_PORT, DB_USER
4
 
5
 
6
  logger = setup_logger()
7
 
8
  class UserManager:
 
9
  def __init__(self):
10
  self.conn = psycopg2.connect(
11
+ dbname=DB_NAME, user=DB_USER, password=DB_PASSWORD, host=DB_HOST, port=DB_PORT
12
  )
13
  self.cursor = self.conn.cursor()
14