mongo_helper

# -*- coding:utf-8 -*-
import os
import configparser
import pymongo
class MongoHelper:
def __init__(self, cOnf=None):
# conf.keys = [mongo_ini, deploy]
self.cOnf= conf
self.mongo_cOnf= None
self.mongo_client = None
self.read_conf()
def read_conf(self):
self.mongo_cOnf= configparser.ConfigParser()
self.mongo_conf.read(self.conf['mongo_ini'], encoding="utf-8")
self.deploy = self.conf['deploy']
def mongo_connet(self):
port_hort_list = self.mongo_conf.get(self.deploy, 'host_port').split('|')
return pymongo.MongoClient(port_hort_list, replicaSet='rs0')
def connet_retry(self, retry=3):
mongo_client, flag = None, False
for _ in range(retry):
try:
mongo_client = self.mongo_connet()
flag = True
except Exception :
#pass
import traceback
traceback.print_exc()
if flag: break
return mongo_client
def check_connet(self, mongo_client, retry=3):
if mongo_client is None:
mongo_client = self.connet_retry(retry)
else:
try:
mongo_client.server_info()
except Exception :
mongo_client = self.connet_retry(retry)
return mongo_client

def update_data_by_one(self, filter_dict, update_data, db_name=None, col_name=None):
cOnnet= self.check_connet(self.mongo_client)
if connet is None:
return ret_value
if db_name is None:
db_name = self.mongo_conf.get(self.deploy, 'db_name')
if col_name is None:
col_name = self.mongo_conf.get(self.deploy, 'collection')
db = connet[db_name]
collection = db[col_name]
collection.update_one(filter_dict, {'$set': update_data})
def mongo_find(self, condention_dict, db_name=None, col_name=None):
querys = None
cOnnet= self.check_connet(self.mongo_client)
if connet is None:
return querys
if db_name is None:
db_name = self.mongo_conf.get(self.deploy, 'db_name')
if col_name is None:
col_name = self.mongo_conf.get(self.deploy, 'collection')
db = connet[db_name]
collection = db[col_name]
if condention_dict is None:
querys = collection.find({})
else:
if condention_dict and isinstance(condention_dict, dict):
querys = collection.find(condention_dict)
return querys