lalithadevi commited on
Commit
3d597fd
1 Parent(s): 214c1ff

Update db_operations/db_read.py

Browse files
Files changed (1) hide show
  1. db_operations/db_read.py +3 -2
db_operations/db_read.py CHANGED
@@ -1,6 +1,7 @@
1
  import pymongo
2
  import os
3
  import pandas as pd
 
4
 
5
 
6
  class DBRead:
@@ -9,8 +10,8 @@ class DBRead:
9
  """
10
  def __init__(self):
11
  self.url = os.getenv('DB_URL')
12
- self.database = "rss_news_db"
13
- self.collection = "rss_news"
14
  self.__client = None
15
  self.__error = 0
16
 
 
1
  import pymongo
2
  import os
3
  import pandas as pd
4
+ from config import DATABASE, COLLECTION
5
 
6
 
7
  class DBRead:
 
10
  """
11
  def __init__(self):
12
  self.url = os.getenv('DB_URL')
13
+ self.database = DATABASE
14
+ self.collection = COLLECTION
15
  self.__client = None
16
  self.__error = 0
17