Added feature to extract references from create table sql

README.md

@@ -1,25 +1,20 @@
-# sqlite2json
+# lite2j -- sqlite to json export utiilty
 
-This script dumps all\* the tables in an sqlite database as json.
+This script dumps all* the tables in an sqlite database as json.
 
 ## Usage
 
-    sqlite2json sqlitefile.db [--exclude=TABLE]
+    lite2j sqlitefile.db [--exclude=TABLE]
 
 
 To dump the sqlite file where iMessages are stored:
 
-    sqlite2json ~/Library/Messages/chat.db
+    lite2j ~/Library/Messages/chat.db
 
 Here is a more useful version of the above example:
 
-    sqlite2json ~/Library/Messages/chat.db | jq -r .message[].text | grep -i loan
+    lite2j ~/Library/Messages/chat.db | jq -r .message[].text | grep -i loan
 
 
 \*The script doesn't dump the contents of of the sqlite_master table 
 
-# Authors
-
-* John Gerlock `<john@pikkey.com>`
-
-* Jeffrey Paul `<sneak@sneak.berlin>`

lite2j/lite2j.py

@@ -0,0 +1,158 @@
+#!/usr/bin/env python
+
+import re
+import bisect
+import pprint
+
+
+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_tables(cursor, exclude_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}
+
+
+def get_tables_info(cursor):
+    cursor.execute('SELECT * FROM main.sqlite_master WHERE type="table"')
+    return cursor.fetchall()
+
+
+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_data(cursor):
+    table_list = get_table_list(cursor)
+    return {
+        table_name: get_table(cursor, table_name) for
+        table_name in table_list}
+
+
+def build_table_map(cursor, exclude_tables=set()):
+    table_list = get_tables_info(cursor)
+    table_names = [table['tbl_name'] for table in table_list]
+    tables = dict()
+    for table in table_list:
+        name = table['tbl_name']
+        sql = table['sql']
+        columns = get_column_names(cursor, name)
+        column_defs = get_column_defs(sql)
+        ref_indexes = []
+        child_column_indexes = []
+        parent_table_indexes = []
+        parent_table_ref_map = {}
+        child_column_ref_map = {}
+        parent_column_ref_map = {}
+        if column_defs:
+            ref_indexes = find_keyword_indexes(
+                column_defs, keyword='REFERENCES')
+            child_column_indexes = find_indexes(column_defs, columns)
+            child_column_ref_map = {ref: find_rightmost_lt_index(
+                child_column_indexes, ref) for ref in ref_indexes}
+            parent_table_indexes = find_indexes(column_defs, table_names)
+            parent_table_ref_map = {ref: find_leftmost_gt_index(
+                parent_table_indexes, ref) for ref in ref_indexes}
+            parent_column_ref_map = {ref: find_parent_column_for_ref(
+                column_defs,
+                ref,
+                parent_table_ref_map, cursor) for ref in ref_indexes}
+
+        tables[name] = dict(sql=sql,
+                            columns=columns,
+                            column_defs=column_defs,
+                            ref_indexes=ref_indexes,
+                            child_column_ref_map=child_column_ref_map,
+                            parent_table_ref_map=parent_table_ref_map,
+                            parent_table_indexes=parent_table_indexes,
+                            child_column_indexes=child_column_indexes,
+                            parent_column_ref_map=parent_column_ref_map)
+    return tables
+
+
+def get_column_defs(sql):
+    columns_pattern = r'^CREATE\s+TABLE\s+\w+\s+\(\s?(?P<column_defs>.*)\)$'
+    matches = re.match(columns_pattern, sql, flags=re.IGNORECASE)
+    if matches:
+        return matches.group('column_defs')
+    else:
+        return None
+
+
+def find_keyword_indexes(text, keyword=None):
+    return [m.start() for m in re.finditer(keyword, text, flags=re.IGNORECASE)]
+
+
+def find_indexes(text, keywords):
+    return {keyword: find_keyword_indexes(text, keyword=keyword)
+            for keyword in keywords}
+
+
+def find_rightmost_lt_index(indexes, ref_index):
+    inverted = build_inverted_index(indexes)
+    idx = find_lt(sorted(inverted.keys()), ref_index)
+    return inverted[idx]
+
+
+def find_leftmost_gt_index(indexes, ref_index):
+    inverted = build_inverted_index(indexes)
+    idx = find_gt(sorted(inverted.keys()), ref_index)
+    return inverted[idx]
+
+
+def find_parent_column_for_ref(text, ref_index, table_name_ref_map, cursor):
+    referenced_table = table_name_ref_map[ref_index]
+    columns = get_column_names(cursor, referenced_table)
+    child_column_indexes = find_indexes(text, columns)
+    return find_leftmost_gt_index(child_column_indexes, ref_index)
+
+
+def build_inverted_index(indexes):
+    inverted = dict()
+    for key, value_list in indexes.iteritems():
+        for value in value_list:
+            inverted[value] = key
+    return inverted
+
+
+def find_lt(a, x):
+    'Find rightmost value less than x'
+    i = bisect.bisect_left(a, x)
+    if i:
+        return a[i - 1]
+    raise ValueError
+
+
+def find_gt(a, x):
+    'Find leftmost value greater than x'
+    i = bisect.bisect_right(a, x)
+    if i != len(a):
+        return a[i]
+    raise ValueError
+
+# debugging stuff
+
+
+def print_tables_info(tables):
+    for name, t in tables.iteritems():
+        if len(t['ref_indexes']) > 0:
+            pprint.pprint(tables)
+
+
+def print_references(tables):
+    for name, t in tables.iteritems():
+        for ref in t['ref_indexes']:
+            col = t['child_column_ref_map'][ref]
+            tbl = t['parent_table_ref_map'][ref]
+            parent_column = t['parent_column_ref_map'][ref]
+            print '%s.%s ----> %s.%s' % (name, col, tbl, parent_column)

lite2j/script.py

@@ -0,0 +1,53 @@
+#!/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))

setup.py

@@ -1,18 +1,14 @@
-from setuptools import setup
-
+from setuptools import setup, find_packages
 setup(
-    name='sqlite2json',
-    version='1.0.1',
-    author='John Gerlock',
-    author_email='john@pikkey.com',
-    packages=['sqlite2json'],
-    url='https://github.com/john-g-g/sqlite2json',
-    license=open('LICENSE').read(),
-    description='SQLite to JSON converter',
-    long_description=open('README.md').read(),
+        name = "lite2j",
+        version = "0.1",
+        description = "Dump sqlite files as json",
+        author = "John Gerlock",
+        author_email = "john.gerlock@gmail.com",
+        url = "http://github.com/john-g-g/lite2j",
         entry_points={
-        'console_scripts': [
-            'sqlite2json = sqlite2json:main'
-        ]
-    }
+            'console_scripts': ['lite2j=lite2j:main',],
+                },
+        packages = find_packages(),
+
 )

sqlite2json/__init__.py

@@ -1,61 +0,0 @@
-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)
-    output = {}
-    for table_name in table_list:
-        output[table_name] = get_table(cursor, table_name)
-    return output
-
-if __name__ == "__main__":
-    main()