]> git.decadent.org.uk Git - dak.git/blob - madison
Validate package name and version numbers. Add ~ as a non-taint character
[dak.git] / madison
1 #!/usr/bin/env python
2
3 # Display information about package(s) (suite, version, etc.)
4 # Copyright (C) 2000, 2001, 2002  James Troup <james@nocrew.org>
5 # $Id: madison,v 1.19 2002-06-05 00:20:16 troup Exp $
6
7 # This program is free software; you can redistribute it and/or modify
8 # it under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2 of the License, or
10 # (at your option) any later version.
11
12 # This program is distributed in the hope that it will be useful,
13 # but WITHOUT ANY WARRANTY; without even the implied warranty of
14 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15 # GNU General Public License for more details.
16
17 # You should have received a copy of the GNU General Public License
18 # along with this program; if not, write to the Free Software
19 # Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
20
21 ################################################################################
22
23 #   And, lo, a great and menacing voice rose from the depths, and with
24 #   great wrath and vehemence it's voice boomed across the
25 #   land... ``hehehehehehe... that *tickles*''
26 #                                                       -- aj on IRC
27
28 ################################################################################
29
30 import pg, string, sys;
31 import utils, db_access;
32 import apt_pkg;
33
34 ################################################################################
35
36 Cnf = None;
37 projectB = None;
38
39 ################################################################################
40
41 def arch_compare (a, b):
42     if a == "source" and b == "source":
43         return 0;
44     elif a == "source":
45         return -1;
46     elif b == "source":
47         return 1;
48
49     return cmp (a, b);
50
51 ################################################################################
52
53 def usage (exit_code=0):
54     print """Usage: madison [OPTION] PACKAGE[...]
55 Display information about PACKAGE(s).
56
57   -a, --architecture=ARCH    only show info for this architecture
58   -r, --regex                treat PACKAGE as a regex
59   -s, --suite=SUITE          only show info for this suite
60   -S, --source-and-binary    show info for the binary children of source pkgs
61   -h, --help                 show this help and exit
62
63 Both ARCH and SUITE can be space seperated lists, e.g.
64     --architecture=\"m68k i386\""""
65     sys.exit(exit_code)
66
67 ################################################################################
68
69 def main ():
70     global Cnf, projectB;
71
72     Cnf = utils.get_conf()
73
74     Arguments = [('a', "architecture", "Madison::Options::Architecture", "HasArg"),
75                  ('r', "regex", "Madison::Options::Regex"),
76                  ('s', "suite", "Madison::Options::Suite", "HasArg"),
77                  ('S', "source-and-binary", "Madison::Options::Source-And-Binary"),
78                  ('h', "help", "Madison::Options::Help")];
79     for i in ["architecture", "regex", "suite", "source-and-binary", "help" ]:
80         if not Cnf.has_key("Madison::Options::%s" % (i)):
81             Cnf["Madison::Options::%s" % (i)] = "";
82
83     packages = apt_pkg.ParseCommandLine(Cnf,Arguments,sys.argv);
84     Options = Cnf.SubTree("Madison::Options")
85
86     if Options["Help"]:
87         usage();
88     if not packages:
89         utils.fubar("need at least one package name as an argument.");
90
91     projectB = pg.connect(Cnf["DB::Name"], Cnf["DB::Host"], int(Cnf["DB::Port"]));
92     db_access.init(Cnf, projectB);
93
94     if Options["Regex"]:
95         comparison_operator = "~";
96     else:
97         comparison_operator = "=";
98
99     if Options["Suite"]:
100         suite_ids_list = [];
101         for suite in string.split(Options["Suite"]):
102             suite_id = db_access.get_suite_id(suite);
103             if suite_id == -1:
104                 utils.warn("suite '%s' not recognised." % (suite));
105             else:
106                 suite_ids_list.append(suite_id);
107         if suite_ids_list:
108             con_suites = "AND su.id IN (%s)" % string.join(map(str, suite_ids_list), ", ");
109         else:
110             utils.fubar("No correct suite given.");
111     else:
112         con_suites = "";
113
114     if Options["Architecture"]:
115         arch_ids_list = [];
116         check_source = 0;
117         for architecture in string.split(Options["Architecture"]):
118             if architecture == "source":
119                 check_source = 1;
120             architecture_id = db_access.get_architecture_id(architecture);
121             if architecture_id == -1:
122                 utils.warn("architecture '%s' not recognised." % (architecture));
123             else:
124                 arch_ids_list.append(architecture_id);
125         if arch_ids_list:
126             con_architectures = "AND a.id IN (%s)" % string.join(map(str, arch_ids_list), ", ");
127         else:
128             utils.fubar("No correct architecture given.");
129     else:
130         con_architectures = "";
131         check_source = 1;
132
133     if Options["Source-And-Binary"]:
134         new_packages = [];
135         for package in packages:
136             q = projectB.query("SELECT DISTINCT package FROM binaries WHERE EXISTS (SELECT s.source FROM source s WHERE binaries.source = s.id AND s.source %s '%s')" % (comparison_operator, package));
137             new_packages.extend(map(lambda x: x[0], q.getresult()));
138             new_packages.append(package);
139         packages = new_packages;
140
141     results = 0;
142     for package in packages:
143         q = projectB.query("SELECT b.package, b.version, a.arch_string, su.suite_name, m.name FROM binaries b, architecture a, suite su, bin_associations ba, maintainer m WHERE b.package %s '%s' AND a.id = b.architecture AND su.id = ba.suite AND b.id = ba.bin AND b.maintainer = m.id %s %s" % (comparison_operator, package, con_suites, con_architectures));
144         ql = q.getresult();
145         if check_source:
146             q = projectB.query("SELECT s.source, s.version, 'source', su.suite_name, m.name FROM source s, suite su, src_associations sa, maintainer m WHERE s.source %s '%s' AND su.id = sa.suite AND s.id = sa.source AND s.maintainer = m.id %s" % (comparison_operator, package, con_suites));
147             ql.extend(q.getresult());
148         d = {};
149         for i in ql:
150             results = results + 1;
151             pkg = i[0];
152             version = i[1];
153             architecture = i[2];
154             suite = i[3];
155             if not d.has_key(pkg):
156                 d[pkg] = {};
157             if not d[pkg].has_key(version):
158                 d[pkg][version] = {};
159             if not d[pkg][version].has_key(suite):
160                 d[pkg][version][suite] = [];
161             d[pkg][version][suite].append(architecture);
162
163         packages = d.keys();
164         packages.sort();
165         for pkg in packages:
166             versions = d[pkg].keys();
167             versions.sort(apt_pkg.VersionCompare);
168             for version in versions:
169                 suites = d[pkg][version].keys();
170                 suites.sort();
171                 for suite in suites:
172                     sys.stdout.write("%10s | %10s | %13s | " % (pkg, version, suite));
173                     arches = d[pkg][version][suite];
174                     arches.sort(arch_compare);
175                     sys.stdout.write(string.join(arches, ", "));
176                     sys.stdout.write('\n');
177
178     if not results:
179         sys.exit(1);
180
181 #######################################################################################
182
183 if __name__ == '__main__':
184     main()
185