我有一个从MongoDB返回数据库连接处理程序的函数.我有各种其他函数调用DB,我想让连接处理程序抛出一个函数,所以我不必在每个函数中定义它.

这看起来不错吗？我想我的问题是,如果它无法连接到数据库服务器,它将打印两个消息无法连接到服务器并且找不到主机如何才能打印“无法连接到服务器”.

def mongodb_conn():
    try:
        conn = pymongo.MongoClient()
    except pymongo.errors.ConnectionFailure, e:
        print "Could not connect to server: %s" % e
    return conn

def get_hosts()
    try:
        conn = mongodb_conn()
        mongodb = conn.dbname.collection

        b = []
        hosts_obj = mongodb.find({'_id': 'PR'})
        for x in hosts_obj:
            print x
    except:
        print "No hosts found"

get_hosts()

解决方法:

将conn = mongodb_conn()调用移出try …除处理程序,并测试是否返回None：

def get_hosts()
    conn = mongodb_conn()
    if conn is None:
        # no connection, exit early
        return

    try:
        mongodb = conn.dbname.collection

        b = []
        hosts_obj = mongodb.find({'_id': 'PR'})
        for x in hosts_obj:
            print x
    except:
        print "No hosts found"

你应该不惜一切代价避免使用毯子;您现在正在捕捉所有内容,包括内存错误和键盘中断,请参阅Why is “except: pass” a bad programming practice?

仅使用特定的例外;您可以使用一个except语句来捕获多个异常类型：

except (AttributeError, pymongo.errors.OperationFailure):

或者你可以使用多个except语句以不同的方式处理不同的异常.

将异常处理程序限制为可以抛出异常的代码部分.例如,hosts_obj：循环中的for x可能不会抛出AttributeError异常,因此它可能不应该是try块的一部分.

请注意,如果从未设置过,则需要调整mongodb_conn()函数,以免尝试使用conn local;如果您这样做,您将收到UnboundLocal错误：

def mongodb_conn():
    try:
        return pymongo.MongoClient()
    except pymongo.errors.ConnectionFailure, e:
        print "Could not connect to server: %s" % e

现在,如果成功,函数将返回连接,如果连接失败,则返回None.