[RFR] Hide implementation of constraint definition storage, courtesy of Raphael Collet

This commit is contained in:
Stefan Rijnhart
2018-08-16 10:33:06 +02:00
committed by Raphael Collet
parent 6327b8f97f
commit b3824aa6cb
2 changed files with 11 additions and 15 deletions
+3 -3
View File
@@ -2445,11 +2445,11 @@ class BaseModel(MetaModel('DummyModel', (object,), {'_register': False})):
def process(key, definition):
conname = '%s_%s' % (self._table, key)
has_definition = tools.constraint_definition(cr, self._table, conname)
if not has_definition:
current_definition = tools.constraint_definition(cr, self._table, conname)
if not current_definition:
# constraint does not exists
tools.add_constraint(cr, self._table, conname, definition)
elif tools.constraint_comment(cr, self._table, conname) != definition:
elif current_definition != definition:
# constraint exists but its definition may have changed
tools.drop_constraint(cr, self._table, conname)
tools.add_constraint(cr, self._table, conname, definition)
+8 -12
View File
@@ -120,17 +120,13 @@ def drop_not_null(cr, tablename, columnname):
def constraint_definition(cr, tablename, constraintname):
""" Return the given constraint's definition. """
cr.execute('SELECT pg_get_constraintdef(c.oid) FROM pg_constraint c '
'JOIN pg_class t ON t.oid = c.conrelid '
'WHERE t.relname = %s AND conname=%s', (tablename, constraintname,))
return cr.fetchone()[0] if cr.rowcount else None
def constraint_comment(cr, tablename, constraintname):
""" Return the given constraint's comment. """
cr.execute('SELECT d.description FROM pg_description d '
'JOIN pg_constraint c ON c.oid = d.objoid '
'JOIN pg_class t ON t.oid = c.conrelid '
'WHERE t.relname = %s AND conname = %s;', (tablename, constraintname,))
query = """
SELECT COALESCE(d.description, pg_get_constraintdef(c.oid))
FROM pg_constraint c
JOIN pg_class t ON t.oid = c.conrelid
LEFT JOIN pg_description d ON c.oid = d.objoid
WHERE t.relname = %s AND conname = %s;"""
cr.execute(query, (tablename, constraintname))
return cr.fetchone()[0] if cr.rowcount else None
def add_constraint(cr, tablename, constraintname, definition):
@@ -145,7 +141,7 @@ def add_constraint(cr, tablename, constraintname, definition):
except Exception:
msg = "Table %r: unable to add constraint %r!\n" \
"If you want to have it, you should update the records and execute manually:\n%s"
_schema.warning(msg, tablename, constraintname, query, exc_info=True)
_schema.warning(msg, tablename, constraintname, query1, exc_info=True)
def drop_constraint(cr, tablename, constraintname):
""" drop the given constraint. """