sfa/util/xrn.py

   1 #----------------------------------------------------------------------
   2 # Copyright (c) 2008 Board of Trustees, Princeton University
   3 #
   4 # Permission is hereby granted, free of charge, to any person obtaining
   5 # a copy of this software and/or hardware specification (the "Work") to
   6 # deal in the Work without restriction, including without limitation the
   7 # rights to use, copy, modify, merge, publish, distribute, sublicense,
   8 # and/or sell copies of the Work, and to permit persons to whom the Work
   9 # is furnished to do so, subject to the following conditions:
  10 #
  11 # The above copyright notice and this permission notice shall be
  12 # included in all copies or substantial portions of the Work.
  13 #
  14 # THE WORK IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  15 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
  16 # MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
  17 # NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT
  18 # HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
  19 # WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 # OUT OF OR IN CONNECTION WITH THE WORK OR THE USE OR OTHER DEALINGS
  21 # IN THE WORK.
  22 #----------------------------------------------------------------------
  23
  24 import re
  25
  26 from sfa.util.faults import SfaAPIError
  27
  28 # for convenience and smoother translation - we should get rid of these functions eventually
  29 def get_leaf(hrn): return Xrn(hrn).get_leaf()
  30 def get_authority(hrn): return Xrn(hrn).get_authority_hrn()
  31 def urn_to_hrn(urn): xrn=Xrn(urn); return (xrn.hrn, xrn.type)
  32 def hrn_to_urn(hrn,type): return Xrn(hrn, type=type).urn
  33 def hrn_authfor_hrn(parenthrn, hrn): return Xrn.hrn_is_auth_for_hrn(parenthrn, hrn)
  34
  35 class Xrn:
  36
  37     ########## basic tools on HRNs
  38     # split a HRN-like string into pieces
  39     # this is like split('.') except for escaped (backslashed) dots
  40     # e.g. hrn_split ('a\.b.c.d') -> [ 'a\.b','c','d']
  41     @staticmethod
  42     def hrn_split(hrn):
  43         return [ x.replace('--sep--','\\.') for x in hrn.replace('\\.','--sep--').split('.') ]
  44
  45     # e.g. hrn_leaf ('a\.b.c.d') -> 'd'
  46     @staticmethod
  47     def hrn_leaf(hrn): return Xrn.hrn_split(hrn)[-1]
  48
  49     # e.g. hrn_auth_list ('a\.b.c.d') -> ['a\.b', 'c']
  50     @staticmethod
  51     def hrn_auth_list(hrn): return Xrn.hrn_split(hrn)[0:-1]
  52
  53     # e.g. hrn_auth ('a\.b.c.d') -> 'a\.b.c'
  54     @staticmethod
  55     def hrn_auth(hrn): return '.'.join(Xrn.hrn_auth_list(hrn))
  56
  57     # e.g. escape ('a.b') -> 'a\.b'
  58     @staticmethod
  59     def escape(token): return re.sub(r'([^\\])\.', r'\1\.', token)
  60
  61     # e.g. unescape ('a\.b') -> 'a.b'
  62     @staticmethod
  63     def unescape(token): return token.replace('\\.','.')
  64
  65     # Return the HRN authority chain from top to bottom.
  66     # e.g. hrn_auth_chain('a\.b.c.d') -> ['a\.b', 'a\.b.c']
  67     @staticmethod
  68     def hrn_auth_chain(hrn):
  69         parts = Xrn.hrn_auth_list(hrn)
  70         chain = []
  71         for i in range(len(parts)):
  72             chain.append('.'.join(parts[:i+1]))
  73         # Include the HRN itself?
  74         #chain.append(hrn)
  75         return chain
  76
  77     # Is the given HRN a true authority over the namespace of the other
  78     # child HRN?
  79     # A better alternative than childHRN.startswith(parentHRN)
  80     # e.g. hrn_is_auth_for_hrn('a\.b', 'a\.b.c.d') -> True,
  81     # but hrn_is_auth_for_hrn('a', 'a\.b.c.d') -> False
  82     # Also hrn_is_auth_for_hrn('a\.b.c.d', 'a\.b.c.d') -> True
  83     @staticmethod
  84     def hrn_is_auth_for_hrn(parenthrn, hrn):
  85         if parenthrn == hrn:
  86             return True
  87         for auth in Xrn.hrn_auth_chain(hrn):
  88             if parenthrn == auth:
  89                 return True
  90         return False
  91
  92     ########## basic tools on URNs
  93     URN_PREFIX = "urn:publicid:IDN"
  94     URN_PREFIX_lower = "urn:publicid:idn"
  95
  96     @staticmethod
  97     def is_urn (text):
  98         return text.lower().startswith(Xrn.URN_PREFIX_lower)
  99
 100     @staticmethod
 101     def urn_full (urn):
 102         if Xrn.is_urn(urn): return urn
 103         else: return Xrn.URN_PREFIX+urn
 104     @staticmethod
 105     def urn_meaningful (urn):
 106         if Xrn.is_urn(urn): return urn[len(Xrn.URN_PREFIX):]
 107         else: return urn
 108     @staticmethod
 109     def urn_split (urn):
 110         return Xrn.urn_meaningful(urn).split('+')
 111
 112     @staticmethod
 113     def filter_type(urns=[], type=None):
 114         urn_list = []
 115         if not type:
 116             return urns
 117
 118         for urn in urns:
 119             xrn = Xrn(xrn=urn)
 120             if not xrn.type:
 121                 # If an xrn doesn't have a type its probably a hrn.
 122                 # We have to make some assumptions on the hrn's type
 123                 # based on the contents of the hrn.
 124                 if type == 'sliver' and '-' in xrn.leaf:
 125                     urn_list.append(urn)
 126                 elif type != 'sliver' and not '-' in xrn.leaf:
 127                     urn_list.append(urn)
 128             elif (xrn.type == type):
 129                 # Xrn is probably a urn so we can just compare types
 130                 urn_list.append(urn)
 131         return urn_list
 132     ####################
 133     # the local fields that are kept consistent
 134     # self.urn
 135     # self.hrn
 136     # self.type
 137     # self.path
 138     # provide either urn, or (hrn + type)
 139     def __init__ (self, xrn="", type=None, id=None):
 140         if not xrn: xrn = ""
 141         # user has specified xrn : guess if urn or hrn
 142         self.id = id
 143         if Xrn.is_urn(xrn):
 144             self.hrn=None
 145             self.urn=xrn
 146             if id:
 147                 self.urn = "%s:%s" % (self.urn, str(id))
 148             self.urn_to_hrn()
 149         else:
 150             self.urn=None
 151             self.hrn=xrn
 152             self.type=type
 153             self.hrn_to_urn()
 154         self._normalize()
 155 # happens all the time ..
 156 #        if not type:
 157 #            debug_logger.debug("type-less Xrn's are not safe")
 158
 159     def __repr__ (self):
 160         result="<XRN u=%s h=%s"%(self.urn,self.hrn)
 161         if hasattr(self,'leaf'): result += " leaf=%s"%self.leaf
 162         if hasattr(self,'authority'): result += " auth=%s"%self.authority
 163         result += ">"
 164         return result
 165
 166     def get_urn(self): return self.urn
 167     def get_hrn(self): return self.hrn
 168     def get_type(self): return self.type
 169     def get_hrn_type(self): return (self.hrn, self.type)
 170
 171     def _normalize(self):
 172         if self.hrn is None: raise SfaAPIError, "Xrn._normalize"
 173         if not hasattr(self,'leaf'):
 174             self.leaf=Xrn.hrn_split(self.hrn)[-1]
 175         # self.authority keeps a list
 176         if not hasattr(self,'authority'):
 177             self.authority=Xrn.hrn_auth_list(self.hrn)
 178
 179     def get_leaf(self):
 180         self._normalize()
 181         return self.leaf
 182
 183     def get_authority_hrn(self):
 184         self._normalize()
 185         return '.'.join( self.authority )
 186
 187     def get_authority_urn(self):
 188         self._normalize()
 189         return ':'.join( [Xrn.unescape(x) for x in self.authority] )
 190
 191     def set_authority(self, authority):
 192         """
 193         update the authority section of an existing urn
 194         """
 195         authority_hrn = self.get_authority_hrn()
 196         if not authority_hrn.startswith(authority):
 197             hrn = ".".join([authority,authority_hrn, self.get_leaf()])
 198         else:
 199             hrn = ".".join([authority_hrn, self.get_leaf()])
 200
 201         self.hrn = hrn
 202         self.hrn_to_urn()
 203         self._normalize()
 204
 205     # sliver_id_parts is list that contains the sliver's
 206     # slice id and node id
 207     def get_sliver_id_parts(self):
 208         sliver_id_parts = []
 209         if self.type == 'sliver' or '-' in self.leaf:
 210             sliver_id_parts = self.leaf.split('-')
 211         return sliver_id_parts
 212
 213
 214     def urn_to_hrn(self):
 215         """
 216         compute tuple (hrn, type) from urn
 217         """
 218
 219 #        if not self.urn or not self.urn.startswith(Xrn.URN_PREFIX):
 220         if not Xrn.is_urn(self.urn):
 221             raise SfaAPIError, "Xrn.urn_to_hrn"
 222
 223         parts = Xrn.urn_split(self.urn)
 224         type=parts.pop(2)
 225         # Remove the authority name (e.g. '.sa')
 226         if type == 'authority':
 227             name = parts.pop()
 228             # Drop the sa. This is a bad hack, but its either this
 229             # or completely change how record types are generated/stored
 230             if name != 'sa':
 231                 type = type + "+" + name
 232             name =""
 233         else:
 234             name = parts.pop(len(parts)-1)
 235         # convert parts (list) into hrn (str) by doing the following
 236         # 1. remove blank parts
 237         # 2. escape dots inside parts
 238         # 3. replace ':' with '.' inside parts
 239         # 3. join parts using '.'
 240         hrn = '.'.join([Xrn.escape(part).replace(':','.') for part in parts if part])
 241         # dont replace ':' in the name section
 242         if name:
 243             parts = name.split(':')
 244             if len(parts) > 1:
 245                 self.id = ":".join(parts[1:])
 246                 name = parts[0]
 247             hrn += '.%s' % Xrn.escape(name)
 248
 249         self.hrn=str(hrn)
 250         self.type=str(type)
 251
 252     def hrn_to_urn(self):
 253         """
 254         compute urn from (hrn, type)
 255         """
 256
 257 #        if not self.hrn or self.hrn.startswith(Xrn.URN_PREFIX):
 258         if Xrn.is_urn(self.hrn):
 259             raise SfaAPIError, "Xrn.hrn_to_urn, hrn=%s"%self.hrn
 260
 261         if self.type and self.type.startswith('authority'):
 262             self.authority = Xrn.hrn_auth_list(self.hrn)
 263             leaf = self.get_leaf()
 264             #if not self.authority:
 265             #    self.authority = [self.hrn]
 266             type_parts = self.type.split("+")
 267             self.type = type_parts[0]
 268             name = 'sa'
 269             if len(type_parts) > 1:
 270                 name = type_parts[1]
 271             auth_parts = [part for part in [self.get_authority_urn(), leaf] if part]
 272             authority_string = ":".join(auth_parts)
 273         else:
 274             self.authority = Xrn.hrn_auth_list(self.hrn)
 275             name = Xrn.hrn_leaf(self.hrn)
 276             authority_string = self.get_authority_urn()
 277
 278         if self.type == None:
 279             urn = "+".join(['',authority_string,Xrn.unescape(name)])
 280         else:
 281             urn = "+".join(['',authority_string,self.type,Xrn.unescape(name)])
 282
 283         if hasattr(self, 'id') and self.id:
 284             urn = "%s:%s" % (urn, self.id)
 285
 286         self.urn = Xrn.URN_PREFIX + urn
 287
 288     def dump_string(self):
 289         result="-------------------- XRN\n"
 290         result += "URN=%s\n"%self.urn
 291         result += "HRN=%s\n"%self.hrn
 292         result += "TYPE=%s\n"%self.type
 293         result += "LEAF=%s\n"%self.get_leaf()
 294         result += "AUTH(hrn format)=%s\n"%self.get_authority_hrn()
 295         result += "AUTH(urn format)=%s\n"%self.get_authority_urn()
 296         return result
 297