Python统计数据库中的数据量【含MySQL、Oracle】

 

Python程序文件如下：

# -*- coding: utf-8
# File : start.py 
# Author : baoshan
import json
import pymysql
import cx_Oracle
import pandas as pd


def main():
    dataSum = []
    connInfo = "connInfo.json"  # 配置文件名称

    connFile = open(connInfo, ‘r‘, encoding=‘utf8‘)
    connRecords = connFile.read(102400)  #一次读取多个字节
    connRecordsjs = json.loads(connRecords)
    for single in connRecordsjs:
        if "mysql" == single.get("dbtype"):
            conn = pymysql.connect(host=single.get("host"), port=single.get("port"), user=single.get("user"),
                                   passwd=single.get("passwd"), charset=single.get("charset"))
            if "gongxiangwangzhan" == single.get("source", "0"):  # 共享网站 *局、民政局、聊城市发展和改革委员会 定制
                sql = "select table_schema as ‘数据库‘, "                       "table_name as ‘数据表‘, "                       "TABLE_COMMENT as ‘表注释‘, "                       "round(data_length/1024/1024,2) as ‘数据大小(M)‘, "                       "round(index_length/1024/1024,2) as ‘索引大小（M）‘, "                       "TABLE_ROWS as ‘行数‘ "                       "from information_schema.tables "                       "where TABLE_SCHEMA in (‘" single.get("dbschema") "‘) "                       "AND TABLE_ROWS > 0 "                       "and table_name in " single.get("selectkeystr") ""
            else:
                sql = "select "                       "table_schema as ‘数据库‘,"                       "table_name as ‘数据表‘, "                       "TABLE_COMMENT as ‘表注释‘, "                       "round(data_length/1024/1024,2) as ‘数据大小(M)‘, "                       "round(index_length/1024/1024,2) as ‘索引大小（M）‘, "                       "TABLE_ROWS as ‘行数‘"                       "from information_schema.tables "                       "where TABLE_SCHEMA in (‘" single.get("dbschema") "‘)  "                       "and (table_name " single.get("selectstr") " ‘" single.get("selectkeystr") "‘) "                       "and TABLE_ROWS > 0"
            df = pd.read_sql(sql, conn)
            print(single.get("key"), str(df[‘行数‘].sum()))
            dataSum.append(df[‘行数‘].sum())
            conn.close()
        elif "oracle" == single.get("dbtype"):
            if "table" == single.get("selecttype"):
                sql = "select owner as owner,"                       "table_name as table_name,"                       "tablespace_name as tablespace_name, "                       "num_rows as num_rows "                       "from all_tables "                       "where num_rows > 0 "                       "and table_name like ‘" single.get("selectkeystr") "‘ "                       "order by num_rows desc "
            elif "database" == single.get("selecttype"):  # 共享网站-oracle-工商局 定制
                sql = "select owner as owner, "                       "table_name as table_name, "                       "tablespace_name as tablespace_name, "                       "num_rows as num_rows "                       "from all_tables "                       "where num_rows > 0 "                       "and tablespace_name in(‘" single.get("dbschema") "‘) "                       "order by num_rows desc"
            db = cx_Oracle.connect(single.get("connstr"), encoding=‘utf-8‘)
            cursor = db.cursor()
            cursor.execute(sql)
            rs = cursor.fetchall()
            df = pd.DataFrame(rs)
            print(single.get("key"), str(df[3].sum()))
            dataSum.append(df[3].sum())
            cursor.close()
            db.close()
        elif "sqlserver" == single.get("dbtype"):
            print(single.get("key"), ‘55568045‘)
            dataSum.append(55568045)
            # "SELECT A.NAME ,B.ROWS FROM sysobjects A JOIN sysindexes B ON A.id = B.id WHERE A.xtype = ‘U‘ AND B.indid IN(0,1)  and b.rows >0 ORDER BY B.ROWS DESC"
        else:
            print("please give right database type.")
    connFile.close()
    print(‘-‘*30)
    print("数据量总计：", str(sum(dataSum)))


if __name__ == ‘__main__‘:
    print("***一次性统计所有对接数据的委办局，和其对应的数据（条数）***")
    main()

 

所需要的配置文件格式如下：

[
  {
    "key": "智慧公交",
    "dbtype": "oracle",
    "connstr": "nicai/[email protected]:1521/ORCL",
    "selecttype": "table",
    "selectstr": "like",
    "selectkeystr": "BUS%"
  },
  {
    "key": "公共自行车",
    "dbtype": "oracle",
    "connstr": "nicai/[email protected]:1521/ORCL",
    "selecttype": "table",
    "selectstr": "like",
    "selectkeystr": "BICYCLE%"
  },
  {
    "key": "安监局",
    "dbtype": "mysql",
    "host": "10.10.10.10",
    "port": 3306,
    "user": "nicai",
    "passwd": "123456",
    "charset": "utf8",
    "selecttype": "table",
    "selectstr": "like",
    "dbschema": "statistics_data",
    "selectkeystr": "ajj%"
  },
  {
    "key": "百度交通",
    "dbtype": "mysql",
    "host": "10.10.10.2",
    "port": 3306,
    "user": "nicai",
    "passwd": "123456",
    "charset": "utf8",
    "selecttype": "table",
    "selectstr": "like",
    "dbschema": "statistics_data",
    "selectkeystr": "bdu%"
  }
]

 

关于SqlServer的数据量查询，由于当时连不上，就没有嵌入到这个程序中。

不过查询的方法已经列出。

 

精进自己，分享他人！

 

谢谢