dak/init_dirs.py

   1 #!/usr/bin/env python
   2
   3 """Initial setup of an archive."""
   4 # Copyright (C) 2002, 2004, 2006  James Troup <james@nocrew.org>
   5
   6 # This program is free software; you can redistribute it and/or modify
   7 # it under the terms of the GNU General Public License as published by
   8 # the Free Software Foundation; either version 2 of the License, or
   9 # (at your option) any later version.
  10
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15
  16 # You should have received a copy of the GNU General Public License
  17 # along with this program; if not, write to the Free Software
  18 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19
  20 ################################################################################
  21
  22 import os, sys
  23 import apt_pkg
  24 from daklib import utils
  25 from daklib.dbconn import *
  26
  27 ################################################################################
  28
  29 Cnf = None
  30
  31 ################################################################################
  32
  33 def usage(exit_code=0):
  34     """Print a usage message and exit with 'exit_code'."""
  35
  36     print """Usage: dak init-dirs
  37 Creates directories for an archive based on dak.conf configuration file.
  38
  39   -h, --help                show this help and exit."""
  40     sys.exit(exit_code)
  41
  42 ################################################################################
  43
  44 def do_dir(target, config_name):
  45     """If 'target' exists, make sure it is a directory.  If it doesn't, create
  46 it."""
  47
  48     if os.path.exists(target):
  49         if not os.path.isdir(target):
  50             utils.fubar("%s (%s) is not a directory."
  51                                % (target, config_name))
  52     else:
  53         print "Creating %s ..." % (target)
  54         os.makedirs(target)
  55
  56 def process_file(config, config_name):
  57     """Create directories for a config entry that's a filename."""
  58
  59     if config.has_key(config_name):
  60         target = os.path.dirname(config[config_name])
  61         do_dir(target, config_name)
  62
  63 def process_tree(config, tree):
  64     """Create directories for a config tree."""
  65
  66     for entry in config.SubTree(tree).List():
  67         entry = entry.lower()
  68         config_name = "%s::%s" % (tree, entry)
  69         target = config[config_name]
  70         do_dir(target, config_name)
  71
  72 def process_morguesubdir(subdir):
  73     """Create directories for morgue sub directories."""
  74
  75     config_name = "%s::MorgueSubDir" % (subdir)
  76     if Cnf.has_key(config_name):
  77         target = os.path.join(Cnf["Dir::Morgue"], Cnf[config_name])
  78         do_dir(target, config_name)
  79
  80 def process_keyring(fullpath, secret=False):
  81     """Create empty keyring if necessary."""
  82
  83     if os.path.exists(fullpath):
  84         return
  85
  86     keydir = os.path.dirname(fullpath)
  87
  88     if not os.path.isdir(keydir):
  89         print "Creating %s ..." % (keydir)
  90         os.makedirs(keydir)
  91         if secret:
  92             # Make sure secret keyring directories are 0700
  93             os.chmod(keydir, 0o700)
  94
  95     # Touch the file
  96     print "Creating %s ..." % (fullpath)
  97     file(fullpath, 'w')
  98     if secret:
  99         os.chmod(fullpath, 0o600)
 100     else:
 101         os.chmod(fullpath, 0o644)
 102
 103 ######################################################################
 104
 105 def create_directories():
 106     """Create directories referenced in dak.conf and apt.conf."""
 107
 108     session = DBConn().session()
 109
 110     # Process directories from dak.conf
 111     process_tree(Cnf, "Dir")
 112
 113     # Process queue directories
 114     for queue in session.query(PolicyQueue):
 115         do_dir(queue.path, '%s queue' % queue.queue_name)
 116
 117     for config_name in [ "Rm::LogFile",
 118                          "Import-Archive::ExportDir" ]:
 119         process_file(Cnf, config_name)
 120
 121     for subdir in [ "Clean-Queues", "Clean-Suites" ]:
 122         process_morguesubdir(subdir)
 123
 124     suite_suffix = "%s" % (Cnf.Find("Dinstall::SuiteSuffix"))
 125
 126     # Process secret keyrings
 127     if Cnf.has_key('Dinstall::SigningKeyring'):
 128         process_keyring(Cnf['Dinstall::SigningKeyring'], secret=True)
 129
 130     if Cnf.has_key('Dinstall::SigningPubKeyring'):
 131         process_keyring(Cnf['Dinstall::SigningPubKeyring'], secret=True)
 132
 133     # Process public keyrings
 134     for keyring in session.query(Keyring).all():
 135         process_keyring(keyring.keyring_name)
 136
 137     # Process pool directories
 138     for component in session.query(Component):
 139         directory = os.path.join( Cnf['Dir::Pool'], component.component_name )
 140
 141         do_dir(directory, '%s pool' % component.component_name)
 142
 143
 144     # Process dists directories
 145     # TODO: Store location of each suite in database
 146     for suite in session.query(Suite):
 147         suite_dir = os.path.join( Cnf['Dir::Root'], 'dists', "%s/%s" % (suite.suite_name, suite_suffix) )
 148
 149         # TODO: Store valid suite/component mappings in database
 150         for component in session.query(Component):
 151             component_name = component.component_name
 152
 153             sc_dir = os.path.join(suite_dir, component_name)
 154
 155             do_dir(sc_dir, "%s/%s" % (suite.suite_name, component_name))
 156
 157             for arch in suite.architectures:
 158                 if arch.arch_string == 'source':
 159                     arch_string = 'source'
 160                 else:
 161                     arch_string = 'binary-%s' % arch.arch_string
 162
 163                 suite_arch_dir = os.path.join(sc_dir, arch_string)
 164
 165                 do_dir(suite_arch_dir, "%s/%s/%s" % (suite.suite_name, component_name, arch_string))
 166
 167 ################################################################################
 168
 169 def main ():
 170     """Initial setup of an archive."""
 171
 172     global Cnf
 173
 174     Cnf = utils.get_conf()
 175     arguments = [('h', "help", "Init-Dirs::Options::Help")]
 176     for i in [ "help" ]:
 177         if not Cnf.has_key("Init-Dirs::Options::%s" % (i)):
 178             Cnf["Init-Dirs::Options::%s" % (i)] = ""
 179
 180     d = DBConn()
 181
 182     arguments = apt_pkg.ParseCommandLine(Cnf, arguments, sys.argv)
 183
 184     options = Cnf.SubTree("Init-Dirs::Options")
 185     if options["Help"]:
 186         usage()
 187     elif arguments:
 188         utils.warn("dak init-dirs takes no arguments.")
 189         usage(exit_code=1)
 190
 191     create_directories()
 192
 193 ################################################################################
 194
 195 if __name__ == '__main__':
 196     main()