summaryrefslogtreecommitdiff
path: root/morphlib/morph2.py
blob: ad9e1d89474e4deee2c2b7f370434c50bc1ac05d (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
# Copyright (C) 2012  Codethink Limited
#
# 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; version 2 of the License.
#
# 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, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.


import json


class Morphology(object):

    '''An in-memory representation of a morphology.

    This is a parsed version of the morphology, with rules for default
    values applied. No other processing.

    '''

    static_defaults = [
        ('configure-commands', None),
        ('build-commands', None),
        ('test-commands', None),
        ('install-commands', None),
        ('chunks', {}),
        ('sources', []),
        ('strata', []),
        ('max-jobs', None),
        ('description', ''),
        ('build-depends', None),
        ('build-system', 'manual'),
        ('arch', None),
        ('system-kind', None),
    ]

    def __init__(self, text):
        self._dict = json.loads(text)
        self._set_defaults()

    def __getitem__(self, key):
        return self._dict[key]

    def __contains__(self, key):
        return key in self._dict

    def keys(self):
        return self._dict.keys()

    def _set_defaults(self):
        if 'max-jobs' in self:
            self._dict['max-jobs'] = int(self['max-jobs'])

        if 'disk-size' in self:
            size = self['disk-size']
            size = size.lower()
            if size.endswith('g'):
                size = int(size[:-1]) * 1024 ** 3
            elif size.endswith('m'):  # pragma: no cover
                size = int(size[:-1]) * 1024 ** 2
            elif size.endswith('k'):  # pragma: no cover
                size = int(size[:-1]) * 1024
            else:  # pragma: no cover
                size = int(size)
            self._dict['disk-size'] = size

        for name, value in self.static_defaults:
            if name not in self._dict:
                self._dict[name] = value

        if self['kind'] == 'stratum':
            self._set_stratum_defaults()

    def _set_stratum_defaults(self):
        for source in self['sources']:
            if 'repo' not in source:
                source['repo'] = source['name']
            if 'morph' not in source:
                source['morph'] = source['name']
            if 'build-depends' not in source:
                source['build-depends'] = None