intro pymongo

2021-11-29 18:27:02 +08:00
parent 3fc3035cdf
commit 3f608c3510
30 changed files with 37974 additions and 95 deletions
--- a/store/init.py
+++ b/store/init.py
--- a/store/pycache/init.cpython-39.pyc
+++ b/store/pycache/init.cpython-39.pyc
--- a/store/pycache/mongo.cpython-39.pyc
+++ b/store/pycache/mongo.cpython-39.pyc
--- a/store/mongo.py
+++ b/store/mongo.py
@ -0,0 +1,15 @@
+import pymongo
+
+class MongoStore:
+    def __init__(self):
+        self.client = pymongo.MongoClient("mongodb://localhost:27017/")
+        self.db = self.client['tiktok']
+    
+    def set_collection(self, collection):
+        self.collection = self.db[collection]
+    
+    def insert_one(self, data):
+        return self.collection.insert_one(data)
+
+    def insert_many(self, data):
+        return self.collection.insert_many(data)