You've already forked Douyin_Web_Live
尝试大改,可读性有点差的版本,基本做到能实时处理弹幕
This commit is contained in:
@ -1,26 +0,0 @@
|
||||
import pymongo
|
||||
|
||||
from config.helper import config
|
||||
|
||||
class MongoStore:
|
||||
def __init__(self):
|
||||
self.client = pymongo.MongoClient(config()['mongo']['uri'])
|
||||
self.db = self.client[config()['mongo']['dbname']]
|
||||
|
||||
def close(self):
|
||||
self.client.close()
|
||||
|
||||
def set_collection(self, collection):
|
||||
self.collection = self.db[collection]
|
||||
|
||||
def replace_one(self, condition, data, upsert=True):
|
||||
return self.collection.replace_one(condition, data, upsert=upsert)
|
||||
|
||||
def insert_one(self, data):
|
||||
return self.collection.insert_one(data)
|
||||
|
||||
def insert_many(self, data):
|
||||
return self.collection.insert_many(data)
|
||||
|
||||
def exists(self, condition):
|
||||
return self.collection.count_documents(condition, limit = 1) != 0
|
Reference in New Issue
Block a user