1 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
3 import os, config, string, urllib, re
4 from common import fileexists, noarchcachedir
5 from baseftptree import BasePkg, BaseFtpTree
12 print "%d error(s) encountered... aborting" % errnum
28 os.rename(src, dst+'/'+src.split('/')[-1])
29 #print "mv: %s %s" % (src, dst+'/'+src.split('/')[-1])
32 def __init__(self, name, tree):
33 self.marked4removal=False
34 self.marked4moving=False
35 BasePkg.__init__(self, name, tree)
37 def mark4moving(self):
38 if not self.marked4moving:
39 self.tree.marked4moving.append(self)
40 self.marked4moving=True
42 def mark4removal(self):
43 if not self.marked4removal:
44 self.tree.marked4removal.append(self)
45 self.marked4removal=True
47 def load(self, content=None):
48 BasePkg.load(self, content)
49 if self.info.has_key('move'):
53 f=open(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', 'w')
54 for bid in self.build.keys():
55 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))
56 for key in self.info.keys():
57 f.write("info:%s:%s\n" % (key, string.join(self.info[key], ':')))
58 for arch in self.files.keys():
59 for rpm in self.files[arch]:
60 f.write("file:%s:%s\n" % (arch, rpm))
63 for arch in self.files.keys():
64 for rpm in self.files[arch]:
65 rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
67 if fileexists(noarchcachedir+rpm+'.filelist'):
68 rm(noarchcachedir+rpm+'.filelist')
69 if fileexists(noarchcachedir+rpm+'.reqlist'):
70 rm(noarchcachedir+rpm+'.reqlist')
71 rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
73 def move(self, dsttree):
74 if dsttree.has_key(self.name):
76 for arch in self.files.keys():
77 if arch in dsttree[self.name].files.keys():
78 pinfo("Arch %s for %s is already present in dest tree; removing from srctree" % (arch, self.name))
79 for rpm in self.files[arch]:
80 rm(self.tree.basedir+'/'+arch+'/RPMS/'+rpm)
83 dsttree[self.name].files[arch]=self.files[arch]
84 for rpm in self.files[arch]:
85 mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
87 for bid in self.build.keys():
88 dsttree[self.name].build[bid]=self.build[bid]
89 dsttree[self.name].writeinfo()
90 rm(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info')
92 for arch in self.files.keys():
93 for rpm in self.files[arch]:
94 mv(self.tree.basedir+'/'+arch+'/RPMS/'+rpm, dsttree.basedir+'/'+arch+'/RPMS/')
95 mv(self.tree.basedir+'/SRPMS/.metadata/'+self.name+'.src.rpm.info', dsttree.basedir+'/SRPMS/.metadata/')
98 class FtpTree(BaseFtpTree):
99 def __init__(self, tree, loadall=False):
100 BaseFtpTree.__init__(self, tree)
102 self.marked4removal=[]
103 self.marked4moving=[]
105 self.__loadpkgnames()
107 for pkgname in self.pkgnames:
108 self.loadedpkgs[pkgname]=Pkg(pkgname, self)
110 def __getitem__(self, key):
111 if self.loadedpkgs.has_key(key):
112 return self.loadedpkgs[key]
113 elif key in self.pkgnames:
115 self.loadedpkgs[key]=pkg
119 def has_key(self, key):
120 if key in self.pkgnames:
127 def testmove(self, dsttree):
129 self.__checkarchs(dsttree)
131 def movepkgs(self, dsttree):
134 self.__checkarchs(dsttree)
136 self.__rmolderfromsrc()
137 self.__rmotherfromdst(dsttree)
139 for pkg in self.marked4moving:
142 def removepkgs(self):
143 for pkg in self.marked4removal:
146 def mark4removal(self, wannabepkgs):
147 self.__mark4something(wannabepkgs, Pkg.mark4removal)
149 def mark4moving(self, wannabepkgs):
150 self.__mark4something(wannabepkgs, Pkg.mark4moving)
153 # Internal functions below
155 def __loadpkgnames(self):
156 def checkfiletype(name):
157 if name[-13:]=='.src.rpm.info':
161 list=filter(checkfiletype, os.listdir(self.basedir+'/SRPMS/.metadata'))
162 self.pkgnames=map((lambda x: x[:-13]), list)
164 def __mark4something(self, wannabepkgs, markfunction):
165 def chopoffextension(pkg):
166 found=pkg.find('.src.rpm')
171 for wannabepkg in wannabepkgs:
172 pkgname=chopoffextension(wannabepkg)
173 if pkgname in self.pkgnames:
174 if not pkgname in self.loadedpkgs.keys():
175 self.loadedpkgs[pkgname]=Pkg(pkgname, self)
176 markfunction(self.loadedpkgs[pkgname])
178 perror(pkgname+" was not found in source tree")
181 def __checkbuild(self):
182 f=urllib.urlopen('http://ep09.pld-linux.org/~builderth/queue.txt')
185 reid=re.compile(r'^.*id=(.*) pri.*$')
186 regb=re.compile(r'^group:.*$|builders:.*$', re.M)
187 for i in re.findall(regb, f.read()):
189 id=reid.sub(r'\1', i)
192 requests[id]=requests[id]+i
194 for pkg in self.marked4moving:
195 for bid in pkg.build.keys():
196 if requests.has_key(bid) and not requests[bid].find('?') == -1:
197 perror("Building of package %s (buildid %s) not finished" % (pkg,bid))
199 def __checkarchs(self, dsttree):
200 for pkg in self.marked4moving:
201 otherpkgnames=self.__find_other_pkgs(pkg, dsttree)
204 for somepkg in otherpkgnames:
205 curarchs.extend(Pkg(somepkg, dsttree).files.keys())
206 for arch in curarchs:
207 if arch not in pkg.files.keys():
208 missingarchs.append(arch)
210 perror('Moving %s would remove archs: %s' % (pkg, missingarchs))
212 def __rmolderfromsrc(self):
213 for pkg in self.marked4moving:
214 olderpkgnames=self.__find_older_pkgs(pkg)
215 for i in olderpkgnames:
216 Pkg(i, self).remove()
218 def __rmotherfromdst(self, dsttree):
219 for pkg in self.marked4moving:
220 pkgnames=self.__find_other_pkgs(pkg, dsttree)
222 Pkg(i, dsttree).remove()
224 # Used more than once filter functions
226 def __find_other_pkgs(self, pkg, tree):
227 ziewre=re.compile(string.join(pkg.name.split('-')[:-2], '-')+'-[^-]*-[^-]*$')
228 def filter_other_pkgs(x):
229 if ziewre.match(x) and not x == pkg.name:
233 return filter(filter_other_pkgs, tree.pkgnames)
235 def __find_older_pkgs(self, pkg):
236 def filter_older_pkgs(x):
237 checking=x.split('-')
238 curpkg=pkg.name.split('-')
239 if checking[-2]<curpkg[-2]:
241 elif checking[-2]==curpkg[-2]:
242 if checking[-1]<curpkg[-1]:
248 return filter(filter_older_pkgs, self.__find_other_pkgs(pkg, self))