Statistics
| Branch: | Tag: | Revision:

root / image_creator / dialog_util.py @ e36b5b14

History | View | Annotate | Download (5.4 kB)

1
#!/usr/bin/env python
2

    
3
# Copyright 2012 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
import os
37
from image_creator.output.dialog import GaugeOutput
38
from image_creator.util import MD5
39

    
40
SMALL_WIDTH = 60
41
WIDTH = 70
42

    
43

    
44
def update_background_title(session):
45
    d = session['dialog']
46
    dev = session['device']
47
    disk = session['disk']
48

    
49
    MB = 2 ** 20
50

    
51
    size = (dev.size + MB - 1) // MB
52
    shrinked = 'shrinked' in session and session['shrinked']
53
    postfix = " (shrinked)" if shrinked else ''
54

    
55
    title = "OS: %s, Distro: %s, Size: %dMB%s, Source: %s" % \
56
            (dev.ostype, dev.distro, size, postfix,
57
             os.path.abspath(disk.source))
58

    
59
    d.setBackgroundTitle(title)
60

    
61

    
62
def confirm_exit(d, msg=''):
63
    return not d.yesno("%s Do you want to exit?" % msg, width=SMALL_WIDTH)
64

    
65

    
66
def confirm_reset(d):
67
    return not d.yesno("Are you sure you want to reset everything?",
68
                       width=SMALL_WIDTH, defaultno=1)
69

    
70

    
71
class Reset(Exception):
72
    pass
73

    
74

    
75
def extract_metadata_string(session):
76
    metadata = ['%s=%s' % (k, v) for (k, v) in session['metadata'].items()]
77

    
78
    if 'task_metadata' in session:
79
        metadata.extend("%s=yes" % m for m in session['task_metadata'])
80

    
81
    return '\n'.join(metadata) + '\n'
82

    
83

    
84
def extract_image(session):
85
    d = session['dialog']
86
    dir = os.getcwd()
87
    while 1:
88
        if dir and dir[-1] != os.sep:
89
            dir = dir + os.sep
90

    
91
        (code, path) = d.fselect(dir, 10, 50, title="Save image as...")
92
        if code in (d.DIALOG_CANCEL, d.DIALOG_ESC):
93
            return False
94

    
95
        if os.path.isdir(path):
96
            dir = path
97
            continue
98

    
99
        if os.path.isdir("%s.meta" % path):
100
            d.msgbox("Can't overwrite directory `%s.meta'" % path,
101
                     width=SMALL_WIDTH)
102
            continue
103

    
104
        if os.path.isdir("%s.md5sum" % path):
105
            d.msgbox("Can't overwrite directory `%s.md5sum'" % path,
106
                     width=SMALL_WIDTH)
107
            continue
108

    
109
        basedir = os.path.dirname(path)
110
        name = os.path.basename(path)
111
        if not os.path.exists(basedir):
112
            d.msgbox("Directory `%s' does not exist" % basedir,
113
                     width=SMALL_WIDTH)
114
            continue
115

    
116
        dir = basedir
117
        if len(name) == 0:
118
            continue
119

    
120
        files = ["%s%s" % (path, ext) for ext in ('', '.meta', '.md5sum')]
121
        overwrite = filter(os.path.exists, files)
122

    
123
        if len(overwrite) > 0:
124
            if d.yesno("The following file(s) exist:\n"
125
                       "%s\nDo you want to overwrite them?" %
126
                       "\n".join(overwrite), width=SMALL_WIDTH):
127
                continue
128

    
129
        gauge = GaugeOutput(d, "Image Extraction", "Extracting image...")
130
        try:
131
            dev = session['device']
132
            out = dev.out
133
            out.add(gauge)
134
            try:
135
                if "checksum" not in session:
136
                    size = dev.size
137
                    md5 = MD5(out)
138
                    session['checksum'] = md5.compute(session['snapshot'],
139
                                                      size)
140

    
141
                # Extract image file
142
                dev.dump(path)
143

    
144
                # Extract metadata file
145
                out.output("Extracting metadata file...")
146
                with open('%s.meta' % path, 'w') as f:
147
                    f.write(extract_metadata_string(session))
148
                out.success('done')
149

    
150
                # Extract md5sum file
151
                out.output("Extracting md5sum file...")
152
                md5str = "%s %s\n" % (session['checksum'], name)
153
                with open('%s.md5sum' % path, 'w') as f:
154
                    f.write(md5str)
155
                out.success("done")
156
            finally:
157
                out.remove(gauge)
158
        finally:
159
            gauge.cleanup()
160
        d.msgbox("Image file `%s' was successfully extracted!" % path,
161
                 width=SMALL_WIDTH)
162
        break
163

    
164
    return True
165

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