added code to connect to MongoDB

This commit is contained in:
simplypower-bbj
2023-03-16 18:41:54 +01:00
parent 9bde702cc0
commit ebe5ac9edb
+7
View File
@@ -7,6 +7,7 @@ import pandas as pd
import psycopg2 import psycopg2
from dotenv import load_dotenv from dotenv import load_dotenv
from sqlalchemy import create_engine from sqlalchemy import create_engine
from pymongo import MongoClient
class Database(object): class Database(object):
@@ -223,3 +224,9 @@ def get_untranslated(dt):
'content': str(df['content'][0]) 'content': str(df['content'][0])
} }
return val_dict return val_dict
def connect_mongo(ip_addr: str = '192.168.1.2', db_name: str = 'twitter', collection_name: str = 'tweet'):
client = MongoClient(ip_addr)
db = client[db_name]
coll = db[collection_name]
return coll