Wypisywanie błędów przy importowaniu książek (obecnie wypisujemy po prostu wyjątek...
[wolnelektury.git] / apps / south / db / postgresql_psycopg2.py
1
2 from django.db import connection
3 from south.db import generic
4
5 class DatabaseOperations(generic.DatabaseOperations):
6
7     """
8     PsycoPG2 implementation of database operations.
9     """
10
11     def rename_column(self, table_name, old, new):
12         if old == new:
13             return []
14         qn = connection.ops.quote_name
15         params = (qn(table_name), qn(old), qn(new))
16         self.execute('ALTER TABLE %s RENAME COLUMN %s TO %s;' % params)
17     
18     def rename_table(self, old_table_name, table_name):
19         "will rename the table and an associated ID sequence and primary key index"
20         # First, rename the table
21         generic.DatabaseOperations.rename_table(self, old_table_name, table_name)
22         # Then, try renaming the ID sequence
23         # (if you're using other AutoFields... your problem, unfortunately)
24         self.commit_transaction()
25         self.start_transaction()
26         try:
27             generic.DatabaseOperations.rename_table(self, old_table_name+"_id_seq", table_name+"_id_seq")
28         except:
29             if self.debug:
30                 print "   ~ No such sequence (ignoring error)"
31             self.rollback_transaction()
32         else:
33             self.commit_transaction()
34         self.start_transaction()
35
36         # Rename primary key index, will not rename other indices on
37         # the table that are used by django (e.g. foreign keys). Until
38         # figure out how, you need to do this yourself.
39         try:
40             generic.DatabaseOperations.rename_table(self, old_table_name+"_pkey", table_name+ "_pkey")
41         except:
42             if self.debug:
43                 print "   ~ No such primary key (ignoring error)"
44             self.rollback_transaction()
45         else:
46             self.commit_transaction()
47         self.start_transaction()
48
49
50     def rename_index(self, old_index_name, index_name):
51         "Rename an index individually"
52         generic.DatabaseOperations.rename_table(self, old_index_name, index_name)