Psycopg2 Postgre连接重试

时间:2015-07-16 07:10:28

标签: python database postgresql heroku psycopg2

我正在尝试用Python编写一些小型Web服务。我使用Heroku和他们的postgre-DB服务(免费)。

但是我遇到了一个很小但很烦人的问题。当我尝试在数据库中搜索某些内容时,程序会连接到数据库,但会继续尝试,即使它第一次运行。

调用search_image函数的部分:

def handle_send(update):
    link = databasecon.search_image(update["message"]["text"], update)

connect_to_database函数:

def connect_to_db():
    global __is_connected
    if "DATABASE_URL" not in os.environ or __is_connected == True:
        print("Environment-variable missing or already connected")
    else:
        urllib.parse.uses_netloc.append("postgres")
        url = urllib.parse.urlparse(os.environ["DATABASE_URL"])
        con = psycopg2.connect(
            database=url.path[1:],
            user=url.username,
            password=url.password,
            host=url.hostname,
            port=url.port
            )
        if con != None:
            __is_connected = True
    return con

search_image函数:

def search_image(image_name, update):
    db_con = connect_to_db()
    cur = db_con.cursor()
    query = """select link from mm_image where id=%s"""
    cur.execute(query, (image_name))
    result = cur.fetchone()
    if result != None:
        image_link = str(result[0])
        disconnect_from_db(db_con)
        return image_link
    else:
        disconnect_from_db(db_con)
        return "Not found"

以下是日志的显示方式,因为handle_send函数被调用: http://i.stack.imgur.com/pTZcF.png

这里有什么问题?
这是我用Python编写的第一个正确的程序,所以如果这是一个明显的错误,我很抱歉:S

2 个答案:

答案 0 :(得分:0)

该行:

 if "DATABASE_URL" not in os.environ or __is_connected == True:
        print("Environment-variable missing or already connected")
    else:

...将检查DATABASE_URL是否不存在,然后检查__is_connected == True(第一次连接后 为真)。

因此,在每次后续检查中,您的程序将转到print("Environment-variable missing or already connected"),因为每次运行都会评估if __is_connected == True

你应该考虑进一步分解它:

def connect_to_db():
    global __is_connected
    if __is_connected == True:
        break;
    elif "DATABASE_URL" not in os.environ:
        print("DATABASE_URL not set")
    else:
        print("
        urllib.parse.uses_netloc.append("postgres")
        url = urllib.parse.urlparse(os.environ["DATABASE_URL"])
        con = psycopg2.connect
        // etc.

答案 1 :(得分:0)

好的,我发现了错误。执行函数需要一个元组作为第二个参数。

从官方使用文档:

  

对于位置变量绑定,第二个参数必须始终是序列,即使它包含单个变量。请记住,Python需要一个逗号来创建单个元素元组