| 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 |  | 
|---|
| 10 | def _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;') | 
|---|