dak/dakdb/update91.py

   1 #!/usr/bin/env python
   2 # coding=utf8
   3
   4 """
   5 per-queue NEW comments and permissions
   6
   7 @contact: Debian FTP Master <ftpmaster@debian.org>
   8 @copyright: 2012 Ansgar Burchardt <ansgar@debian.org>
   9 @license: GNU General Public License version 2 or later
  10 """
  11
  12 # This program is free software; you can redistribute it and/or modify
  13 # it under the terms of the GNU General Public License as published by
  14 # the Free Software Foundation; either version 2 of the License, or
  15 # (at your option) any later version.
  16
  17 # This program is distributed in the hope that it will be useful,
  18 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  19 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  20 # GNU General Public License for more details.
  21
  22 # You should have received a copy of the GNU General Public License
  23 # along with this program; if not, write to the Free Software
  24 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  25
  26 ################################################################################
  27
  28 import psycopg2
  29 from daklib.dak_exceptions import DBUpdateError
  30 from daklib.config import Config
  31
  32 statements = [
  33 """
  34 ALTER TABLE new_comments
  35 ADD COLUMN policy_queue_id INTEGER REFERENCES policy_queue(id)
  36 """,
  37
  38 """
  39 UPDATE new_comments
  40 SET policy_queue_id = (SELECT id FROM policy_queue WHERE queue_name = 'new')
  41 """,
  42
  43 """
  44 ALTER TABLE new_comments ALTER COLUMN policy_queue_id SET NOT NULL
  45 """,
  46
  47 """
  48 CREATE OR REPLACE FUNCTION trigger_check_policy_queue_permission() RETURNS TRIGGER
  49 SET search_path = public, pg_temp
  50 LANGUAGE plpgsql
  51 AS $$
  52 DECLARE
  53   v_row RECORD;
  54   v_suite_id suite.id%TYPE;
  55   v_policy_queue_name policy_queue.queue_name%TYPE;
  56 BEGIN
  57
  58   CASE TG_OP
  59     WHEN 'INSERT', 'UPDATE' THEN
  60       v_row := NEW;
  61     WHEN 'DELETE' THEN
  62       v_row := OLD;
  63     ELSE
  64       RAISE EXCEPTION 'Unexpected TG_OP (%)', TG_OP;
  65   END CASE;
  66
  67   IF TG_OP = 'UPDATE' AND OLD.policy_queue_id != NEW.policy_queue_id THEN
  68     RAISE EXCEPTION 'Cannot change policy_queue_id';
  69   END IF;
  70
  71   SELECT suite_id, queue_name INTO STRICT v_suite_id, v_policy_queue_name
  72     FROM policy_queue WHERE id = v_row.policy_queue_id;
  73   IF NOT has_suite_permission(TG_OP, v_suite_id) THEN
  74     RAISE EXCEPTION 'Not allowed to % in %', TG_OP, v_policy_queue_name;
  75   END IF;
  76
  77   RETURN v_row;
  78
  79 END;
  80 $$
  81 """,
  82
  83 """
  84 CREATE CONSTRAINT TRIGGER trigger_new_comments_permission
  85   AFTER INSERT OR UPDATE OR DELETE
  86   ON new_comments
  87   FOR EACH ROW
  88   EXECUTE PROCEDURE trigger_check_policy_queue_permission()
  89 """,
  90 ]
  91
  92 ################################################################################
  93 def do_update(self):
  94     print __doc__
  95     try:
  96         cnf = Config()
  97
  98         c = self.db.cursor()
  99
 100         for stmt in statements:
 101             c.execute(stmt)
 102
 103         c.execute("UPDATE config SET value = '91' WHERE name = 'db_revision'")
 104         self.db.commit()
 105
 106     except psycopg2.ProgrammingError as msg:
 107         self.db.rollback()
 108         raise DBUpdateError('Unable to apply sick update 91, rollback issued. Error message: {0}'.format(msg))