3 """Initial setup of an archive."""
4 # Copyright (C) 2002, 2004, 2006 James Troup <james@nocrew.org>
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.
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.
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
20 ################################################################################
24 from daklib import utils
25 from daklib.dbconn import *
27 ################################################################################
31 ################################################################################
33 def usage(exit_code=0):
34 """Print a usage message and exit with 'exit_code'."""
36 print """Usage: dak init-dirs
37 Creates directories for an archive based on dak.conf configuration file.
39 -h, --help show this help and exit."""
42 ################################################################################
44 def do_dir(target, config_name):
45 """If 'target' exists, make sure it is a directory. If it doesn't, create
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))
53 print "Creating %s ..." % (target)
56 def process_file(config, config_name):
57 """Create directories for a config entry that's a filename."""
59 if config.has_key(config_name):
60 target = os.path.dirname(config[config_name])
61 do_dir(target, config_name)
63 def process_tree(config, tree):
64 """Create directories for a config tree."""
66 for entry in config.subtree(tree).list():
68 config_name = "%s::%s" % (tree, entry)
69 target = config[config_name]
70 do_dir(target, config_name)
72 def process_morguesubdir(subdir):
73 """Create directories for morgue sub directories."""
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)
80 def process_keyring(fullpath, secret=False):
81 """Create empty keyring if necessary."""
83 if os.path.exists(fullpath):
86 keydir = os.path.dirname(fullpath)
88 if not os.path.isdir(keydir):
89 print "Creating %s ..." % (keydir)
92 # Make sure secret keyring directories are 0700
93 os.chmod(keydir, 0o700)
96 print "Creating %s ..." % (fullpath)
99 os.chmod(fullpath, 0o600)
101 os.chmod(fullpath, 0o644)
103 ######################################################################
105 def create_directories():
106 """Create directories referenced in dak.conf."""
108 session = DBConn().session()
110 # Process directories from dak.conf
111 process_tree(Cnf, "Dir")
113 # Process queue directories
114 for queue in session.query(PolicyQueue):
115 do_dir(queue.path, '%s queue' % queue.queue_name)
117 for config_name in [ "Rm::LogFile",
118 "Import-Archive::ExportDir" ]:
119 process_file(Cnf, config_name)
121 for subdir in [ "Clean-Queues", "Clean-Suites" ]:
122 process_morguesubdir(subdir)
124 suite_suffix = "%s" % (Cnf.find("Dinstall::SuiteSuffix"))
126 # Process secret keyrings
127 if Cnf.has_key('Dinstall::SigningKeyring'):
128 process_keyring(Cnf['Dinstall::SigningKeyring'], secret=True)
130 if Cnf.has_key('Dinstall::SigningPubKeyring'):
131 process_keyring(Cnf['Dinstall::SigningPubKeyring'], secret=True)
133 # Process public keyrings
134 for keyring in session.query(Keyring).filter_by(active=True):
135 process_keyring(keyring.keyring_name)
137 # Process dists directories
138 # TODO: Store location of each suite in database
139 for suite in session.query(Suite):
140 suite_dir = os.path.join(suite.archive.path, 'dists', suite.suite_name, suite_suffix)
142 # TODO: Store valid suite/component mappings in database
143 for component in session.query(Component):
144 component_name = component.component_name
146 sc_dir = os.path.join(suite_dir, component_name)
148 do_dir(sc_dir, "%s/%s" % (suite.suite_name, component_name))
150 for arch in suite.architectures:
151 if arch.arch_string == 'source':
152 arch_string = 'source'
154 arch_string = 'binary-%s' % arch.arch_string
156 suite_arch_dir = os.path.join(sc_dir, arch_string)
158 do_dir(suite_arch_dir, "%s/%s/%s" % (suite.suite_name, component_name, arch_string))
160 ################################################################################
163 """Initial setup of an archive."""
167 Cnf = utils.get_conf()
168 arguments = [('h', "help", "Init-Dirs::Options::Help")]
170 if not Cnf.has_key("Init-Dirs::Options::%s" % (i)):
171 Cnf["Init-Dirs::Options::%s" % (i)] = ""
175 arguments = apt_pkg.parse_commandline(Cnf, arguments, sys.argv)
177 options = Cnf.subtree("Init-Dirs::Options")
181 utils.warn("dak init-dirs takes no arguments.")
186 ################################################################################
188 if __name__ == '__main__':