]> git.pld-linux.org Git - projects/pld-ftp-admin.git/blobdiff - modules/ftptree.py
- _RPMVSF_NOSIGNATURES and _RPMVSF_NODIGESTS have been removed in rpm 5
[projects/pld-ftp-admin.git] / modules / ftptree.py
index 8848f5ec93f3b2f39ea34d2911d58849768481bb..36970964e5fb0b207981768a00f878e6c54b1bdd 100644 (file)
@@ -1,11 +1,19 @@
 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
 
-import os, config, string, urllib, re
+import os, config, string, urllib, re, rpm
 from common import fileexists, noarchcachedir
 from baseftptree import BasePkg, BaseFtpTree
-errnum=0
+from sign import is_signed
 
-SomeError="Oh no"
+errnum = 0
+quietmode = False
+
+class SomeError(Exception):
+    def __init__(self):
+        return
+
+    def __str__(self):
+        print "","An Error occured!"
 
 def bailoutonerror():
     if not errnum == 0:
@@ -13,47 +21,129 @@ def bailoutonerror():
         raise SomeError
 
 def pinfo(msg):
-    print msg
+    print 'INFO: ' + msg
 
 def perror(msg):
     global errnum
-    errnum=errnum+1
+    errnum = errnum + 1
     print 'ERR: ' + msg
 
 def pwarning(msg):
     print 'WARN: ' + msg
 
-def rm(file):
-    os.remove(file)
-    #print 'rm: '+file
-
-def mv(src, dst):
-    os.rename(src, dst+'/'+src.split('/')[-1])
-    #print "mv: %s %s" % (src, dst+'/'+src.split('/')[-1])
+def rm(file, test = False):
+    if test:
+        if not os.path.exists(file):
+            pinfo("TEST os.remove(%s): file doesn't exists" % file)
+    else:
+        try:
+            os.remove(file)
+        except OSError, e:
+            pinfo("os.remove(%s): %s" % (file, e))
+            #raise
+
+def mv(src, dst, test = False):
+    fsrc = src
+    fdst = dst + '/' + src.split('/')[-1]
+    if test:
+        if not os.path.exists(src):
+            pinfo("TEST os.rename(%s, %s): source doesn't exists" % (fsrc, fdst))
+        if not os.path.exists(dst):
+            pinfo("TEST destination doesn't exist: %s" % dst)
+    else:
+        try:
+            os.rename(fsrc, fdst)
+        except OSError, e:
+            pinfo("os.rename(%s, %s): %s" % (fsrc, fdst, e))
+            raise
 
 class Pkg(BasePkg):
-    def __init__(self, name, tree):
-        self.marked4removal=False
-        self.marked4moving=False
-        BasePkg.__init__(self, name, tree)
+    def __init__(self, nvr, tree):
+        BasePkg.__init__(self, nvr, tree)
+        self.name = string.join(nvr.split('-')[:-2], '-')
+        self.version = nvr.split('-')[-2]
+        self.release = nvr.split('-')[-1]
+        self.marked4removal = False
+        self.marked4moving = False
+        self.marked4movingpool = []
+        self.errors = []
+        self.warnings = []
+
+    def __cmp__(self, pkg):
+        if self.name > pkg.name:
+            return 1
+        elif self.name < pkg.name:
+            return -1
+        else:
+            return rpm.labelCompare(('0', self.version, self.release),
+                                    ('0', pkg.version, pkg.release))
+
+
+    # unfortunately can't do new Pkg(NVR), and have no "tree" in this pkg context
+    # so this static function
+    def is_debuginfo(self, nvr):
+        """
+        returns true if NVR is debuginfo package and separate debuginfo is enabled
+        """
+        if not config.separate_debuginfo:
+            return False
+        pkg = nvr.split('-')[:-2]
+        return pkg[-1] == 'debuginfo'
+
+    def is_sourcefile(self, file):
+        """
+        returns true if file is source package
+        """
+        return file[-8:] == '.src.rpm'
+
+    # returns true if package build is integer
+    def is_release(self):
+        """
+        To account Release tags with subver macros, we consider integer release
+        if it contains odd number of dots:
+
+        1 -> True
+        0.1 -> False
+        0.%{subver}.%{rel}, %{rel} = 1 -> 0.20010.1 -> True
+        0.%{subver}.%{rel}, %{rel} = 0.1 -> 0.20010.0.1 -> False
+        """
+        return self.release.count('.') % 2 == 0
 
     def mark4moving(self):
         if not self.marked4moving:
