dak/export_suite.py

   1 #! /usr/bin/env python
   2 #
   3 # Copyright (C) 2012, Ansgar Burchardt <ansgar@debian.org>
   4 #
   5 # This program is free software; you can redistribute it and/or modify
   6 # it under the terms of the GNU General Public License as published by
   7 # the Free Software Foundation; either version 2 of the License, or
   8 # (at your option) any later version.
   9 #
  10 # This program is distributed in the hope that it will be useful,
  11 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 # GNU General Public License for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License along
  16 # with this program; if not, write to the Free Software Foundation, Inc.,
  17 # 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18
  19 import apt_pkg
  20 import os
  21 import sys
  22
  23 from daklib.config import Config
  24 from daklib.dbconn import *
  25 from daklib.fstransactions import FilesystemTransaction
  26
  27 def usage():
  28     print """Usage: dak export-suite -s <suite> [options]
  29
  30 Export binaries and sources from a suite to a flat directory structure.
  31
  32  -c --copy         copy files instead of symlinking them
  33  -d <directory>    target directory to export packages to
  34                    default: current directory
  35  -s <suite>        suite to grab uploads from
  36 """
  37
  38 def main(argv=None):
  39     if argv is None:
  40         argv = sys.argv
  41
  42     arguments = [('h', 'help', 'Export::Options::Help'),
  43                  ('c', 'copy', 'Export::Options::Copy'),
  44                  ('d', 'directory', 'Export::Options::Directory', 'HasArg'),
  45                  ('s', 'suite', 'Export::Options::Suite', 'HasArg')]
  46
  47     cnf = Config()
  48     apt_pkg.parse_commandline(cnf.Cnf, arguments, argv)
  49     options = cnf.subtree('Export::Options')
  50
  51     if 'Help' in options or 'Suite' not in options:
  52         usage()
  53         sys.exit(0)
  54
  55     session = DBConn().session()
  56
  57     suite = session.query(Suite).filter_by(suite_name=options['Suite']).first()
  58     if suite is None:
  59         print "Unknown suite '{0}'".format(options['Suite'])
  60         sys.exit(1)
  61
  62     directory = options.get('Directory')
  63     if not directory:
  64         print "No target directory."
  65         sys.exit(1)
  66
  67     symlink = 'Copy' not in options
  68
  69     binaries = suite.binaries
  70     sources = suite.sources
  71
  72     files = []
  73     files.extend([ b.poolfile for b in binaries ])
  74     for s in sources:
  75         files.extend([ ds.poolfile for ds in s.srcfiles ])
  76
  77     with FilesystemTransaction() as fs:
  78         for f in files:
  79             dst = os.path.join(directory, f.basename)
  80             fs.copy(f.fullpath, dst, symlink=symlink)
  81         fs.commit()
  82
  83 if __name__ == '__main__':
  84     main()