cmdatahandlers/tests/mocked_dependencies/serviceprofiles/profiles.py

   1 # Copyright 2019 Nokia
   2
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #     http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12 # See the License for the specific language governing permissions and
  13 # limitations under the License.
  14
  15 import os
  16 import re
  17 import inspect
  18
  19 class Profile(object):
  20     def __init__(self):
  21         self.name = None
  22         self.description = None
  23         self.inherits = []
  24         self.included_profiles = []
  25     def __str__(self):
  26         return 'name:{}\ndescription:{}\ninherits:{}\nincluded_profiles:{}\n'.format(self.name, self.description, self.inherits, self.included_profiles)
  27
  28 class Profiles(object):
  29     def __init__(self, location=os.path.dirname(os.path.abspath(inspect.getfile(inspect.currentframe())))):
  30         self.location = location
  31         self.profiles = {}
  32         try:
  33             self._load_profiles()
  34         except:
  35             pass
  36
  37     def _load_profiles(self):
  38         files = self._get_profiles_files()
  39         for f in files:
  40             self._profile_from_file(f)
  41
  42         #update the included profiles
  43         for name, profile in self.profiles.iteritems():
  44             included_profiles = []
  45             self._update_included_profiles(profile, included_profiles)
  46             profile.included_profiles = included_profiles
  47
  48
  49     def _update_included_profiles(self, profile, included_profiles):
  50         included_profiles.append(profile.name)
  51         for b in profile.inherits:
  52             self._update_included_profiles(self.profiles[b], included_profiles)
  53
  54     def _get_profiles_files(self):
  55         files = os.listdir(self.location)
  56         pattern = re.compile('.*[.]profile$')
  57         result = []
  58         for f in files:
  59             fullpath = self.location + '/' + f
  60             if os.path.isfile(fullpath) and pattern.match(f):
  61                 result.append(fullpath)
  62         return result
  63
  64
  65
  66     def _profile_from_file(self, filename):
  67         profile = Profile()
  68         with open(filename) as f:
  69             lines=f.read().splitlines()
  70             for l in lines:
  71                 data = l.split(':')
  72                 if len(data) != 2:
  73                     raise Exception('Invalid line %s in file %s' % (l, filename))
  74                 elif data[0] == 'name':
  75                     profile.name = data[1]
  76                 elif data[0] == 'description':
  77                     profile.description = data[1]
  78                 elif data[0] == 'inherits':
  79                     profile.inherits = data[1].split(',')
  80                 else:
  81                     raise Exception('Invalid line %s in file %s' % (l, filename))
  82         self.profiles[profile.name] = profile
  83
  84     def get_included_profiles(self, name):
  85         return self.profiles[name].included_profiles
  86
  87     def get_profiles(self):
  88         return self.profiles
  89
  90     def get_children_profiles(self, name):
  91         ret = []
  92         for pfname, profile in self.profiles.iteritems():
  93             if name in profile.inherits:
  94                 ret.append(pfname)
  95         return ret
  96
  97
  98 if __name__ == '__main__':
  99     import sys
 100     import traceback
 101     import argparse
 102
 103     parser = argparse.ArgumentParser(description='Test service profiles',
 104             prog=sys.argv[0])
 105
 106
 107     parser.add_argument('--get-included-profiles',
 108             dest='get_included_profiles',
 109             help='Get the profiles included in some profile name',
 110             action='store_true')
 111
 112     parser.add_argument('--get-all-profiles',
 113             help='Get the profiles list',
 114             dest='get_all_profiles',
 115             action='store_true')
 116
 117     parser.add_argument('--get-children-profiles',
 118             dest='get_children_profiles',
 119             help='Get the children of a profile',
 120             action='store_true')
 121
 122     parser.add_argument('--name',
 123             metavar='NAME',
 124             dest='name',
 125             help='The name of the profile',
 126             type=str,
 127             action='store')
 128     try:
 129         args = parser.parse_args(sys.argv[1:])
 130         profiles = Profiles()
 131         if args.get_included_profiles or args.get_children_profiles:
 132             if not args.name:
 133                 raise Exception('Missing profile name')
 134
 135             if args.get_included_profiles:
 136                 included_profiles = profiles.get_included_profiles(args.name)
 137                 print('Included profiles')
 138                 for p in included_profiles:
 139                     print(p)
 140             if args.get_children_profiles:
 141                 children_profiles = profiles.get_children_profiles(args.name)
 142                 print('Children profiles')
 143                 for p in children_profiles:
 144                     print(p)
 145         elif args.get_all_profiles:
 146             all = profiles.get_profiles()
 147             for name, p in all.iteritems():
 148                 print(p)
 149     except Exception as exp:
 150         print('Failed with error %s' % exp)
 151         traceback.print_exc()
 152         sys.exit(1)