python调用postgres


import json

import psycopg2

USER = "user"
PASSWORD = "password"
PORT = 5432
HOST = "10.0.0.17"
DATABASE = "unittest"


def call_postgresql(sql, flag=1):
    # "flag = 1表示获取多条数据 0表示获取单条数据"

    conn = psycopg2.connect(database=DATABASE, port=PORT,
                            host=HOST, user=USER, password=PASSWORD)
    cursor = conn.cursor()
    cursor.execute(sql)
    if flag:
        results = cursor.fetchall()
    else:
        results = None
    conn.commit()
    conn.close()
    return results

if __name__ == "__main__":
    sql = "select * from unittest_result"
    call_postgresql(sql)