diff --git a/database.py b/database.py new file mode 100644 index 0000000..073a677 --- /dev/null +++ b/database.py @@ -0,0 +1,7 @@ +from pymongo import MongoClient + +def connect(ip_addr: str = '192.168.1.2', db_name: str = 'twitter', collection_name: str = 'tweet'): + client = MongoClient(ip_addr) + db = client[db_name] + collection = db[collection_name] + return collection \ No newline at end of file