Statistics
| Branch: | Tag: | Revision:

root / image_creator / main.py @ f165adc0

History | View | Annotate | Download (6.5 kB)

1
#!/usr/bin/env python
2

    
3
# Copyright 2011 GRNET S.A. All rights reserved.
4
#
5
# Redistribution and use in source and binary forms, with or
6
# without modification, are permitted provided that the following
7
# conditions are met:
8
#
9
#   1. Redistributions of source code must retain the above
10
#      copyright notice, this list of conditions and the following
11
#      disclaimer.
12
#
13
#   2. Redistributions in binary form must reproduce the above
14
#      copyright notice, this list of conditions and the following
15
#      disclaimer in the documentation and/or other materials
16
#      provided with the distribution.
17
#
18
# THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS
19
# OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
20
# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
21
# PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR
22
# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
23
# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
24
# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
25
# USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
26
# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27
# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
28
# ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
29
# POSSIBILITY OF SUCH DAMAGE.
30
#
31
# The views and conclusions contained in the software and
32
# documentation are those of the authors and should not be
33
# interpreted as representing official policies, either expressed
34
# or implied, of GRNET S.A.
35

    
36
from image_creator import get_os_class
37
from image_creator import __version__ as version
38
from image_creator.disk import Disk
39
from image_creator.util import get_command, error, success, output, FatalError
40
from image_creator import util
41
import sys
42
import os
43
import optparse
44

    
45
dd = get_command('dd')
46

    
47

    
48
def check_writable_dir(option, opt_str, value, parser):
49
    dirname = os.path.dirname(value)
50
    name = os.path.basename(value)
51
    if dirname and not os.path.isdir(dirname):
52
        parser.error("`%s' is not an existing directory" % dirname)
53

    
54
    if not name:
55
        parser.error("`%s' is not a valid file name" % dirname)
56

    
57
    setattr(parser.values, option.dest, value)
58

    
59

    
60
def parse_options(input_args):
61
    usage = "Usage: %prog [options] <input_media>"
62
    parser = optparse.OptionParser(version=version, usage=usage)
63

    
64
    parser.add_option("-f", "--force", dest="force", default=False,
65
        action="store_true", help="overwrite output files if they exist")
66

    
67
    parser.add_option("--no-sysprep", dest="sysprep", default=True,
68
        help="don't perform system preperation", action="store_false")
69

    
70
    parser.add_option("--no-shrink", dest="shrink", default=True,
71
        help="don't shrink any partition", action="store_false")
72

    
73
    parser.add_option("-o", "--outfile", type="string", dest="outfile",
74
        default=None, action="callback", callback=check_writable_dir,
75
        help="dump image to FILE", metavar="FILE")
76

    
77
    parser.add_option("--enable-sysprep", dest="enabled_syspreps", default=[],
78
        help="Run SYSPREP operation on the input media",
79
        action="append", metavar="SYSPREP")
80

    
81
    parser.add_option("--disable-sysprep", dest="disabled_syspreps",
82
        help="Prevent SYSPREP operation from running on the input media",
83
        default=[], action="append", metavar="SYSPREP")
84

    
85
    parser.add_option("--print-sysprep", dest="print_sysprep", default=False,
86
        help="Print the enabled and disabled sysprep operations for this "
87
        "input media", action="store_true")
88

    
89
    parser.add_option("-s", "--silent", dest="silent", default=False,
90
        help="silent mode, only output errors", action="store_true")
91

    
92
    parser.add_option("-u", "--upload", dest="upload", default=False,
93
        help="upload the image to pithos", action="store_true")
94

    
95
    parser.add_option("-r", "--register", dest="register", default=False,
96
        help="register the image to ~okeanos", action="store_true")
97

    
98
    options, args = parser.parse_args(input_args)
99

    
100
    if len(args) != 1:
101
        parser.error('Wrong number of arguments')
102
    options.source = args[0]
103
    if not os.path.exists(options.source):
104
        parser.error('input media is not accessible')
105

    
106
    if options.register:
107
        options.upload = True
108

    
109
    return options
110

    
111

    
112
def image_creator():
113
    options = parse_options(sys.argv[1:])
114

    
115
    if options.silent:
116
        util.silent = True
117

    
118
    if options.outfile is None and not options.upload \
119
                                            and not options.print_sysprep:
120
        raise FatalError("At least one of `-o', `-u' or" \
121
                            "`--print-sysprep' must be set")
122

    
123
    output('snf-image-creator %s\n' % version)
124

    
125
    if os.geteuid() != 0:
126
        raise FatalError("You must run %s as root" \
127
                        % os.path.basename(sys.argv[0]))
128

    
129
    if not options.force and options.outfile is not None:
130
        for extension in ('', '.meta'):
131
            filename = "%s%s" % (options.outfile, extension)
132
            if os.path.exists(filename):
133
                raise FatalError("Output file %s exists "
134
                    "(use --force to overwrite it)." % filename)
135

    
136
    disk = Disk(options.source)
137
    try:
138
        dev = disk.get_device()
139
        dev.mount()
140

    
141
        osclass = get_os_class(dev.distro, dev.ostype)
142
        image_os = osclass(dev.root, dev.g)
143
        metadata = image_os.get_metadata()
144

    
145
        output()
146

    
147
        for sysprep in options.disabled_syspreps:
148
            image_os.disable_sysprep(sysprep)
149

    
150
        for sysprep in options.enabled_syspreps:
151
            image_os.enable_sysprep(sysprep)
152

    
153
        if options.print_sysprep:
154
            image_os.print_syspreps()
155
            output()
156

    
157
        if options.outfile is None and not options.upload:
158
            return 0
159

    
160
        if options.sysprep:
161
            image_os.do_sysprep()
162

    
163
        dev.umount()
164

    
165
        size = options.shrink and dev.shrink() or dev.size()
166
        metadata['SIZE'] = str(size // 2 ** 20)
167

    
168
        if options.outfile is not None:
169
            f = open('%s.%s' % (options.outfile, 'meta'), 'w')
170
            try:
171
                for key in metadata.keys():
172
                    f.write("%s=%s\n" % (key, metadata[key]))
173
            finally:
174
                f.close()
175

    
176
            dev.dump(options.outfile)
177
    finally:
178
        output('cleaning up...')
179
        disk.cleanup()
180

    
181
    return 0
182

    
183

    
184
def main():
185
    try:
186
        ret = image_creator()
187
        sys.exit(ret)
188
    except FatalError as e:
189
        error(e)
190
        sys.exit(1)
191

    
192

    
193
if __name__ == '__main__':
194
    main()
195

    
196
# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :