summaryrefslogtreecommitdiffstats
path: root/scripts/maketreeinfo.py
blob: c53a672ea28c7dd64287d356796cf39ef62b9355 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
#!/usr/bin/python
#
# makes a .treeinfo file.  if information isn't provided, emit some warnings.
#
# Copyright (C) 2007  Red Hat, Inc.  All rights reserved.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Author(s): Will Woods <wwoods@redhat.com>
#

import os,sys,string
import getopt
import time
import ConfigParser


def usage():
    args = ""
    for key in data:
        args = "%s [--%s=%s]" %(args, key, key)
    print "%s: %s" % (sys.argv[0], args)
    sys.exit(1)

# TODO: add composeid, images, etc.
# TODO: take releasestr as an option and break it up into family/variant/version?

data = {"timestamp": time.time(),
        "family": None,
        "variant": None,
        "version": None,
        "arch": None,
        "discnum": None,
        "totaldiscs": None,
        "packagedir": None,
        "outfile": None}
allDiscs = None

opts = []
for key in data.keys():
    opts.append("%s=" % (key,))
opts.append("allDiscs")

(args, extra) = getopt.getopt(sys.argv[1:], '', opts)
if len(extra) > 0:
    print "had extra args: %s" % extra
    usage()

for (str, arg) in args:
    if str[2:] in data.keys():
        data[str[2:]] = arg
    elif str == "--allDiscs":
        allDiscs = 1
    else:
	print "unknown str of ", str
        usage()

# Make sure timestamp is actually a float
if type(data["timestamp"]) != float:
    data["timestamp"] = float(data["timestamp"])

if data["family"] is None:
    print >> sys.stderr, "--family missing! This is probably bad!"
    data["family"] = ""

if data["variant"] is None:
    print >> sys.stderr, "--variant missing, but that's OK."
    data["variant"] = ""

if data["version"] is None:
    print >> sys.stderr, "--version missing! This is probably bad!"
    data["version"] = ""

if data["arch"] is None:
    print >> sys.stderr, "--arch missing! This is probably bad!"
    data["arch"] = ""
    
if data["discnum"] is None and allDiscs is None:
    print >> sys.stderr, "--discnum missing; assuming disc 1"
    data["discnum"] = "1"

if data["totaldiscs"] is None and allDiscs is None:
    print >> sys.stderr, "--totaldiscs missing; assuming 1"
    data["totaldiscs"] = "1"

if data["packagedir"] is None:
    print >> sys.stderr, "--packagedir missing. This might cause some weirdness."
    data["packagedir"] = ""


if data["outfile"] is None:
    f = sys.stdout
else:
    f = open(data["outfile"], "w")

section='general'
c=ConfigParser.ConfigParser()
c.add_section(section)
for k,v in data.items(): 
    if k != 'outfile':
        c.set(section,k,v)
c.write(f)