lalithadevi commited on
Commit
074c9bf
1 Parent(s): 16d3623

Update db_operations/db_read.py

Browse files
Files changed (1) hide show
  1. db_operations/db_read.py +4 -4
db_operations/db_read.py CHANGED
@@ -1,7 +1,7 @@
1
  import pymongo
2
  import os
3
  import pandas as pd
4
- from config import DATABASE, COLLECTION
5
  from logger import get_logger
6
 
7
 
@@ -12,9 +12,9 @@ class DBRead:
12
  Reads news from MongoDB
13
  """
14
  def __init__(self):
15
- self.url = os.getenv('DB_URL')
16
- self.database = DATABASE
17
- self.collection = COLLECTION
18
  self.__client = None
19
  self.__error = 0
20
 
 
1
  import pymongo
2
  import os
3
  import pandas as pd
4
+ from config import PRODUCTION_URL, PRODUCTION_DATABASE, PRODUCTION_COLLECTION
5
  from logger import get_logger
6
 
7
 
 
12
  Reads news from MongoDB
13
  """
14
  def __init__(self):
15
+ self.url = PRODUCTION_URL
16
+ self.database = PRODUCTION_DATABASE
17
+ self.collection = PRODUCTION_COLLECTION
18
  self.__client = None
19
  self.__error = 0
20