|
@@ -54,7 +54,9 @@ class DatabaseOperations(BaseDatabaseOperations):
|
|
|
return '%s'
|
|
|
|
|
|
def last_insert_id(self, cursor, table_name, pk_name):
|
|
|
- cursor.execute("SELECT CURRVAL('\"%s_%s_seq\"')" % (table_name, pk_name))
|
|
|
+ # Use pg_get_serial_sequence to get the underlying sequence name
|
|
|
+ # from the table name and column name (available since PostgreSQL 8)
|
|
|
+ cursor.execute("SELECT CURRVAL(pg_get_serial_sequence('%s','%s'))" % (table_name, pk_name))
|
|
|
return cursor.fetchone()[0]
|
|
|
|
|
|
def no_limit_value(self):
|
|
@@ -90,13 +92,14 @@ class DatabaseOperations(BaseDatabaseOperations):
|
|
|
for sequence_info in sequences:
|
|
|
table_name = sequence_info['table']
|
|
|
column_name = sequence_info['column']
|
|
|
- if column_name and len(column_name) > 0:
|
|
|
- sequence_name = '%s_%s_seq' % (table_name, column_name)
|
|
|
- else:
|
|
|
- sequence_name = '%s_id_seq' % table_name
|
|
|
- sql.append("%s setval('%s', 1, false);" % \
|
|
|
+ if not (column_name and len(column_name) > 0):
|
|
|
+ # This will be the case if it's an m2m using an autogenerated
|
|
|
+ # intermediate table (see BaseDatabaseIntrospection.sequence_list)
|
|
|
+ column_name = 'id'
|
|
|
+ sql.append("%s setval(pg_get_serial_sequence('%s','%s'), 1, false);" % \
|
|
|
(style.SQL_KEYWORD('SELECT'),
|
|
|
- style.SQL_FIELD(self.quote_name(sequence_name)))
|
|
|
+ style.SQL_TABLE(table_name),
|
|
|
+ style.SQL_FIELD(column_name))
|
|
|
)
|
|
|
return sql
|
|
|
else:
|
|
@@ -110,11 +113,15 @@ class DatabaseOperations(BaseDatabaseOperations):
|
|
|
# Use `coalesce` to set the sequence for each model to the max pk value if there are records,
|
|
|
# or 1 if there are none. Set the `is_called` property (the third argument to `setval`) to true
|
|
|
# if there are records (as the max pk value is already in use), otherwise set it to false.
|
|
|
+ # Use pg_get_serial_sequence to get the underlying sequence name from the table name
|
|
|
+ # and column name (available since PostgreSQL 8)
|
|
|
+
|
|
|
for f in model._meta.local_fields:
|
|
|
if isinstance(f, models.AutoField):
|
|
|
- output.append("%s setval('%s', coalesce(max(%s), 1), max(%s) %s null) %s %s;" % \
|
|
|
+ output.append("%s setval(pg_get_serial_sequence('%s','%s'), coalesce(max(%s), 1), max(%s) %s null) %s %s;" % \
|
|
|
(style.SQL_KEYWORD('SELECT'),
|
|
|
- style.SQL_FIELD(qn('%s_%s_seq' % (model._meta.db_table, f.column))),
|
|
|
+ style.SQL_TABLE(model._meta.db_table),
|
|
|
+ style.SQL_FIELD(f.column),
|
|
|
style.SQL_FIELD(qn(f.column)),
|
|
|
style.SQL_FIELD(qn(f.column)),
|
|
|
style.SQL_KEYWORD('IS NOT'),
|
|
@@ -123,9 +130,10 @@ class DatabaseOperations(BaseDatabaseOperations):
|
|
|
break # Only one AutoField is allowed per model, so don't bother continuing.
|
|
|
for f in model._meta.many_to_many:
|
|
|
if not f.rel.through:
|
|
|
- output.append("%s setval('%s', coalesce(max(%s), 1), max(%s) %s null) %s %s;" % \
|
|
|
+ output.append("%s setval(pg_get_serial_sequence('%s','%s'), coalesce(max(%s), 1), max(%s) %s null) %s %s;" % \
|
|
|
(style.SQL_KEYWORD('SELECT'),
|
|
|
- style.SQL_FIELD(qn('%s_id_seq' % f.m2m_db_table())),
|
|
|
+ style.SQL_TABLE(model._meta.db_table),
|
|
|
+ style.SQL_FIELD('id'),
|
|
|
style.SQL_FIELD(qn('id')),
|
|
|
style.SQL_FIELD(qn('id')),
|
|
|
style.SQL_KEYWORD('IS NOT'),
|