import glob
from daklib import utils
-from daklib.dbconn import get_suite, get_suite_architectures
+from daklib.dbconn import Archive, Component, DBConn, Suite, get_suite, get_suite_architectures
#from daklib.regexes import re_includeinpdiff
import re
re_includeinpdiff = re.compile(r"(Translation-[a-zA-Z_]+\.(?:bz2|xz))")
Write out ed-style diffs to Packages/Source lists
-h, --help show this help and exit
+ -a <archive> generate diffs for suites in <archive>
-c give the canonical path of the file
-p name for the patch (defaults to current time)
- -r use a different archive root
-d name for the hardlink farm for status
-m how many diffs to generate
-n take no action
self.history_order = []
self.max = max
self.readpath = readpath
- self.filesizesha1 = None
+ self.filesizehashes = None
if readpath:
try:
f = open(readpath + "/Index")
x = f.readline()
- def read_hashs(ind, f, self, x=x):
+ def read_hashs(ind, hashind, f, self, x=x):
while 1:
x = f.readline()
if not x or x[0] != " ": break
l = x.split()
- if not self.history.has_key(l[2]):
- self.history[l[2]] = [None,None]
- self.history_order.append(l[2])
- self.history[l[2]][ind] = (l[0], int(l[1]))
+ fname = l[2]
+ if fname.endswith('.gz'):
+ fname = fname[:-3]
+ if not self.history.has_key(fname):
+ self.history[fname] = [None,None,None]
+ self.history_order.append(fname)
+ if not self.history[fname][ind]:
+ self.history[fname][ind] = (int(l[1]), None, None)
+ if hashind == 1:
+ self.history[fname][ind] = (int(self.history[fname][ind][0]), l[0], self.history[fname][ind][2])
+ else:
+ self.history[fname][ind] = (int(self.history[fname][ind][0]), self.history[fname][ind][1], l[0])
return x
while x:
continue
if l[0] == "SHA1-History:":
- x = read_hashs(0,f,self)
+ x = read_hashs(0,1,f,self)
+ continue
+
+ if l[0] == "SHA256-History:":
+ x = read_hashs(0,2,f,self)
continue
if l[0] == "SHA1-Patches:":
- x = read_hashs(1,f,self)
+ x = read_hashs(1,1,f,self)
+ continue
+
+ if l[0] == "SHA256-Patches:":
+ x = read_hashs(1,2,f,self)
+ continue
+
+ if l[0] == "SHA1-Download:":
+ x = read_hashs(2,1,f,self)
+ continue
+
+ if l[0] == "SHA256-Download:":
+ x = read_hashs(2,2,f,self)
continue
if l[0] == "Canonical-Name:" or l[0]=="Canonical-Path:":
self.can_path = l[1]
if l[0] == "SHA1-Current:" and len(l) == 3:
- self.filesizesha1 = (l[1], int(l[2]))
+ if not self.filesizehashes:
+ self.filesizehashes = (int(l[2]), None, None)
+ self.filesizehashes = (int(self.filesizehashes[0]), l[1], self.filesizehashes[2])
+
+ if l[0] == "SHA256-Current:" and len(l) == 3:
+ if not self.filesizehashes:
+ self.filesizehashes = (int(l[2]), None, None)
+ self.filesizehashes = (int(self.filesizehashes[0]), self.filesizehashes[2], l[1])
x = f.readline()
if self.can_path:
out.write("Canonical-Path: %s\n" % (self.can_path))
- if self.filesizesha1:
- out.write("SHA1-Current: %s %7d\n" % (self.filesizesha1))
+ if self.filesizehashes:
+ if self.filesizehashes[1]:
+ out.write("SHA1-Current: %s %7d\n" % (self.filesizehashes[1], self.filesizehashes[0]))
+ if self.filesizehashes[2]:
+ out.write("SHA256-Current: %s %7d\n" % (self.filesizehashes[2], self.filesizehashes[0]))
hs = self.history
l = self.history_order[:]
out.write("SHA1-History:\n")
for h in l:
- out.write(" %s %7d %s\n" % (hs[h][0][0], hs[h][0][1], h))
+ if hs[h][0] and hs[h][0][1]:
+ out.write(" %s %7d %s\n" % (hs[h][0][1], hs[h][0][0], h))
+ out.write("SHA256-History:\n")
+ for h in l:
+ if hs[h][0] and hs[h][0][2]:
+ out.write(" %s %7d %s\n" % (hs[h][0][2], hs[h][0][0], h))
out.write("SHA1-Patches:\n")
for h in l:
- out.write(" %s %7d %s\n" % (hs[h][1][0], hs[h][1][1], h))
+ if hs[h][1] and hs[h][1][1]:
+ out.write(" %s %7d %s\n" % (hs[h][1][1], hs[h][1][0], h))
+ out.write("SHA256-Patches:\n")
+ for h in l:
+ if hs[h][1] and hs[h][1][2]:
+ out.write(" %s %7d %s\n" % (hs[h][1][2], hs[h][1][0], h))
+ out.write("SHA1-Download:\n")
+ for h in l:
+ if hs[h][2] and hs[h][2][1]:
+ out.write(" %s %7d %s.gz\n" % (hs[h][2][1], hs[h][2][0], h))
+ out.write("SHA256-Download:\n")
+ for h in l:
+ if hs[h][2] and hs[h][2][2]:
+ out.write(" %s %7d %s.gz\n" % (hs[h][2][2], hs[h][2][0], h))
def create_temp_file(r):
f = tempfile.TemporaryFile()
f.seek(0)
return f
-def sizesha1(f):
+def sizehashes(f):
size = os.fstat(f.fileno())[6]
f.seek(0)
sha1sum = apt_pkg.sha1sum(f)
- return (sha1sum, size)
+ f.seek(0)
+ sha256sum = apt_pkg.sha256sum(f)
+ return (size, sha1sum, sha256sum)
def genchanges(Options, outdir, oldfile, origfile, maxdiffs = 56):
if Options.has_key("NoAct"):
return
oldf = smartopen(oldfile)
- oldsizesha1 = sizesha1(oldf)
+ oldsizehashes = sizehashes(oldf)
# should probably early exit if either of these checks fail
# alternatively (optionally?) could just trim the patch history
if os.path.exists(newfile): os.unlink(newfile)
smartlink(origfile, newfile)
newf = open(newfile, "r")
- newsizesha1 = sizesha1(newf)
+ newsizehashes = sizehashes(newf)
newf.close()
- if newsizesha1 == oldsizesha1:
+ if newsizehashes == oldsizehashes:
os.unlink(newfile)
oldf.close()
#print "%s: unchanged" % (origfile)
oldf.close()
difff = smartopen(difffile)
- difsizesha1 = sizesha1(difff)
+ difsizehashes = sizehashes(difff)
difff.close()
- upd.history[patchname] = (oldsizesha1, difsizesha1)
+ difffgz = open(difffile + ".gz", "r")
+ difgzsizehashes = sizehashes(difffgz)
+ difffgz.close()
+
+ upd.history[patchname] = (oldsizehashes, difsizehashes, difgzsizehashes)
upd.history_order.append(patchname)
- upd.filesizesha1 = newsizesha1
+ upd.filesizehashes = newsizehashes
os.unlink(oldfile + oldext)
os.link(origfile + origext, oldfile + origext)
os.unlink(newfile)
- f = open(outdir + "/Index", "w")
- upd.dump(f)
- f.close()
+ with open(outdir + "/Index.new", "w") as f:
+ upd.dump(f)
+ os.rename(outdir + "/Index.new", outdir + "/Index")
def main():
Cnf = utils.get_conf()
Arguments = [ ('h', "help", "Generate-Index-Diffs::Options::Help"),
+ ('a', 'archive', 'Generate-Index-Diffs::Options::Archive', 'hasArg'),
('c', None, "Generate-Index-Diffs::Options::CanonicalPath", "hasArg"),
('p', "patchname", "Generate-Index-Diffs::Options::PatchName", "hasArg"),
- ('r', "rootdir", "Generate-Index-Diffs::Options::RootDir", "hasArg"),
('d', "tmpdir", "Generate-Index-Diffs::Options::TempDir", "hasArg"),
('m', "maxdiffs", "Generate-Index-Diffs::Options::MaxDiffs", "hasArg"),
('n', "n-act", "Generate-Index-Diffs::Options::NoAct"),
format = "%Y-%m-%d-%H%M.%S"
Options["PatchName"] = time.strftime( format )
- AptCnf = apt_pkg.newConfiguration()
- apt_pkg.ReadConfigFileISC(AptCnf,utils.which_apt_conf_file())
-
- if Options.has_key("RootDir"):
- Cnf["Dir::Root"] = Options["RootDir"]
+ session = DBConn().session()
if not suites:
- suites = Cnf.subtree("Suite").list()
+ query = session.query(Suite.suite_name)
+ if Options.get('Archive'):
+ query = query.join(Suite.archive).filter(Archive.archive_name == Options['Archive'])
+ suites = [ s.suite_name for s in query ]
for suitename in suites:
print "Processing: " + suitename
- SuiteBlock = Cnf.subtree("Suite::" + suitename)
- suiteobj = get_suite(suitename.lower())
+ suiteobj = get_suite(suitename.lower(), session=session)
# Use the canonical version of the suite name
suite = suiteobj.suite_name
print "Skipping: " + suite + " (untouchable)"
continue
- architectures = get_suite_architectures(suite, skipall=True)
-
- if SuiteBlock.has_key("Components"):
- components = SuiteBlock.value_list("Components")
- else:
- components = []
+ architectures = get_suite_architectures(suite, skipall=True, session=session)
+ components = [ c.component_name for c in session.query(Component.component_name) ]
suite_suffix = Cnf.find("Dinstall::SuiteSuffix")
if components and suite_suffix:
else:
longsuite = suite
- tree = SuiteBlock.get("Tree", "dists/%s" % (longsuite))
-
- if AptCnf.has_key("tree::%s" % (tree)):
- sections = AptCnf["tree::%s::Sections" % (tree)].split()
- elif AptCnf.has_key("bindirectory::%s" % (tree)):
- sections = AptCnf["bindirectory::%s::Sections" % (tree)].split()
- else:
- aptcnf_filename = os.path.basename(utils.which_apt_conf_file())
- print "ALERT: suite %s not in %s, nor untouchable!" % (suite, aptcnf_filename)
- continue
+ tree = os.path.join(suiteobj.archive.path, 'dists', longsuite)
# See if there are Translations which might need a new pdiff
cwd = os.getcwd()
- for component in sections:
+ for component in components:
#print "DEBUG: Working on %s" % (component)
- workpath=os.path.join(Cnf["Dir::Root"], tree, component, "i18n")
+ workpath=os.path.join(tree, component, "i18n")
if os.path.isdir(workpath):
os.chdir(workpath)
for dirpath, dirnames, filenames in os.walk(".", followlinks=True, topdown=True):
for archobj in architectures:
architecture = archobj.arch_string
- # use sections instead of components since dak.conf
- # treats "foo/bar main" as suite "foo", suitesuffix "bar" and
- # component "bar/main". suck.
-
- for component in sections:
+ for component in components:
if architecture == "source":
longarch = architecture
packages = "Sources"
maxsuite = maxsources
else:
- longarch = "binary-%s"% (architecture)
+ longarch = "binary-%s" % (architecture)
packages = "Packages"
maxsuite = maxpackages
- # Process Contents
- file = "%s/%s/Contents-%s" % (Cnf["Dir::Root"] + tree, component,
- architecture)
- storename = "%s/%s_%s_contents_%s" % (Options["TempDir"], suite, component, architecture)
- genchanges(Options, file + ".diff", storename, file, \
- Cnf.get("Suite::%s::Generate-Index-Diffs::MaxDiffs::Contents" % (suite), maxcontents))
-
- file = "%s/%s/%s/%s" % (Cnf["Dir::Root"] + tree,
- component, longarch, packages)
+
+ # Process Contents
+ file = "%s/%s/Contents-%s" % (tree, component, architecture)
+ storename = "%s/%s_%s_contents_%s" % (Options["TempDir"], suite, component, architecture)
+ genchanges(Options, file + ".diff", storename, file, maxcontents)
+
+ file = "%s/%s/%s/%s" % (tree, component, longarch, packages)
storename = "%s/%s_%s_%s" % (Options["TempDir"], suite, component, architecture)
- genchanges(Options, file + ".diff", storename, file, \
- Cnf.get("Suite::%s::Generate-Index-Diffs::MaxDiffs::%s" % (suite, packages), maxsuite))
+ genchanges(Options, file + ".diff", storename, file, maxsuite)
################################################################################