1# Mimic the sqlite3 console shell's .dump command 2# Author: Paul Kippes <kippesp@gmail.com> 3 4# Every identifier in sql is quoted based on a comment in sqlite 5# documentation "SQLite adds new keywords from time to time when it 6# takes on new features. So to prevent your code from being broken by 7# future enhancements, you should normally quote any identifier that 8# is an English language word, even if you do not have to." 9 10def _iterdump(connection): 11 """ 12 Returns an iterator to the dump of the database in an SQL text format. 13 14 Used to produce an SQL dump of the database. Useful to save an in-memory 15 database for later restoration. This function should not be called 16 directly but instead called from the Connection method, iterdump(). 17 """ 18 19 cu = connection.cursor() 20 yield('BEGIN TRANSACTION;') 21 22 # sqlite_master table contains the SQL CREATE statements for the database. 23 q = """ 24 SELECT "name", "type", "sql" 25 FROM "sqlite_master" 26 WHERE "sql" NOT NULL AND 27 "type" == 'table' 28 ORDER BY "name" 29 """ 30 schema_res = cu.execute(q) 31 for table_name, type, sql in schema_res.fetchall(): 32 if table_name == 'sqlite_sequence': 33 yield('DELETE FROM "sqlite_sequence";') 34 elif table_name == 'sqlite_stat1': 35 yield('ANALYZE "sqlite_master";') 36 elif table_name.startswith('sqlite_'): 37 continue 38 # NOTE: Virtual table support not implemented 39 #elif sql.startswith('CREATE VIRTUAL TABLE'): 40 # qtable = table_name.replace("'", "''") 41 # yield("INSERT INTO sqlite_master(type,name,tbl_name,rootpage,sql)"\ 42 # "VALUES('table','{0}','{0}',0,'{1}');".format( 43 # qtable, 44 # sql.replace("''"))) 45 else: 46 yield('%s;' % sql) 47 48 # Build the insert statement for each row of the current table 49 table_name_ident = table_name.replace('"', '""') 50 res = cu.execute('PRAGMA table_info("{0}")'.format(table_name_ident)) 51 column_names = [str(table_info[1]) for table_info in res.fetchall()] 52 q = """SELECT 'INSERT INTO "{0}" VALUES({1})' FROM "{0}";""".format( 53 table_name_ident, 54 ",".join("""'||quote("{0}")||'""".format(col.replace('"', '""')) for col in column_names)) 55 query_res = cu.execute(q) 56 for row in query_res: 57 yield("%s;" % row[0]) 58 59 # Now when the type is 'index', 'trigger', or 'view' 60 q = """ 61 SELECT "name", "type", "sql" 62 FROM "sqlite_master" 63 WHERE "sql" NOT NULL AND 64 "type" IN ('index', 'trigger', 'view') 65 """ 66 schema_res = cu.execute(q) 67 for name, type, sql in schema_res.fetchall(): 68 yield('%s;' % sql) 69 70 yield('COMMIT;') 71