# @Time     :2019/11/26 11:25
# @Author   :dengyuting
# @File     :db_func.py
import psycopg2

# 获得连接
def GetConnect():
    conn = False
    try:
        conn = psycopg2.connect(database="partner_test",
                                    user="bianxinhua_user",
                                    password="Bianxinhua_user#123",
                                    host="101.37.68.23",
                                    port="15432")
    except Exception as err:
        print("连接数据库失败，%s" % err)
    return conn

def get_admin_id_from_database(tbl):
    db = GetConnect()
    cursor = db.cursor()
    sql= "select id from " + tbl + ' ' + 'order by create_at desc '
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()

def get_admin_account_from_database(tbl):
    db = GetConnect()
    cursor = db.cursor()
    sql= "select account from " + tbl  + ' ' + 'order by create_at desc '
    try:
        cursor.execute(sql)
        result = cursor.fetchone()
        return result[0]
    except Exception as err:
        # 发生错误时回滚
        print(err)
        db.rollback()
    db.close()