获取pg某个数据库中总行数

湖魂

已于 2023-02-22 21:01:37 修改

阅读量483

点赞数

分类专栏： postgresql 文章标签： postgresql Powered by 金山文档

于 2023-02-22 21:01:21 首次发布

本文链接：https://blog.csdn.net/Woody453253/article/details/129170545

版权

postgresql 专栏收录该内容

10 篇文章 0 订阅

订阅专栏

import psycopg2
USER = "user"
PASSWORD = "123123123"
PORT = 5432
HOST = "localhost"
DB = "imdb"
CONNECTION_STR = "dbname=" + DB + " user=" + USER + " password=" + PASSWORD + " host=localhost port=" + str(PORT)
get_tables = "select tablename from pg_tables where schemaname=\'public\';"


def run_query(q,):
    conn = psycopg2.connect(CONNECTION_STR)
    conn.set_client_encoding('UTF8')
    result = None
    cur = conn.cursor()
    cur.execute(q)
    result = cur.fetchall()
    return result

CONNECTION_STR = "dbname=" + DB + " user=" + USER + " password=" + PASSWORD + " host=localhost port=" + str(PORT)

# 获取所有表名
table_list1 = run_query(get_tables)
table_list1 = [x[0] for x in table_list1]

# 获取每个表的行数
row_list1 = [run_query("select count(*) from " + x +";") for x in table_list1]
row_list1 = [x[0][0] for x in row_list1]
total_row1 = sum(row_list1)