bin/pfa-genindex

   1 #!/usr/bin/env python
   2 # vi: encoding=utf-8 ts=8 sts=4 sw=4 et
   3
   4 import getopt
   5 import sys, os
   6 sys.path.insert(0, os.environ['HOME']+'/pld-ftp-admin/modules')
   7 from common import checkdir
   8 from config import ftp_dir,all_ftp_archs
   9 import config
  10 import ftpio
  11
  12 try:
  13     opts, args = getopt.getopt(sys.argv[1:], 'q',
  14         [
  15             "quiet", "index=",
  16             "nopoldek", "noyum", "norpmrepo",
  17             "poldek", "yum", "rpmrepo"
  18         ]
  19     )
  20 except getopt.GetoptError:
  21     print >>sys.stderr, "ERR: not enough parameters given"
  22     print >>sys.stderr, "gen-indexes.py [--quiet] [--[no]poldek] [--[no]yum] [--[no]rpmrepo] tree [tree2...]"
  23     sys.exit(1)
  24
  25 do_poldek = True
  26 do_yum = False
  27 do_rpmrepo = False
  28 quiet = False
  29
  30 for o, a in opts:
  31     if o == "--nopoldek":
  32         do_poldek = False
  33     if o == "--noyum":
  34         do_yum = False
  35     if o == "--norpmrepo":
  36         do_rpmrepo = False
  37
  38     if o == "--poldek":
  39         do_poldek = True
  40     if o == "--yum":
  41         do_yum = True
  42     if o == "--rpmrepo":
  43         do_rpmrepo = True
  44
  45     if o == "-q" or o == "--quiet":
  46         quiet = True
  47
  48     if o == "--index":
  49         do_poldek = do_yum = do_rpmrepo = False
  50         for v in a.split(','):
  51             if v == 'poldek':
  52                 do_poldek = True
  53             if v == 'yum':
  54                 do_yum = True
  55             if v == 'rpmrepo':
  56                 do_rpmrepo = True
  57
  58 if not quiet:
  59     print "poldek: %s; yum: %s; rpmrepo: %s" % (do_poldek, do_yum, do_rpmrepo)
  60
  61 if not do_poldek and not do_yum and not do_rpmrepo:
  62     print >>sys.stderr, "ERR: speciy at least one action"
  63     sys.exit(1)
  64
  65 trees = args
  66
  67 for tree in trees:
  68     checkdir(tree)
  69
  70 ftpio.connect('gen-indexes')
  71
  72 locked = []
  73
  74 for tree in trees:
  75     if ftpio.lock(tree, True):
  76         locked.append(tree)
  77     else:
  78         if not quiet:
  79             print >>sys.stderr, "ERR: %s tree already locked" % tree
  80         for i in locked:
  81             ftpio.unlock(i)
  82         sys.exit(1)
  83
  84 home = os.environ['HOME']
  85
  86 os.umask(022)
  87 os.nice(19)
  88
  89 if do_poldek:
  90     poldek = '%s.stat/bin/poldek-new --cachedir=%s/tmp/poldek --conf %s.stat/etc/poldek.conf --mkidxz' % (ftp_dir, home, ftp_dir)
  91
  92     for tree in trees:
  93         print '\n-------------------------- %s --------------------------' % tree
  94         for arch in all_ftp_archs:
  95             print '\ngenerate poldek index for %s' % arch
  96             os.system('%s -s %s%s/%s/RPMS/ --mkidxz --mkidx-type pndir' % (poldek, ftp_dir, tree, arch))
  97             if arch != 'noarch' and config.separate_debuginfo:
  98                 os.system('%s -s %s%s/%s/debuginfo/ --mkidxz --mkidx-type pndir' % (poldek, ftp_dir, tree, arch))
  99
 100 if do_yum:
 101     os.system('%s cd %s.stat/repodata && cvs %s up comps.xml' % ("" if quiet else "set -x;", ftp_dir, "" if quiet else "-Q"))
 102     yum = '%s.stat/bin/createrepo -d -v --skip-stat --workers=12 -g %s.stat/repodata/comps.xml' % (ftp_dir, ftp_dir)
 103     for tree in trees:
 104         print '\n-------------------------- %s --------------------------' % tree
 105         cachedir = '%s/tmp/createrepo/%s' % (home, tree)
 106         treedir = "%s%s" % (ftp_dir, tree)
 107         for arch in all_ftp_archs:
 108             print '\ngenerate repodata for %s using createrepo' % arch
 109             archdir = "%s/%s" % (treedir, arch)
 110             print ('%s %s --cache %s-%s %s/RPMS' % ("" if quiet else "time", yum, cachedir, arch, archdir))
 111             os.system('%s %s --cache %s-%s %s/RPMS' % ("" if quiet else "time", yum, cachedir, arch, archdir))
 112             if arch != 'noarch' and config.separate_debuginfo:
 113                 os.system('%s %s --cache %s-%s %s/debuginfo' % ("" if quiet else "time", yum, cachedir, arch, archdir))
 114
 115 if do_rpmrepo:
 116     os.system('%s cd %s.stat/repodata && cvs %s up comps.xml' % ("" if quiet else "set -x;", ftp_dir, "" if quiet else "-Q"))
 117     for tree in trees:
 118         print '\n-------------------------- %s --------------------------' % tree
 119         for arch in all_ftp_archs:
 120             dir = '%s/%s/%s/RPMS' % (ftp_dir, tree, arch)
 121             if not quiet:
 122                 print '\ngenerate repodata for %s using rpmrepo (in %s)' % (arch, dir)
 123             os.system('%s rpmrepo %s -o %s' % ("" if quiet else "set -x;", dir, dir))
 124             if not quiet:
 125                 print 'copy comps.xml'
 126             comps = '%s.stat/repodata/comps.xml' % ftp_dir
 127             os.system('%s cp -p %s %s/repodata' % ("" if quiet else "set -x;", comps, dir))
 128
 129 for tree in trees:
 130     ftpio.unlock(tree)