+            # Only one pkg in this pool can be marked for moving
+            for pkg in self.marked4movingpool:
+                pkg.unmark4moving()
             self.tree.marked4moving.append(self)
             self.marked4moving=True
 
+    def unmark4moving(self):
+        if self.marked4moving:
+            self.tree.marked4moving.remove(self)
+            self.marked4moving=False
+
     def mark4removal(self):
         if not self.marked4removal:
             self.tree.marked4removal.append(self)
             self.marked4removal=True
 
+    def error(self, msg):
+        self.errors.append(msg)
+        if not quietmode:
+            perror('%s %s' % (self.nvr, msg))
+
+    def warning(self, msg):
+        self.warnings.append(msg)
+        if not quietmode:
+            pwarning('%s %s' % (self.nvr, msg))
+
     def load(self, content=None):
         BasePkg.load(self, content)
         if self.info.has_key('move'):
             self.mark4moving()
 
     def writeinfo(self):
-        f=open(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', 'w')
+        f = open(self.tree.basedir+'/SRPMS/.metadata/'+self.nvr+'.src.rpm.info', 'w')
         for bid in self.build.keys():
             f.write("info:build:%s:requester:%s\ninfo:build:%s:requester_email:%s\n" % (bid, self.build[bid].requester, bid, self.build[bid].requester_email))
         for key in self.info.keys():
@@ -62,55 +152,124 @@ class Pkg(BasePkg):
             for rpm in self.files[arch]:
                 f.write("file:%s:%s\n" % (arch, rpm))
         
-    def remove(self):
+    def remove(self, test = False):
+        """
+        Remove package from ftp
+        """
         for arch in self.files.keys():
             for rpm in self.files[arch]:
-                rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
-                if arch=='noarch':
-                    if fileexists(noarchcachedir+rpm+'.filelist'):
-                        rm(noarchcachedir+rpm+'.filelist')
-                    if fileexists(noarchcachedir+rpm+'.reqlist'):
-                        rm(noarchcachedir+rpm+'.reqlist')
-        rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
-
-    def move(self, dsttree):
-        if dsttree.has_key(self.name):
-            movedany=False
+                if self.is_debuginfo(rpm):
+                    rm(self.tree.basedir + '/' + arch + '/debuginfo/' + rpm, test)
+                else:
+                    rm(self.tree.basedir + '/' + arch + '/RPMS/' + rpm, test)
+                if arch == 'noarch':
+                    if fileexists(noarchcachedir + rpm + '.filelist'):
+                        rm(noarchcachedir + rpm + '.filelist', test)
+                    if fileexists(noarchcachedir + rpm + '.reqlist'):
+                        rm(noarchcachedir + rpm + '.reqlist', test)
+        rm(self.tree.basedir + '/SRPMS/.metadata/' + self.nvr + '.src.rpm.info', test)
+
+    def rpmfiles(self, debugfiles = True, sourcefiles  = True):
+        """
+        Return rpm files related to this package
+        """
+        files = []
+        for arch, rpms in self.files.items():
+            for nvr in rpms:
+                if self.is_debuginfo(nvr):
+                    if debugfiles:
+                        files.append(self.tree.basedir + '/' + arch + '/debuginfo/' + nvr)
+                else:
+                    if self.is_sourcefile(nvr):
+                        if sourcefiles:
+                            files.append(self.tree.basedir + '/' + arch + '/RPMS/' + nvr)
+                    else:
+                        files.append(self.tree.basedir + '/' + arch + '/RPMS/' + nvr)
+        return files
+
+    def obsoletes(self):
+        """
+        Return obsoletes for all packages in Pkg:
+
+        {'php-geshi': set(['geshi'])}
+
+        """
+        def rpmhdr(pkg):
+            ts = rpm.ts()
+            fdno = os.open(pkg, os.O_RDONLY)
+            hdr = ts.hdrFromFdno(fdno)
+            os.close(fdno)
+            return hdr
+
+        obsoletes = {}
+        for rpmfile in self.rpmfiles():
+            if not os.path.exists(rpmfile):
+                continue
+            hdr = rpmhdr(rpmfile)
+            if not hdr[rpm.RPMTAG_OBSOLETES]:
+                continue
+
+            name = hdr[rpm.RPMTAG_NAME]
+            if not name in obsoletes:
+                obsoletes[name] = set()
+
+            for tag in hdr[rpm.RPMTAG_OBSOLETES]:
+                obsoletes[name].add(tag)
+
+        return obsoletes
+
+    def move(self, dsttree, test = False):
+        if dsttree.has_key(self.nvr):
+            movedany = False
             for arch in self.files.keys():
-                if arch in dsttree[self.name].files.keys():
-                    pinfo("Arch %s for %s is already present in dest tree; removing from srctree" % (arch, self.name))
+                if arch in dsttree[self.nvr].files.keys():
+                    msg = ""
+                    if test:
+                        msg = "TEST "
+                    pinfo("%sArch %s for %s is already present in dest tree; removing from srctree" % (msg, arch, self.nvr))
                     for rpm in self.files[arch]:
-                        rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
+                        if self.is_debuginfo(rpm):
+                            rm(self.tree.basedir + '/' + arch + '/debuginfo/' + rpm, test)
+                        else:
+                            rm(self.tree.basedir + '/' + arch + '/RPMS/' + rpm, test)
                 else:
-                    movedany=True
-                    dsttree[self.name].files[arch]=self.files[arch]
+                    movedany = True
+                    dsttree[self.nvr].files[arch] = self.files[arch]
                     for rpm in self.files[arch]:
-                        mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
-            if movedany:
+                        if self.is_debuginfo(rpm):
+                            mv(self.tree.basedir + '/' + arch + '/debuginfo/' + rpm, dsttree.basedir + '/' + arch + '/debuginfo/', test)
+                        else:
+                            mv(self.tree.basedir + '/' + arch + '/RPMS/' + rpm, dsttree.basedir + '/' + arch + '/RPMS/', test)
+            if not test and movedany:
                 for bid in self.build.keys():
-                    dsttree[self.name].build[bid]=self.build[bid]
-                dsttree[self.name].writeinfo()
-            rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
+                    dsttree[self.nvr].build[bid] = self.build[bid]
+                dsttree[self.nvr].writeinfo()
+            rm(self.tree.basedir + '/SRPMS/.metadata/' + self.nvr + '.src.rpm.info', test)
         else:
+            # move files
             for arch in self.files.keys():
                 for rpm in self.files[arch]:
-                    mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
-            mv(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', dsttree.basedir+'/SRPMS/.metadata/')
+                    if self.is_debuginfo(rpm):
+                        mv(self.tree.basedir + '/' + arch + '/debuginfo/' + rpm, dsttree.basedir + '/' + arch + '/debuginfo/', test)
+                    else:
+                        mv(self.tree.basedir + '/' + arch + '/RPMS/' + rpm, dsttree.basedir + '/' + arch + '/RPMS/', test)
 
+            # move metadata
+            mv(self.tree.basedir + '/SRPMS/.metadata/' + self.nvr + '.src.rpm.info', dsttree.basedir + '/SRPMS/.metadata/', test)
 
 class FtpTree(BaseFtpTree):
     def __init__(self, tree, loadall=False):
         BaseFtpTree.__init__(self, tree)
-        self.loadedpkgs={}
-        self.marked4removal=[]
-        self.marked4moving=[]
-        self.pkgnames=[]
+        self.loadedpkgs = {}
+        self.marked4removal = []
+        self.marked4moving = []
+        self.pkgnames = []
         self.__loadpkgnames()
         if loadall:
             for pkgname in self.pkgnames:
-                self.loadedpkgs[pkgname]=Pkg(pkgname, self)
+                self.loadedpkgs[pkgname] = Pkg(pkgname, self)
         # Tests:
-        self.do_checkbuild=True
+        self.do_checkbuild = True
 
     def __getitem__(self, key):
         if self.loadedpkgs.has_key(key):
@@ -131,23 +290,62 @@ class FtpTree(BaseFtpTree):
     def keys(self):
         return self.pkgnames
 
-    def testmove(self, dsttree):
-        self.__checkbuild()
-        self.__checkarchs(dsttree)
+    def values(self):
+        return self.loadedpkgs.values()
+
+    def checktree(self, dsttree):
+        self.__checkbuild(self.loadedpkgs.values())
+        self.__checkarchs(dsttree, self.loadedpkgs.values())
+
+    def testmove(self, dsttree, archivetree = None):
+        self.__checkbuild(self.marked4moving)
+        self.__checkarchs(dsttree, self.marked4moving)
+        self.__checkduplicates(self.marked4moving)
+
+        self.__checksigns(dsttree, self.marked4moving, test = True)
+        self.__checkforobsoletes(dsttree, self.marked4moving, test = True)
+        self.__checkforrelease(dsttree, self.marked4moving, test = True)
+
+        if not self.treename.count("archive"):
+            self.__rmolderfromsrc(test = True)
 
-    def movepkgs(self, dsttree):
+        self.__rmotherfromdst(dsttree, test = True, archivetree = archivetree)
+
+        for pkg in self.marked4moving:
+            pkg.move(dsttree, test = True)
+
+    def movepkgs(self, dsttree, archivetree = None):
         if self.do_checkbuild:
-            self.__checkbuild()
+            self.__checkbuild(self.marked4moving)
         bailoutonerror()
-        self.__checkarchs(dsttree)
+
+        self.__checkarchs(dsttree, self.marked4moving)
         bailoutonerror()
-        self.__rmolderfromsrc()
-        self.__rmotherfromdst(dsttree)
+
+        self.__checksigns(dsttree, self.marked4moving)
+        bailoutonerror()
+
+        if not self.treename.count("archive"):
+            self.__rmolderfromsrc()
+
+        self.__rmotherfromdst(dsttree, archivetree = archivetree)
 
         for pkg in self.marked4moving:
             pkg.move(dsttree)
 
+    def rpmfiles(self, debugfiles = True, sourcefiles = True):
+        if self.do_checkbuild:
+            self.__checkbuild(self.marked4moving)
+
+        files = []
+        for pkg in self.marked4moving:
+            files += pkg.rpmfiles(debugfiles = debugfiles, sourcefiles = sourcefiles)
+        return files
+
     def removepkgs(self):
+        if self.do_checkbuild:
+            self.__checkbuild(self.marked4removal)
+        bailoutonerror()
         for pkg in self.marked4removal:
             pkg.remove()
 
@@ -159,6 +357,12 @@ class FtpTree(BaseFtpTree):
         
 
     # Internal functions below
+    def __arch_stringify(self, list):
+        ret = []
+        dist = config.ftp_dist;
+        for arch in list:
+            ret.append(dist + '-' + arch)
+        return ' '.join(ret)
 
     def __loadpkgnames(self):
         def checkfiletype(name):
@@ -166,18 +370,19 @@ class FtpTree(BaseFtpTree):
                 return True
             else:
                 return False
-        list=filter(checkfiletype, os.listdir(self.basedir+'/SRPMS/.metadata'))
-        self.pkgnames=map((lambda x: x[:-13]), list)
+        list = filter(checkfiletype, os.listdir(self.basedir+'/SRPMS/.metadata'))
+        self.pkgnames = map((lambda x: x[:-13]), list)
 
     def __mark4something(self, wannabepkgs, markfunction):
         def chopoffextension(pkg):
-            found=pkg.find('.src.rpm')
-            if found==-1:
+            found = pkg.find('.src.rpm')
+            if found == -1:
                 return pkg
             else:
                 return pkg[:found]
+
         for wannabepkg in wannabepkgs:
-            pkgname=chopoffextension(wannabepkg)
+            pkgname = chopoffextension(wannabepkg)
             if pkgname in self.pkgnames:
                 if not pkgname in self.loadedpkgs.keys():
                     self.loadedpkgs[pkgname]=Pkg(pkgname, self)
@@ -186,71 +391,93 @@ class FtpTree(BaseFtpTree):
                 perror('%s not found in source tree' % pkgname)
         bailoutonerror()
 
-    def __checkbuild(self):
-        f=urllib.urlopen('http://ep09.pld-linux.org/~builderth/queue.txt')
-        #f=open('queue.txt')
-        requests={}
-        reid=re.compile(r'^.*id=(.*) pri.*$')
-        regb=re.compile(r'^group:.*$|builders:.*$', re.M)
+    def __checkbuild(self, marked):
+        """
+        Checks queue file if all arches are built
+
+        Reads config.builderqueue to grab the info
+        """
+        f = urllib.urlopen(config.builderqueue)
+        requests = {}
+        reid = re.compile(r'^.*id=(.*) pri.*$')
+        regb = re.compile(r'^group:.*$|builders:.*$', re.M)
         for i in re.findall(regb, f.read()):
-            if i[0]=='g':
-                id=reid.sub(r'\1', i)
-                requests[id]=""
+            if i[0] == 'g':
+                id = reid.sub(r'\1', i)
+                requests[id] = ""
             elif i[0]=='b':
-                requests[id]=requests[id]+i
+                requests[id] = requests[id] + i
         f.close()
-        for pkg in self.marked4moving:
+
+        for pkg in marked:
             for bid in pkg.build.keys():
                 if requests.has_key(bid) and not requests[bid].find('?') == -1:
-                    perror("%s (buildid %s) building not finished" % (pkg,bid))
+                    pkg.error("(buildid %s) building not finished" % bid)
 
-    def __checkarchs(self, dsttree):
-        for pkg in self.marked4moving:
+    def __checkarchs(self, dsttree, marked):
+        """
+        Checks marked pkgs it is built on all archs.
+        """
+        for pkg in marked:
             if len(pkg.files.keys()) <= 1:
-                perror('%s has only src.rpm built' % pkg)
+                pkg.error('has only src.rpm built')
                 continue
-            otherpkgnames=self.__find_other_pkgs(pkg, dsttree)
-            if otherpkgnames: # check if we're not removing some archs
-                curarchs=[]
-                missingarchs=[]
+            otherpkgnames = self.__find_other_pkgs(pkg, dsttree)
+
+            # check if we're not removing some archs
+            if otherpkgnames:
+                curarchs = []
+                missingarchs = []
                 for somepkg in otherpkgnames:
                     curarchs.extend(Pkg(somepkg, dsttree).files.keys())
                 for arch in curarchs:
                     if arch not in pkg.files.keys():
                         missingarchs.append(arch)
                 if missingarchs:
-                    perror('%s moving would remove archs: %s' %
-                                                            (pkg, missingarchs))
-            else: # warn if a package isn't built for all archs
-                if (config.separate_noarch and 'noarch' in pkg.files.keys() and
-                                    len(pkg.files.keys())==2):
+                    pkg.error('moving would remove archs: %s' % self.__arch_stringify(missingarchs))
+            else:
+                # warn if a package isn't built for all archs
+                if (config.separate_noarch and 'noarch' in pkg.files.keys() and len(pkg.files.keys()) == 2):
                     continue
-                elif len(pkg.files.keys()) != len(config.ftp_archs)+1:
-                    missingarchs=[]
+                elif len(pkg.files.keys()) != len(config.ftp_archs) + 1:
+                    missingarchs = []
                     for arch in config.ftp_archs:
                         if arch not in pkg.files.keys():
                             missingarchs.append(arch)
-                    pwarning('%s not built for archs: %s' %
-                                                            (pkg, missingarchs))
+                    pkg.warning('not built for archs: %s' % self.__arch_stringify(missingarchs))
+
+    def __checkduplicates(self, marked):
+        """
+        Checks if marked packages contain duplicate packages (with different versions)
+        """
+        for pkg in marked:
+            olderpkgnames = self.__find_older_pkgs(pkg)
+            for i in olderpkgnames:
+                markednames = [str(x) for x in marked]
+                if i in markednames:
+                    pkg.error('duplicate package: %s' % i)
 
-    def __rmolderfromsrc(self):
+    def __rmolderfromsrc(self, test = False):
         for pkg in self.marked4moving:
-            olderpkgnames=self.__find_older_pkgs(pkg)
+            olderpkgnames = self.__find_older_pkgs(pkg)
             for i in olderpkgnames:
-                Pkg(i, self).remove()
+                Pkg(i, self).remove(test)
 
-    def __rmotherfromdst(self, dsttree):
+    def __rmotherfromdst(self, dsttree, test = False, archivetree = None):
         for pkg in self.marked4moving:
-            pkgnames=self.__find_other_pkgs(pkg, dsttree)
+            pkgnames = self.__find_other_pkgs(pkg, dsttree)
             for i in pkgnames:
-                Pkg(i, dsttree).remove()
+                if archivetree == None:
+                    Pkg(i, dsttree).remove(test)
+                else:
+                    Pkg(i, dsttree).move(archivetree, test = test)
 
     # Used more than once filter functions
-
     def __find_other_pkgs(self, pkg, tree):
-        ziewre=re.compile(string.join(pkg.name.split('-')[:-2], '-')+'-[^-]*-[^-]*$')
+        escapedpkgname = pkg.name.replace('.', '\.').replace('+', '\+')
+        ziewre = re.compile(escapedpkgname + '-[^-]*-[^-]*$')
         def filter_other_pkgs(x):
-            if ziewre.match(x) and not x == pkg.name:
+            if ziewre.match(x) and not x == pkg.nvr:
                 return True
             else:
                 return False
@@ -258,17 +485,71 @@ class FtpTree(BaseFtpTree):
 
     def __find_older_pkgs(self, pkg):
         def filter_older_pkgs(x):
-            checking=x.split('-')
-            curpkg=pkg.name.split('-')
-            if checking[-2]<curpkg[-2]:
+            c = x.split('-')
+            rc = rpm.labelCompare(('0', pkg.version, pkg.release),
+                                                        ('0', c[-2], c[-1]))
+            if rc == 1: # pkg > x
                 return True
-            elif checking[-2]==curpkg[-2]:
-                if checking[-1]<curpkg[-1]:
-                    return True
-                else:
-                    return False
             else:
                 return False
         return filter(filter_older_pkgs, self.__find_other_pkgs(pkg, self))
 
+    def __checksigns(self, tree, pkgs, test = False):
+        """
+        Checks if pkgs in tree are all signed.
+
+        in case of test = true, error flag is set for unsigned packages
+        """
+        if not tree.treename in config.signed_trees:
+            return
+
+        for pkg in pkgs:
+            unsigned = 0
+            for file in pkg.rpmfiles():
+                if not is_signed(file):
+                    unsigned += 1
+
+            if unsigned != 0:
+                if test == True:
+                    if not quietmode:
+                        pkg.warning('%d files not signed' % unsigned)
+                else:
+                    pkg.error('%d files not signed' % unsigned)
+
+    def __checkforobsoletes(self, tree, pkgs, test = False):
+        """
+        Checks queue file if package obsoletes something in destination tree and suggest for removal.
+
+        Only NAME tag is compared, i.e virtual packages do not get reported.
+
+        """
+        if test != True:
+            return
+
+        def findbyname(name):
+            def x(nvr):
+                return '-'.join(nvr.split('-')[:-2]) == name
+            return filter(x, tree.pkgnames)
+
+        for pkg in pkgs:
+            obsoletes = pkg.obsoletes()
+            if not obsoletes:
+                continue
+
+            for pn, setlist in obsoletes.items():
+                for item in setlist:
+                    p = findbyname(item)
+                    if p:
+                        pkg.warning('obsoletes %s (via %s) in dest tree, perhaps you want rmpkg' % (p,pn))
+
+    def __checkforrelease(self, tree, pkgs, test = False):
+        """
+        Checks queue file if package release is non integer.
+
+        """
+        if test != True:
+            return
 
+        for pkg in pkgs:
+            if not pkg.is_release():
+                pkg.warning('non-integer release: %s' % pkg.release)
This page took 0.06345 seconds and 4 git commands to generate.