36 lines
631 B
Python
36 lines
631 B
Python
|
# -*- coding: UTF-8 -*-
|
|||
|
|
"""
|
|||
|
|
@description 工具类，数据库
|
|||
|
|
@author wendell
|
|||
|
|
"""
|
|||
|
|
from pymongo import MongoClient
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_client():
|
|||
|
|
"""
|
|||
|
|
获取连接客户端
|
|||
|
|
:return: 链接客户端
|
|||
|
|
"""
|
|||
|
|
return MongoClient("mongodb://localhost:27017/", connect=False)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_db(db=None):
|
|||
|
|
"""
|
|||
|
|
获取数据库
|
|||
|
|
:param db: 默认链接到python数据库
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
if not db:
|
|||
|
|
db = 'python'
|
|||
|
|
return get_client().get_database(name=db)
|
|||
|
|
|
|||
|
|
|
|||
|
|
def close_client(client):
|
|||
|
|
"""
|
|||
|
|
关闭客户端连接
|
|||
|
|
:param client: 连接客户端
|
|||
|
|
:return:
|
|||
|
|
"""
|
|||
|
|
if client:
|
|||
|
|
client.close()
|