59 lines
1.7 KiB
Python
Executable File
59 lines
1.7 KiB
Python
Executable File
import sqlite3
|
|
import sys
|
|
import json
|
|
import base64
|
|
|
|
def main():
|
|
exclude_tables = []
|
|
|
|
try:
|
|
for i, token in enumerate(sys.argv):
|
|
if i < 1:
|
|
continue
|
|
if i == 1:
|
|
db_file = token
|
|
if i > 1:
|
|
exclude_tables.append(token.split('--exclude=')[1])
|
|
assert(db_file)
|
|
except (IndexError, NameError):
|
|
show_usage()
|
|
sys.exit(127)
|
|
|
|
exclude_tables = set(exclude_tables)
|
|
print sqlite2json(
|
|
db_file=db_file,
|
|
exclude_tables=exclude_tables
|
|
)
|
|
|
|
def show_usage():
|
|
print 'Usage:\n\tsqlite2json sqlitefile.db --exclude=table_name --exclude=table_name'
|
|
|
|
def sqlite2json(exclude_tables=None,db_file=None):
|
|
# Make sure blobs are base64 encoded
|
|
sqlite3.register_converter('BLOB', base64.b64encode)
|
|
|
|
conn = sqlite3.connect(db_file, detect_types=sqlite3.PARSE_DECLTYPES)
|
|
cursor = conn.cursor()
|
|
|
|
return json.dumps(get_tables(cursor))
|
|
|
|
def get_table_list(cursor, exclude_tables=set([])):
|
|
cursor.execute('SELECT * FROM main.sqlite_master WHERE type="table"')
|
|
return set(row[1] for row in cursor.fetchall()) - exclude_tables
|
|
|
|
def get_column_names(cursor, table_name):
|
|
cursor.execute('SELECT * FROM %s' % table_name)
|
|
return [c[0] for c in cursor.description]
|
|
|
|
def get_table(cursor, table_name):
|
|
column_names = get_column_names(cursor,table_name)
|
|
cursor.execute('SELECT * FROM main.%s' % table_name)
|
|
return [dict(zip(column_names, row)) for row in cursor.fetchall()]
|
|
|
|
def get_tables(cursor):
|
|
table_list = get_table_list(cursor)
|
|
return [{table_name : get_table(cursor, table_name) for table_name in table_list}]
|
|
|
|
if __name__ == "__main__":
|
|
main()
|