Make reset-postgresql-sequences work on empty tables

2012-10-19 13:35:05 -07:00 · 2012-10-19 13:35:05 -07:00 · 787a7b864b
parent c4d360e0ec
commit 787a7b864b
1 changed files with 3 additions and 2 deletions
--- a/bin/reset-postgresql-sequences
+++ b/bin/reset-postgresql-sequences
@ -33,8 +33,9 @@ for table in sorted(mapped_classes):
    for c in sequence_columns(table):
        max_value, = session.query(func.max(c)).one()
        if options.verbose:
-            print "%s.%s <- %s" % (table_name, c.name, max_value)
+            print "%s.%s <- %s" % (table_name, c.name, max_value or 0)
        session.execute(func.setval(
            func.pg_get_serial_sequence(table_name, c.name),
-            max_value,
+            max_value or 1,
+            max_value is not None,
        ))