Last active
January 2, 2020 18:53
-
-
Save pfdevmuller/4072091 to your computer and use it in GitHub Desktop.
Python script to output the number of tables, columns and rows in a SQLite database
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# <author>Pieter Muller</author> | |
# <date>2012-11-14</date> | |
import sys | |
import sqlite3 as sqlite | |
tablesToIgnore = ["sqlite_sequence"] | |
outputFilename = None | |
def Print(msg): | |
if (outputFilename != None): | |
outputFile = open(outputFilename,'a') | |
print >> outputFile, msg | |
outputFile.close() | |
else: | |
print msg | |
def Describe(dbFile): | |
connection = sqlite.connect(dbFile) | |
cursor = connection.cursor() | |
Print("TableName\tColumns\tRows\tCells") | |
totalTables = 0 | |
totalColumns = 0 | |
totalRows = 0 | |
totalCells = 0 | |
# Get List of Tables: | |
tableListQuery = "SELECT name FROM sqlite_master WHERE type='table' ORDER BY Name" | |
cursor.execute(tableListQuery) | |
tables = map(lambda t: t[0], cursor.fetchall()) | |
for table in tables: | |
if (table in tablesToIgnore): | |
continue | |
columnsQuery = "PRAGMA table_info(%s)" % table | |
cursor.execute(columnsQuery) | |
numberOfColumns = len(cursor.fetchall()) | |
rowsQuery = "SELECT Count() FROM %s" % table | |
cursor.execute(rowsQuery) | |
numberOfRows = cursor.fetchone()[0] | |
numberOfCells = numberOfColumns*numberOfRows | |
Print("%s\t%d\t%d\t%d" % (table, numberOfColumns, numberOfRows, numberOfCells)) | |
totalTables += 1 | |
totalColumns += numberOfColumns | |
totalRows += numberOfRows | |
totalCells += numberOfCells | |
Print( "" ) | |
Print( "Number of Tables:\t%d" % totalTables ) | |
Print( "Total Number of Columns:\t%d" % totalColumns ) | |
Print( "Total Number of Rows:\t%d" % totalRows ) | |
Print( "Total Number of Cells:\t%d" % totalCells ) | |
cursor.close() | |
connection.close() | |
if __name__ == "__main__": | |
if (len(sys.argv) == 2): | |
dbFile = sys.argv[1] | |
Describe(dbFile) | |
elif (len(sys.argv) == 3): | |
dbFile = sys.argv[1] | |
outputFilename = sys.argv[2] | |
Describe(dbFile) | |
else: | |
print "\n\tUsage:" | |
print "\n\t\tDBDescribe.py {dbFile}" | |
print "\t\t\tPrints summary of {dbFile} to standard output." | |
print "\n\t\tDBDescribe.py {dbFile} {outputFile}" | |
print "\t\t\tAppends summary of {dbFile} to {outputFile}." | |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Thanks for this, Pieter; it does exactly what I was looking for! I had some too-long tables in my DB file, so I added some justification to make sure the table stays square. It's here.