def getMongoConn(self, *args, **kwargs):
conf = self.conf.get(kwargs.get("link_type", 'default'))
city = kwargs.get("city")
host = conf.get("host")
port = conf.get("port")
user = conf.get("username", "")
passwd = conf.get("password", "")
db_name = kwargs.get("db_name")
self.db_name = db_name if db_name else get_db(type=self.conf_name,
city=city)
if user and passwd:
uri = "mongodb://{username}:{password}@{host}:{port}".\
format(username=user, password=passwd, host=host, port= port)
else:
uri = "mongodb://{host}:{port}".format(host=host, port=port)
conn = MongoClient(uri)
try:
client = conn.get_database(self.db_name)
yield client
except Exception as e:
logging.error('ERROR', e)
finally:
# conn.close()
pass
def get_conn(self, *args, **kwargs):
conn = MysqlDB.__pool.get(self.conf_name)
if conn:
conn.ping()
else:
#获取主从连接
city = kwargs.get("city")
conf_name = getConfigName(city=city, type=self.conf_name)
conf1 = mysql_conf.get(conf_name)
conf = conf1.get(kwargs.get("link_type", 'default'))
host = conf.get("host")
port = conf.get("port")
user = conf.get("user")
passwd = conf.get("passwd")
db_name = get_db(type=conf_name, city=city)
conn = pymysql.connect(host=host, port=port, user=user, password=passwd, db=db_name, charset="utf8")
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
try:
yield cursor
conn.commit()
except Exception as e:
conn.rollback()
logging.error(e)
finally:
cursor.close()
conn.close()
def get_conn(self, *args, **kwargs):
city = kwargs.get("city")
self.conf_name = getConfigName(city=city, type=self.conf_name)
conf = self.conf.get(kwargs.get("link_type", 'default'))
# 获取主从连接
host = conf.get("host")
port = conf.get("port")
user = conf.get("user")
passwd = conf.get("passwd")
db_name = get_db(type=self.conf_name, city=city)
conn = pymysql.connect(host=host,
port=port,
user=user,
password=passwd,
db=db_name,
charset="utf8")
cursor = conn.cursor(cursor=pymysql.cursors.DictCursor)
try:
yield cursor
logging.info(cursor._last_executed)
except Exception as e:
logging.error(e)
finally:
conn.commit()
cursor.close()
conn.close()
def get_connection(self, *args, **kwargs):
# 获取主从连接
conf = self.conf.get(kwargs.get("link_type", 'default'))
city = kwargs.get("city")
host = conf.get("host")
port = conf.get("port")
user = conf.get("user")
passwd = conf.get("passwd")
db_name = get_db(type=self.conf_name, city=city)
conn = pymysql.connect(host=host, port=port, user=user, password=passwd, db=db_name, charset="utf8")
return conn
