dak/dakdb/update78.py

   1 #!/usr/bin/env python
   2 # coding=utf8
   3
   4 """
   5 Alter permissions for new tables and set default 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 ################################################################################
  33 def do_update(self):
  34     print __doc__
  35     try:
  36         cnf = Config()
  37
  38         c = self.db.cursor()
  39
  40         c.execute("GRANT SELECT ON files_archive_map, policy_queue_byhand_file, policy_queue_upload, policy_queue_upload_binaries_map TO PUBLIC")
  41         c.execute("GRANT ALL ON files_archive_map, policy_queue_byhand_file, policy_queue_upload, policy_queue_upload_binaries_map TO ftpmaster")
  42
  43         c.execute("ALTER DEFAULT PRIVILEGES GRANT SELECT ON TABLES TO PUBLIC")
  44         c.execute("ALTER DEFAULT PRIVILEGES GRANT ALL ON TABLES TO ftpmaster")
  45         c.execute("ALTER DEFAULT PRIVILEGES GRANT SELECT ON SEQUENCES TO PUBLIC")
  46         c.execute("ALTER DEFAULT PRIVILEGES GRANT ALL ON SEQUENCES TO ftpmaster")
  47         c.execute("ALTER DEFAULT PRIVILEGES GRANT ALL ON FUNCTIONS TO ftpmaster")
  48
  49         c.execute("UPDATE config SET value = '78' WHERE name = 'db_revision'")
  50         self.db.commit()
  51
  52     except psycopg2.ProgrammingError as msg:
  53         self.db.rollback()
  54         raise DBUpdateError('Unable to apply sick update 78, rollback issued. Error message: {0}'.format(msg))