dak/dakdb/update2.py

   1 #!/usr/bin/env python
   2 # coding=utf8
   3
   4 """
   5 debversion
   6
   7 @contact: Debian FTP Master <ftpmaster@debian.org>
   8 @copyright: 2008  Michael Casadevall <mcasadevall@debian.org>
   9 @copyright: 2008  Roger Leigh <rleigh@debian.org>
  10 @license: GNU General Public License version 2 or later
  11 """
  12
  13 # This program is free software; you can redistribute it and/or modify
  14 # it under the terms of the GNU General Public License as published by
  15 # the Free Software Foundation; either version 2 of the License, or
  16 # (at your option) any later version.
  17
  18 # This program is distributed in the hope that it will be useful,
  19 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  20 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21 # GNU General Public License for more details.
  22
  23 # You should have received a copy of the GNU General Public License
  24 # along with this program; if not, write to the Free Software
  25 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  26
  27 ################################################################################
  28
  29 import psycopg2
  30 import time
  31 from daklib.dak_exceptions import DBUpdateError
  32
  33 ################################################################################
  34
  35 def do_update(self):
  36     print "Note: to be able to enable the the PL/Perl (plperl) procedural language, we do"
  37     print "need postgresql-plperl-$postgres-version installed. Make sure that this is the"
  38     print "case before you continue. Interrupt if it isn't, sleeping 5 seconds now."
  39     print "(We need to be database superuser for this to work!)"
  40     time.sleep (5)
  41
  42     try:
  43         c = self.db.cursor()
  44
  45         print "Enabling PL/Perl language"
  46         c.execute("CREATE LANGUAGE plperl;")
  47         c.execute("CREATE LANGUAGE plpgsql;")
  48
  49         print "Adding debversion type to database."
  50
  51 # Not present in all databases, maybe PL/Perl version-dependent?
  52 #        c.execute("SET SESSION plperl.use_strict TO 't';")
  53
  54         c.execute("CREATE DOMAIN debversion AS TEXT;")
  55         c.execute("COMMENT ON DOMAIN debversion IS 'Debian package version number';")
  56
  57         c.execute("""ALTER DOMAIN debversion
  58                      ADD CONSTRAINT debversion_syntax
  59                      CHECK (VALUE !~ '[^-+:.0-9a-zA-Z~]');""")
  60
  61         # From Dpkg::Version::parseversion
  62         c.execute("""CREATE OR REPLACE FUNCTION debversion_split (debversion)
  63   RETURNS text[] AS $$
  64     my $ver = shift;
  65     my %verhash;
  66     if ($ver =~ /:/)
  67     {
  68         $ver =~ /^(\d+):(.+)/ or die "bad version number '$ver'";
  69         $verhash{epoch} = $1;
  70         $ver = $2;
  71     }
  72     else
  73     {
  74         $verhash{epoch} = 0;
  75     }
  76     if ($ver =~ /(.+)-(.*)$/)
  77     {
  78         $verhash{version} = $1;
  79         $verhash{revision} = $2;
  80     }
  81     else
  82     {
  83         $verhash{version} = $ver;
  84         $verhash{revision} = 0;
  85     }
  86
  87     return [$verhash{'epoch'}, $verhash{'version'}, $verhash{'revision'}];
  88 $$
  89   LANGUAGE plperl
  90   IMMUTABLE STRICT;""")
  91         c.execute("""COMMENT ON FUNCTION debversion_split (debversion)
  92                    IS 'Split debian version into epoch, upstream version and revision';""")
  93
  94         c.execute("""CREATE OR REPLACE FUNCTION debversion_epoch (version debversion)
  95   RETURNS text AS $$
  96 DECLARE
  97   split text[];
  98 BEGIN
  99   split := debversion_split(version);
 100   RETURN split[1];
 101 END;
 102 $$
 103   LANGUAGE plpgsql
 104   IMMUTABLE STRICT;
 105 COMMENT ON FUNCTION debversion_epoch (debversion)
 106   IS 'Get debian version epoch';
 107
 108 CREATE OR REPLACE FUNCTION debversion_version (version debversion)
 109   RETURNS text AS $$
 110 DECLARE
 111   split text[];
 112 BEGIN
 113   split := debversion_split(version);
 114   RETURN split[2];
 115 END;
 116 $$
 117   LANGUAGE plpgsql
 118   IMMUTABLE STRICT;""")
 119         c.execute("""COMMENT ON FUNCTION debversion_version (debversion)
 120                    IS 'Get debian version upstream version';""")
 121
 122         c.execute("""CREATE OR REPLACE FUNCTION debversion_revision (version debversion)
 123   RETURNS text AS $$
 124 DECLARE
 125   split text[];
 126 BEGIN
 127   split := debversion_split(version);
 128   RETURN split[3];
 129 END;
 130 $$
 131   LANGUAGE plpgsql
 132   IMMUTABLE STRICT;""")
 133         c.execute("""COMMENT ON FUNCTION debversion_revision (debversion)
 134                    IS 'Get debian version revision';""")
 135
 136 # From Dpkg::Version::parseversion
 137         c.execute("""CREATE OR REPLACE FUNCTION debversion_compare_single (version1 text, version2 text)
 138   RETURNS integer AS $$
 139      sub order{
 140           my ($x) = @_;
 141           ##define order(x) ((x) == '~' ? -1 \
 142           #           : cisdigit((x)) ? 0 \
 143           #           : !(x) ? 0 \
 144           #           : cisalpha((x)) ? (x) \
 145           #           : (x) + 256)
 146           # This comparison is out of dpkg's order to avoid
 147           # comparing things to undef and triggering warnings.
 148           if (not defined $x or not length $x) {
 149                return 0;
 150           }
 151           elsif ($x eq '~') {
 152                return -1;
 153           }
 154           elsif ($x =~ /^\d$/) {
 155                return 0;
 156           }
 157           elsif ($x =~ /^[A-Z]$/i) {
 158                return ord($x);
 159           }
 160           else {
 161                return ord($x) + 256;
 162           }
 163      }
 164
 165      sub next_elem(\@){
 166           my $a = shift;
 167           return @{$a} ? shift @{$a} : undef;
 168      }
 169      my ($val, $ref) = @_;
 170      $val = "" if not defined $val;
 171      $ref = "" if not defined $ref;
 172      my @val = split //,$val;
 173      my @ref = split //,$ref;
 174      my $vc = next_elem @val;
 175      my $rc = next_elem @ref;
 176      while (defined $vc or defined $rc) {
 177           my $first_diff = 0;
 178           while ((defined $vc and $vc !~ /^\d$/) or
 179                  (defined $rc and $rc !~ /^\d$/)) {
 180                my $vo = order($vc); my $ro = order($rc);
 181                # Unlike dpkg's verrevcmp, we only return 1 or -1 here.
 182                return (($vo - $ro > 0) ? 1 : -1) if $vo != $ro;
 183                $vc = next_elem @val; $rc = next_elem @ref;
 184           }
 185           while (defined $vc and $vc eq '0') {
 186                $vc = next_elem @val;
 187           }
 188           while (defined $rc and $rc eq '0') {
 189                $rc = next_elem @ref;
 190           }
 191           while (defined $vc and $vc =~ /^\d$/ and
 192                  defined $rc and $rc =~ /^\d$/) {
 193                $first_diff = ord($vc) - ord($rc) if !$first_diff;
 194                $vc = next_elem @val; $rc = next_elem @ref;
 195           }
 196           return 1 if defined $vc and $vc =~ /^\d$/;
 197           return -1 if defined $rc and $rc =~ /^\d$/;
 198           return (($first_diff  > 0) ? 1 : -1) if $first_diff;
 199      }
 200      return 0;
 201 $$
 202   LANGUAGE plperl
 203   IMMUTABLE STRICT;""")
 204         c.execute("""COMMENT ON FUNCTION debversion_compare_single (text, text)
 205                    IS 'Compare upstream or revision parts of Debian versions';""")
 206
 207 # Logic only derived from Dpkg::Version::parseversion
 208         c.execute("""CREATE OR REPLACE FUNCTION debversion_compare (version1 debversion, version2 debversion)
 209   RETURNS integer AS $$
 210 DECLARE
 211   split1 text[];
 212   split2 text[];
 213   result integer;
 214 BEGIN
 215   result := 0;
 216   split1 := debversion_split(version1);
 217   split2 := debversion_split(version2);
 218
 219   -- RAISE NOTICE 'Version 1: %', version1;
 220   -- RAISE NOTICE 'Version 2: %', version2;
 221   -- RAISE NOTICE 'Split 1: %', split1;
 222   -- RAISE NOTICE 'Split 2: %', split2;
 223
 224   IF split1[1] > split2[1] THEN
 225     result := 1;
 226   ELSIF split1[1] < split2[1] THEN
 227     result := -1;
 228   ELSE
 229     result := debversion_compare_single(split1[2], split2[2]);
 230     IF result = 0 THEN
 231       result := debversion_compare_single(split1[3], split2[3]);
 232     END IF;
 233   END IF;
 234
 235   RETURN result;
 236 END;
 237 $$
 238   LANGUAGE plpgsql
 239   IMMUTABLE STRICT;""")
 240         c.execute("""COMMENT ON FUNCTION debversion_compare (debversion, debversion)
 241   IS 'Compare Debian versions';""")
 242
 243         c.execute("""CREATE OR REPLACE FUNCTION debversion_eq (version1 debversion, version2 debversion)
 244   RETURNS boolean AS $$
 245 DECLARE
 246   comp integer;
 247   result boolean;
 248 BEGIN
 249   comp := debversion_compare(version1, version2);
 250   result := comp = 0;
 251   RETURN result;
 252 END;
 253 $$
 254   LANGUAGE plpgsql
 255   IMMUTABLE STRICT;""")
 256         c.execute("""COMMENT ON FUNCTION debversion_eq (debversion, debversion)
 257   IS 'debversion equal';""")
 258
 259         c.execute("""CREATE OR REPLACE FUNCTION debversion_ne (version1 debversion, version2 debversion)
 260   RETURNS boolean AS $$
 261 DECLARE
 262   comp integer;
 263   result boolean;
 264 BEGIN
 265   comp := debversion_compare(version1, version2);
 266   result := comp <> 0;
 267   RETURN result;
 268 END;
 269 $$
 270   LANGUAGE plpgsql
 271   IMMUTABLE STRICT;""")
 272         c.execute("""COMMENT ON FUNCTION debversion_ne (debversion, debversion)
 273   IS 'debversion not equal';""")
 274
 275         c.execute("""CREATE OR REPLACE FUNCTION debversion_lt (version1 debversion, version2 debversion)
 276   RETURNS boolean AS $$
 277 DECLARE
 278   comp integer;
 279   result boolean;
 280 BEGIN
 281   comp := debversion_compare(version1, version2);
 282   result := comp < 0;
 283   RETURN result;
 284 END;
 285 $$
 286   LANGUAGE plpgsql
 287   IMMUTABLE STRICT;""")
 288         c.execute("""COMMENT ON FUNCTION debversion_lt (debversion, debversion)
 289                    IS 'debversion less-than';""")
 290
 291         c.execute("""CREATE OR REPLACE FUNCTION debversion_gt (version1 debversion, version2 debversion) RETURNS boolean AS $$
 292 DECLARE
 293   comp integer;
 294   result boolean;
 295 BEGIN
 296   comp := debversion_compare(version1, version2);
 297   result := comp > 0;
 298   RETURN result;
 299 END;
 300 $$
 301   LANGUAGE plpgsql
 302   IMMUTABLE STRICT;""")
 303         c.execute("""COMMENT ON FUNCTION debversion_gt (debversion, debversion)
 304                    IS 'debversion greater-than';""")
 305
 306         c.execute("""CREATE OR REPLACE FUNCTION debversion_le (version1 debversion, version2 debversion)
 307   RETURNS boolean AS $$
 308 DECLARE
 309   comp integer;
 310   result boolean;
 311 BEGIN
 312   comp := debversion_compare(version1, version2);
 313   result := comp <= 0;
 314   RETURN result;
 315 END;
 316 $$
 317   LANGUAGE plpgsql
 318   IMMUTABLE STRICT;""")
 319         c.execute("""COMMENT ON FUNCTION debversion_le (debversion, debversion)
 320                    IS 'debversion less-than-or-equal';""")
 321
 322         c.execute("""CREATE OR REPLACE FUNCTION debversion_ge (version1 debversion, version2 debversion)
 323   RETURNS boolean AS $$
 324 DECLARE
 325   comp integer;
 326   result boolean;
 327 BEGIN
 328   comp := debversion_compare(version1, version2);
 329   result := comp >= 0;
 330   RETURN result;
 331 END;
 332 $$
 333   LANGUAGE plpgsql
 334   IMMUTABLE STRICT;""")
 335         c.execute("""COMMENT ON FUNCTION debversion_ge (debversion, debversion)
 336                    IS 'debversion greater-than-or-equal';""")
 337
 338         c.execute("""CREATE OPERATOR = (
 339                    PROCEDURE = debversion_eq,
 340                    LEFTARG = debversion,
 341                    RIGHTARG = debversion,
 342                    COMMUTATOR = =,
 343                    NEGATOR = !=);""")
 344         c.execute("""COMMENT ON OPERATOR = (debversion, debversion)
 345                    IS 'debversion equal';""")
 346
 347         c.execute("""CREATE OPERATOR != (
 348                    PROCEDURE = debversion_eq,
 349                    LEFTARG = debversion,
 350                    RIGHTARG = debversion,
 351                    COMMUTATOR = !=,
 352                    NEGATOR = =);""")
 353         c.execute("""COMMENT ON OPERATOR != (debversion, debversion)
 354                    IS 'debversion not equal';""")
 355
 356         c.execute("""CREATE OPERATOR < (
 357                    PROCEDURE = debversion_lt,
 358                    LEFTARG = debversion,
 359                    RIGHTARG = debversion,
 360                    COMMUTATOR = >,
 361                    NEGATOR = >=);""")
 362         c.execute("""COMMENT ON OPERATOR < (debversion, debversion)
 363                    IS 'debversion less-than';""")
 364
 365         c.execute("""CREATE OPERATOR > (
 366                    PROCEDURE = debversion_gt,
 367                    LEFTARG = debversion,
 368                    RIGHTARG = debversion,
 369                    COMMUTATOR = <,
 370                    NEGATOR = >=);""")
 371         c.execute("""COMMENT ON OPERATOR > (debversion, debversion)
 372                    IS 'debversion greater-than';""")
 373
 374         c.execute("""CREATE OPERATOR <= (
 375                    PROCEDURE = debversion_le,
 376                    LEFTARG = debversion,
 377                    RIGHTARG = debversion,
 378                    COMMUTATOR = >=,
 379                    NEGATOR = >);""")
 380         c.execute("""COMMENT ON OPERATOR <= (debversion, debversion)
 381                    IS 'debversion less-than-or-equal';""")
 382
 383         c.execute("""CREATE OPERATOR >= (
 384                    PROCEDURE = debversion_ge,
 385                    LEFTARG = debversion,
 386                    RIGHTARG = debversion,
 387                    COMMUTATOR = <=,
 388                    NEGATOR = <);""")
 389         c.execute("""COMMENT ON OPERATOR >= (debversion, debversion)
 390                    IS 'debversion greater-than-or-equal';""")
 391
 392         c.execute("ALTER TABLE source ALTER COLUMN version TYPE debversion;")
 393         c.execute("ALTER TABLE binaries ALTER COLUMN version TYPE debversion;")
 394
 395         c.execute("UPDATE config SET value = '2' WHERE name = 'db_revision'")
 396
 397         self.db.commit()
 398
 399     except psycopg2.ProgrammingError, msg:
 400         self.db.rollback()
 401         raise DBUpdateError, "Unable to appy debversion updates, rollback issued. Error message : %s" % (str(msg))