ftp_dlink_eu_downloader.py 文件源码

def psql(query:str, var=None):
    import psycopg2
    try:
        db2=psycopg2.connect(database='firmware', user='firmadyne',
                             password='firmadyne', host='127.0.0.1')
        cur = db2.cursor()
        cur.execute(query,var)
        if not query.startswith('SELECT'):
            db2.commit()
            return
        else:
            rows = cur.fetchall()
            return rows
    except psycopg2.Error as ex:
        print(ex, file=sys.stderr)
        db2.rollback()
        raise ex
    finally:
        db2.close()