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)
python调用postgres
524 views