Skip to content

Commit

Permalink
fix delete sample prep
Browse files Browse the repository at this point in the history
  • Loading branch information
antgonza committed Jun 26, 2017
1 parent 464f70b commit e819ca2
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 0 deletions.
6 changes: 6 additions & 0 deletions qiita_db/metadata_template/base_metadata_template.py
Expand Up @@ -619,9 +619,15 @@ def _common_delete_sample_steps(self, sample_name):
self._table_name(self._id))
qdb.sql_connection.TRN.add(sql, [sample_name])

sql = 'ALTER TABLE qiita.{0} DISABLE TRIGGER ALL'.format(
self._table)
qdb.sql_connection.TRN.add(sql)
sql = "DELETE FROM qiita.{0} WHERE sample_id=%s AND {1}=%s".format(
self._table, self._id_column)
qdb.sql_connection.TRN.add(sql, [sample_name, self.id])
sql = 'ALTER TABLE qiita.{0} ENABLE TRIGGER ALL'.format(
self._table)
qdb.sql_connection.TRN.add(sql)

qdb.sql_connection.TRN.execute()

Expand Down
8 changes: 8 additions & 0 deletions qiita_db/metadata_template/prep_template.py
Expand Up @@ -256,14 +256,22 @@ def delete(cls, id_):
qdb.sql_connection.TRN.add(sql, args)

# Remove the row from study_prep_template
sql = 'ALTER TABLE qiita.study_prep_template DISABLE TRIGGER ALL'
qdb.sql_connection.TRN.add(sql)
sql = """DELETE FROM qiita.study_prep_template
WHERE {0} = %s""".format(cls._id_column)
qdb.sql_connection.TRN.add(sql, args)
sql = 'ALTER TABLE qiita.study_prep_template ENABLE TRIGGER ALL'
qdb.sql_connection.TRN.add(sql)

# Remove the row from prep_template
sql = 'ALTER TABLE qiita.prep_template DISABLE TRIGGER ALL'
qdb.sql_connection.TRN.add(sql)
sql = "DELETE FROM qiita.prep_template WHERE {0} = %s".format(
cls._id_column)
qdb.sql_connection.TRN.add(sql, args)
sql = 'ALTER TABLE qiita.prep_template ENABLE TRIGGER ALL'
qdb.sql_connection.TRN.add(sql)

qdb.sql_connection.TRN.execute()

Expand Down
6 changes: 6 additions & 0 deletions qiita_db/metadata_template/sample_template.py
Expand Up @@ -135,9 +135,15 @@ def delete(cls, id_):
sql = "DROP TABLE qiita.{0}".format(table_name)
qdb.sql_connection.TRN.add(sql)

sql = 'ALTER TABLE qiita.{0} DISABLE TRIGGER ALL'.format(
cls._table)
qdb.sql_connection.TRN.add(sql)
sql = "DELETE FROM qiita.{0} WHERE {1} = %s".format(
cls._table, cls._id_column)
qdb.sql_connection.TRN.add(sql, args)
sql = 'ALTER TABLE qiita.{0} ENABLE TRIGGER ALL'.format(
cls._table)
qdb.sql_connection.TRN.add(sql)

qdb.sql_connection.TRN.execute()

Expand Down

0 comments on commit e819ca2

Please sign in to comment.