Update db_operations/db_write.py
Browse files
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 =
|
12 |
-
self.collection =
|
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 |
|