modules/ftptree.py

   1 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
   2
   3 import os, config, string, urllib, re
   4 from common import fileexists, noarchcachedir
   5 from baseftptree import BasePkg, BaseFtpTree
   6 errnum=0
   7
   8 SomeError="Oh no"
   9
  10 def bailoutonerror():
  11     if not errnum == 0:
  12         print "%d error(s) encountered... aborting" % errnum
  13         raise SomeError
  14
  15 def pinfo(msg):
  16     print msg
  17
  18 def perror(msg):
  19     global errnum
  20     errnum=errnum+1
  21     print msg
  22
  23 def pwarning(msg):
  24     print msg
  25
  26 def rm(file):
  27     os.remove(file)
  28     #print 'rm: '+file
  29
  30 def mv(src, dst):
  31     os.rename(src, dst+'/'+src.split('/')[-1])
  32     #print "mv: %s %s" % (src, dst+'/'+src.split('/')[-1])
  33
  34 class Pkg(BasePkg):
  35     def __init__(self, name, tree):
  36         self.marked4removal=False
  37         self.marked4moving=False
  38         BasePkg.__init__(self, name, tree)
  39
  40     def mark4moving(self):
  41         if not self.marked4moving:
  42             self.tree.marked4moving.append(self)
  43             self.marked4moving=True
  44
  45     def mark4removal(self):
  46         if not self.marked4removal:
  47             self.tree.marked4removal.append(self)
  48             self.marked4removal=True
  49
  50     def load(self, content=None):
  51         BasePkg.load(self, content)
  52         if self.info.has_key('move'):
  53             self.mark4moving()
  54
  55     def writeinfo(self):
  56         f=open(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', 'w')
  57         for bid in self.build.keys():
  58             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))
  59         for key in self.info.keys():
  60             f.write("info:%s:%s\n" % (key, string.join(self.info[key], ':')))
  61         for arch in self.files.keys():
  62             for rpm in self.files[arch]:
  63                 f.write("file:%s:%s\n" % (arch, rpm))
  64
  65     def remove(self):
  66         for arch in self.files.keys():
  67             for rpm in self.files[arch]:
  68                 rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
  69                 if arch=='noarch':
  70                     if fileexists(noarchcachedir+rpm+'.filelist'):
  71                         rm(noarchcachedir+rpm+'.filelist')
  72                     if fileexists(noarchcachedir+rpm+'.reqlist'):
  73                         rm(noarchcachedir+rpm+'.reqlist')
  74         rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
  75
  76     def move(self, dsttree):
  77         if dsttree.has_key(self.name):
  78             movedany=False
  79             for arch in self.files.keys():
  80                 if arch in dsttree[self.name].files.keys():
  81                     pinfo("Arch %s for %s is already present in dest tree; removing from srctree" % (arch, self.name))
  82                     for rpm in self.files[arch]:
  83                         rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
  84                 else:
  85                     movedany=True
  86                     dsttree[self.name].files[arch]=self.files[arch]
  87                     for rpm in self.files[arch]:
  88                         mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
  89             if movedany:
  90                 for bid in self.build.keys():
  91                     dsttree[self.name].build[bid]=self.build[bid]
  92                 dsttree[self.name].writeinfo()
  93             rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
  94         else:
  95             for arch in self.files.keys():
  96                 for rpm in self.files[arch]:
  97                     mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
  98             mv(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', dsttree.basedir+'/SRPMS/.metadata/')
  99
 100
 101 class FtpTree(BaseFtpTree):
 102     def __init__(self, tree, loadall=False):
 103         BaseFtpTree.__init__(self, tree)
 104         self.loadedpkgs={}
 105         self.marked4removal=[]
 106         self.marked4moving=[]
 107         self.pkgnames=[]
 108         self.__loadpkgnames()
 109         if loadall:
 110             for pkgname in self.pkgnames:
 111                 self.loadedpkgs[pkgname]=Pkg(pkgname, self)
 112         # Tests:
 113         self.do_checkbuild=True
 114
 115     def __getitem__(self, key):
 116         if self.loadedpkgs.has_key(key):
 117             return self.loadedpkgs[key]
 118         elif key in self.pkgnames:
 119             pkg=Pkg(key, self)
 120             self.loadedpkgs[key]=pkg
 121             return pkg
 122         else:
 123             raise KeyError, key
 124
 125     def has_key(self, key):
 126         if key in self.pkgnames:
 127             return True
 128         else:
 129             return False
 130
 131     def keys(self):
 132         return self.pkgnames
 133
 134     def testmove(self, dsttree):
 135         self.__checkbuild()
 136         self.__checkarchs(dsttree)
 137
 138     def movepkgs(self, dsttree):
 139         if self.do_checkbuild:
 140             self.__checkbuild()
 141         bailoutonerror()
 142         self.__checkarchs(dsttree)
 143         bailoutonerror()
 144         self.__rmolderfromsrc()
 145         self.__rmotherfromdst(dsttree)
 146
 147         for pkg in self.marked4moving:
 148             pkg.move(dsttree)
 149
 150     def removepkgs(self):
 151         for pkg in self.marked4removal:
 152             pkg.remove()
 153
 154     def mark4removal(self, wannabepkgs):
 155         self.__mark4something(wannabepkgs, Pkg.mark4removal)
 156
 157     def mark4moving(self, wannabepkgs):
 158         self.__mark4something(wannabepkgs, Pkg.mark4moving)
 159
 160
 161     # Internal functions below
 162
 163     def __loadpkgnames(self):
 164         def checkfiletype(name):
 165             if name[-13:]=='.src.rpm.info':
 166                 return True
 167             else:
 168                 return False
 169         list=filter(checkfiletype, os.listdir(self.basedir+'/SRPMS/.metadata'))
 170         self.pkgnames=map((lambda x: x[:-13]), list)
 171
 172     def __mark4something(self, wannabepkgs, markfunction):
 173         def chopoffextension(pkg):
 174             found=pkg.find('.src.rpm')
 175             if found==-1:
 176                 return pkg
 177             else:
 178                 return pkg[:found]
 179         for wannabepkg in wannabepkgs:
 180             pkgname=chopoffextension(wannabepkg)
 181             if pkgname in self.pkgnames:
 182                 if not pkgname in self.loadedpkgs.keys():
 183                     self.loadedpkgs[pkgname]=Pkg(pkgname, self)
 184                 markfunction(self.loadedpkgs[pkgname])
 185             else:
 186                 perror(pkgname+" was not found in source tree")
 187         bailoutonerror()
 188
 189     def __checkbuild(self):
 190         f=urllib.urlopen('http://ep09.pld-linux.org/~builderth/queue.txt')
 191         #f=open('queue.txt')
 192         requests={}
 193         reid=re.compile(r'^.*id=(.*) pri.*$')
 194         regb=re.compile(r'^group:.*$|builders:.*$', re.M)
 195         for i in re.findall(regb, f.read()):
 196             if i[0]=='g':
 197                 id=reid.sub(r'\1', i)
 198                 requests[id]=""
 199             elif i[0]=='b':
 200                 requests[id]=requests[id]+i
 201         f.close()
 202         for pkg in self.marked4moving:
 203             for bid in pkg.build.keys():
 204                 if requests.has_key(bid) and not requests[bid].find('?') == -1:
 205                     perror("Building of package %s (buildid %s) not finished" % (pkg,bid))
 206
 207     def __checkarchs(self, dsttree):
 208         for pkg in self.marked4moving:
 209             if len(pkg.files.keys()) <= 1:
 210                 perror('Package %s has only src.rpm built' % pkg)
 211                 continue
 212             otherpkgnames=self.__find_other_pkgs(pkg, dsttree)
 213             if otherpkgnames: # check if we're not removing some archs
 214                 curarchs=[]
 215                 missingarchs=[]
 216                 for somepkg in otherpkgnames:
 217                     curarchs.extend(Pkg(somepkg, dsttree).files.keys())
 218                 for arch in curarchs:
 219                     if arch not in pkg.files.keys():
 220                         missingarchs.append(arch)
 221                 if missingarchs:
 222                     perror('Moving %s would remove archs: %s' %
 223                                                             (pkg, missingarchs))
 224             else: # warn if a package isn't built for all archs
 225                 pass
 226
 227     def __rmolderfromsrc(self):
 228         for pkg in self.marked4moving:
 229             olderpkgnames=self.__find_older_pkgs(pkg)
 230             for i in olderpkgnames:
 231                 Pkg(i, self).remove()
 232
 233     def __rmotherfromdst(self, dsttree):
 234         for pkg in self.marked4moving:
 235             pkgnames=self.__find_other_pkgs(pkg, dsttree)
 236             for i in pkgnames:
 237                 Pkg(i, dsttree).remove()
 238
 239     # Used more than once filter functions
 240
 241     def __find_other_pkgs(self, pkg, tree):
 242         ziewre=re.compile(string.join(pkg.name.split('-')[:-2], '-')+'-[^-]*-[^-]*$')
 243         def filter_other_pkgs(x):
 244             if ziewre.match(x) and not x == pkg.name:
 245                 return True
 246             else:
 247                 return False
 248         return filter(filter_other_pkgs, tree.pkgnames)
 249
 250     def __find_older_pkgs(self, pkg):
 251         def filter_older_pkgs(x):
 252             checking=x.split('-')
 253             curpkg=pkg.name.split('-')
 254             if checking[-2]<curpkg[-2]:
 255                 return True
 256             elif checking[-2]==curpkg[-2]:
 257                 if checking[-1]<curpkg[-1]:
 258                     return True
 259                 else:
 260                     return False
 261             else:
 262                 return False
 263         return filter(filter_older_pkgs, self.__find_other_pkgs(pkg, self))
 264
 265