]> git.decadent.org.uk Git - dak.git/blob - dak/init_dirs.py
dinstall
[dak.git] / 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
26 ################################################################################
27
28 Cnf = None
29 AptCnf = 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         if tree == "Dir":
69             if entry in [ "poolroot", "queue" , "morguereject" ]:
70                 continue
71         config_name = "%s::%s" % (tree, entry)
72         target = config[config_name]
73         do_dir(target, config_name)
74
75 def process_morguesubdir(subdir):
76     """Create directories for morgue sub directories."""
77
78     config_name = "%s::MorgueSubDir" % (subdir)
79     if Cnf.has_key(config_name):
80         target = os.path.join(Cnf["Dir::Morgue"], Cnf[config_name])
81         do_dir(target, config_name)
82
83 ######################################################################
84
85 def create_directories():
86     """Create directories referenced in dak.conf and apt.conf."""
87
88     # Process directories from dak.conf
89     process_tree(Cnf, "Dir")
90     process_tree(Cnf, "Dir::Queue")
91     for config_name in [ "Dinstall::LockFile", "Rm::LogFile",
92                          "Import-Archive::ExportDir" ]:
93         process_file(Cnf, config_name)
94     for subdir in [ "Clean-Queues", "Clean-Suites" ]:
95         process_morguesubdir(subdir)
96
97     # Process directories from apt.conf
98     process_tree(AptCnf, "Dir")
99     for tree in AptCnf.SubTree("Tree").List():
100         config_name = "Tree::%s" % (tree)
101         tree_dir = os.path.join(Cnf["Dir::Root"], tree)
102         do_dir(tree_dir, tree)
103         for filename in [ "FileList", "SourceFileList" ]:
104             process_file(AptCnf, "%s::%s" % (config_name, filename))
105         for component in AptCnf["%s::Sections" % (config_name)].split():
106             for architecture in AptCnf["%s::Architectures" \
107                                        % (config_name)].split():
108                 if architecture != "source":
109                     architecture = "binary-"+architecture
110                 target = os.path.join(tree_dir, component, architecture)
111                 do_dir(target, "%s, %s, %s" % (tree, component, architecture))
112
113
114 ################################################################################
115
116 def main ():
117     """Initial setup of an archive."""
118
119     global AptCnf, Cnf
120
121     Cnf = utils.get_conf()
122     arguments = [('h', "help", "Init-Dirs::Options::Help")]
123     for i in [ "help" ]:
124         if not Cnf.has_key("Init-Dirs::Options::%s" % (i)):
125             Cnf["Init-Dirs::Options::%s" % (i)] = ""
126
127     arguments = apt_pkg.ParseCommandLine(Cnf, arguments, sys.argv)
128
129     options = Cnf.SubTree("Init-Dirs::Options")
130     if options["Help"]:
131         usage()
132     elif arguments:
133         utils.warn("dak init-dirs takes no arguments.")
134         usage(exit_code=1)
135
136     AptCnf = apt_pkg.newConfiguration()
137     apt_pkg.ReadConfigFileISC(AptCnf, utils.which_apt_conf_file())
138
139     create_directories()
140
141 ################################################################################
142
143 if __name__ == '__main__':
144     main()