Hot-keys on this page

r m x p   toggle line displays

j k   next/prev highlighted chunk

0   (zero) top of page

1   (one) first highlighted chunk

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

# -*- Mode:Python; indent-tabs-mode:nil; tab-width:4 -*- 

# 

# Copyright (C) 2015 Canonical Ltd 

# 

# This program is free software: you can redistribute it and/or modify 

# it under the terms of the GNU General Public License version 3 as 

# published by the Free Software Foundation. 

# 

# 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/>. 

 

import logging 

import requests 

import yaml 

 

PARTS_URI = 'https://wiki.ubuntu.com/Snappy/Parts' 

PARTS_URI_PARAMS = {'action': 'raw'} 

 

_WIKI_OPEN = '{{{' 

_WIKI_CLOSE = '}}}' 

 

logging.getLogger("urllib3").setLevel(logging.CRITICAL) 

 

 

class Wiki: 

    wiki_parts = None 

 

    def _fetch(self): 

        if self.wiki_parts is None: 

            raw_content = requests.get(PARTS_URI, params=PARTS_URI_PARAMS) 

            content = raw_content.text.strip() 

 

40            if content.startswith(_WIKI_OPEN): 

                content = content[len(_WIKI_OPEN):].strip() 

43            if content.endswith(_WIKI_CLOSE): 

                content = content[:-len(_WIKI_CLOSE)] 

 

            self.wiki_parts = yaml.load(content) 

 

    def get_part(self, name): 

        self._fetch() 

 

        if name in self.wiki_parts: 

50            if 'plugin' and 'type' in self.wiki_parts[name]: 

                del self.wiki_parts[name]['type'] 

            return self.wiki_parts[name] 

 

    def compose(self, name, properties): 

        """Return properties composed with the ones from part name in the wiki. 

 

        :param str name: The name of the part to query from the wiki 

        :param dict properties: The current set of properties 

        :return: Part properties from the wiki composed with the properties 

                 passed as a parameter. If there is no wiki part named name, 

                 properties will be returned. 

        :rtype: dict 

        :raises KeyError: if the part named name is not found in the wiki. 

        """ 

        self._fetch() 

 

        wiki_properties = self.wiki_parts[name] 

        for key in wiki_properties: 

            properties[key] = properties.get(key, wiki_properties[key]) 

        properties['plugin'] = wiki_properties.get('plugin', None) 

 

        return properties