X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=db-config.d%2F000-functions;h=5779b0e3e313d6f23a00cc0a7d921c67f7f1f5cd;hb=19d4a01ccf66af9e00914351b3eacd5fc880f988;hp=5c1ed2456625e40e5a6d7651ac8b784fe1e09152;hpb=a7ba28837e8c6f949f29aff64a0297ac48cbe065;p=plcapi.git diff --git a/db-config.d/000-functions b/db-config.d/000-functions index 5c1ed24..5779b0e 100644 --- a/db-config.d/000-functions +++ b/db-config.d/000-functions @@ -1,9 +1,6 @@ # -*-python-*- -# $Id$ -# $URL$ #################### import sys, os -import resource g_url = "" def GetMyPLCURL(): return g_url @@ -26,18 +23,42 @@ def SetRole(level, role): g_known_tag_types = [tag_type['tagname'] for tag_type in GetTagTypes()] g_known_tag_types.sort() +def AllPersonRoles (): return [ 'pi','user','tech' ] + def SetTagType(tag_type): - global g_known_tag_types - # Create/update default slice tag types - if tag_type['tagname'] not in g_known_tag_types: - AddTagType(tag_type) - g_known_tag_types.append(tag_type['tagname']) - g_known_tag_types.sort() - else: - UpdateTagType(tag_type['tagname'], tag_type) - if 'roles' in tag_type: - for role in tag_type['roles']: - AddRoleToTagType(role,tag_type['tagname']) + try: + tagname=tag_type['tagname'] + global g_known_tag_types + # handle 'roles' field differently + if 'roles' in tag_type: + roles=tag_type['roles'] + del tag_type['roles'] + else: + roles=['admin'] + # just in case + if 'min_role_id' in tag_type: + print "WARNING: ignoring deprecated field min_role_id for tagtype %s"%tagname + del tag_type['min_role_id'] + # Create/update default slice tag types + if tagname not in g_known_tag_types: + AddTagType(tag_type) + g_known_tag_types.append(tagname) + g_known_tag_types.sort() + else: + UpdateTagType(tagname, tag_type) + # enforce provided roles if present + old_roles=GetTagTypes(tagname)[0]['roles'] + for minus_role in set(old_roles).difference(set(roles)): + DeleteRoleFromTagType(minus_role,tagname) + for plus_role in set(roles).difference(set(old_roles)): + AddRoleToTagType(plus_role,tagname) + except: + # something went wrong for that tagname, + # but don't want to break the whole startup sequence + print "Could not enforce tagtype %s --- beg"%tagname + import traceback + traceback.print_exc() + print "Could not enforce tagtype %s --- end"%tagname # Get list of existing (enabled, global) files g_conf_files = GetConfFiles() @@ -75,56 +96,65 @@ def SetConfFile(conf_file): UpdateConfFile(conf_file_id, conf_file) def SetSlice(slice, tags): - # Create or Update slice - slice_name = slice['name'] - slices = GetSlices([slice_name]) - if len(slices)==1: - slice_id = slices[0]['slice_id'] - if slice.has_key('name'): - del slice['name'] - UpdateSlice(slice_id, slice) - slice['name']=slice_name - else: - expires = None - if slice.has_key('expires'): - expires = slice['expires'] - del slice['expires'] - slice_id = AddSlice(slice) - if expires <> None: - UpdateSlice(slice_id, {'expires':expires}) - - # Get slice structure with all fields - slice = GetSlices([slice_name])[0] - - # Create/delete all tags - # NOTE: update is not needed, since unspecified tags are deleted, - # and new tags are added - slice_tags = [] - if slice['slice_tag_ids']: - # Delete unknown attributes - for slice_tag in GetSliceTags(slice['slice_tag_ids']): - # ignore sliver tags, as those are custom/run-time values - if slice_tag['node_id'] <> None: continue - if (slice_tag['tagname'], slice_tag['value']) not in tags: - DeleteSliceTag(slice_tag['slice_tag_id']) - else: - slice_tags.append((slice_tag['tagname'],slice_tag['value'])) - - # only add slice tags that are new - for (name, value) in tags: - if (name,value) not in slice_tags: - AddSliceTag(slice_name, name, value) + try: + # Create or Update slice + slice_name = slice['name'] + slices = GetSlices([slice_name]) + if len(slices)==1: + slice_id = slices[0]['slice_id'] + if slice.has_key('name'): + del slice['name'] + UpdateSlice(slice_id, slice) + slice['name']=slice_name else: - # NOTE: this confirms that the user-specified tag is - # returned by GetSliceTags - pass + expires = None + if slice.has_key('expires'): + expires = slice['expires'] + del slice['expires'] + slice_id = AddSlice(slice) + if expires <> None: + UpdateSlice(slice_id, {'expires':expires}) + + # Get slice structure with all fields + slice = GetSlices([slice_name])[0] + + # Create/delete all tags + # NOTE: update is not needed, since unspecified tags are deleted, + # and new tags are added + slice_tags = [] + if slice['slice_tag_ids']: + # Delete unknown attributes + for slice_tag in GetSliceTags(slice['slice_tag_ids']): + # ignore sliver tags, as those are custom/run-time values + if slice_tag['node_id'] <> None: continue + if (slice_tag['tagname'], slice_tag['value']) not in tags: + DeleteSliceTag(slice_tag['slice_tag_id']) + else: + slice_tags.append((slice_tag['tagname'],slice_tag['value'])) + + # only add slice tags that are new + for (name, value) in tags: + if (name,value) not in slice_tags: + AddSliceTag(slice_name, name, value) + else: + # NOTE: this confirms that the user-specified tag is + # returned by GetSliceTags + pass + except: + # something went wrong for that tagname, + print "Could not create init slice %s --- beg"%slice['name'] + import traceback + traceback.print_exc() + print "Could not create init slice %s --- end"%slice['name'] def SetMessage(message): messages = GetMessages([message['message_id']]) if len(messages)==0: AddMessage(message) - else: - UpdateMessage(message['message_id'],message) + ### Thierry 2012-03 + # let people customize their messages if they want to + #else: + # UpdateMessage(message['message_id'],message) # Get all model names g_pcu_models = [type['model'] for type in GetPCUTypes()]