[2] | 1 | # Mimic the sqlite3 console shell's .dump command
|
---|
| 2 | # Author: Paul Kippes <kippesp@gmail.com>
|
---|
| 3 |
|
---|
[391] | 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 |
|
---|
[2] | 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 = """
|
---|
[391] | 24 | SELECT "name", "type", "sql"
|
---|
| 25 | FROM "sqlite_master"
|
---|
| 26 | WHERE "sql" NOT NULL AND
|
---|
| 27 | "type" == 'table'
|
---|
| 28 | ORDER BY "name"
|
---|
[2] | 29 | """
|
---|
| 30 | schema_res = cu.execute(q)
|
---|
| 31 | for table_name, type, sql in schema_res.fetchall():
|
---|
| 32 | if table_name == 'sqlite_sequence':
|
---|
[391] | 33 | yield('DELETE FROM "sqlite_sequence";')
|
---|
[2] | 34 | elif table_name == 'sqlite_stat1':
|
---|
[391] | 35 | yield('ANALYZE "sqlite_master";')
|
---|
[2] | 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)"\
|
---|
[391] | 42 | # "VALUES('table','{0}','{0}',0,'{1}');".format(
|
---|
[2] | 43 | # qtable,
|
---|
[391] | 44 | # sql.replace("''")))
|
---|
[2] | 45 | else:
|
---|
| 46 | yield('%s;' % sql)
|
---|
| 47 |
|
---|
| 48 | # Build the insert statement for each row of the current table
|
---|
[391] | 49 | table_name_ident = table_name.replace('"', '""')
|
---|
| 50 | res = cu.execute('PRAGMA table_info("{0}")'.format(table_name_ident))
|
---|
[2] | 51 | column_names = [str(table_info[1]) for table_info in res.fetchall()]
|
---|
[391] | 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)
|
---|
[2] | 56 | for row in query_res:
|
---|
| 57 | yield("%s;" % row[0])
|
---|
| 58 |
|
---|
| 59 | # Now when the type is 'index', 'trigger', or 'view'
|
---|
| 60 | q = """
|
---|
[391] | 61 | SELECT "name", "type", "sql"
|
---|
| 62 | FROM "sqlite_master"
|
---|
| 63 | WHERE "sql" NOT NULL AND
|
---|
| 64 | "type" IN ('index', 'trigger', 'view')
|
---|
[2] | 65 | """
|
---|
| 66 | schema_res = cu.execute(q)
|
---|
| 67 | for name, type, sql in schema_res.fetchall():
|
---|
| 68 | yield('%s;' % sql)
|
---|
| 69 |
|
---|
| 70 | yield('COMMIT;')
|
---|