54 lines
1.5 KiB
Python
Executable File
54 lines
1.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
import sqlite3
|
|
import sys
|
|
import json
|
|
import base64
|
|
import lite2j
|
|
|
|
def show_usage():
|
|
print 'Usage:\n\tlite2j sqlitefile.db --exclude=table_name --exclude=table_name'
|
|
|
|
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)
|
|
|
|
# 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()
|
|
|
|
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, excude_tables):
|
|
table_list = get_table_list(cursor, exclude_tables=exclude_tables)
|
|
return {table_name : get_table(cursor, table_name) for table_name in table_list}
|
|
|
|
if __name__ == '__main__':
|
|
print json.dumps(lite2j.get_tables(cursor, eclude_tables))
|