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

Update db_operations/db_write.py

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