postgres_query.py 文件源码

def main():
    """The main program"""
    args = parse_args()
    items = []
    now = str(int(time()))

    with connect(database=args.dbname) as conn:
        conn.set_session(
            isolation_level=ISOLATION_LEVEL_REPEATABLE_READ,
            readonly=True,
        )

        for query in args.queries:
            rows = execute(conn, query)
            if len(rows) == 0:
                raise Exception('No result')
            items.extend(
                get_row_data(args.key_column, rows) if args.key_column
                else get_column_data(rows)
            )

    for key, value in items:
        print(args.prefix + '.' + key, value, now)