tryton-do/tryton
2
0 Fork 0
Code Issues Pull Requests 2 Actions 2 Packages Projects Releases Wiki Activity

Set default value for id and __id column for issue886

Browse Source
This commit is contained in:
Cédric Krier committed 2009-03-22 23:24:12 +01:00
1 parent 2d5396946c
commit 9db2aa4958
1 file changed
+13 -8
Download Patch File Download Diff File Expand all files Collapse all files
trytond/backend/postgresql/table.py
+13 -8
View File
@@ -23,22 +23,26 @@ class TableHandler(TableHandlerInterface):
# Create new table if necessary
if not self.table_exist(self.cursor, self.table_name):
self.cursor.execute('CREATE TABLE "%s" ' \
'(id INTEGER NOT NULL)' % self.table_name)
if not self.history:
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD PRIMARY KEY(id)' % self.table_name)
self.cursor.execute('CREATE TABLE "%s" ()' % self.table_name)
self._update_definitions()
if 'id' not in self._columns:
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD COLUMN id INTEGER NOT NULL' % self.table_name)
if not self.history:
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD COLUMN id INTEGER ' \
'DEFAULT nextval(\'%s\') NOT NULL' % \
(self.table_name, self.sequence_name))
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD PRIMARY KEY(id)' % self.table_name)
else:
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD COLUMN id INTEGER' % self.table_name)
self._update_definitions()
if self.history and not '__id' in self._columns:
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD COLUMN __id INTEGER NOT NULL' % self.table_name)
'ADD COLUMN __id INTEGER ' \
'DEFAULT nextval(\'%s\') NOT NULL' % \
(self.table_name, self.sequence_name))
self.cursor.execute('ALTER TABLE "%s" ' \
'ADD PRIMARY KEY(__id)' % self.table_name)
if self.history:
@@ -49,6 +53,7 @@ class TableHandler(TableHandlerInterface):
self.cursor.execute('ALTER TABLE "%s" ' \
'ALTER id SET DEFAULT nextval(\'%s\')' % (self.table_name,
self.sequence_name))
self._update_definitions()
@staticmethod
def table_exist(cursor, table_name):