sfa/util/namespace.py

   1 ### $Id$
   2 ### $URL$
   3 import re
   4 from sfa.util.faults import *
   5 URN_PREFIX = "urn:publicid:IDN"
   6
   7 def get_leaf(hrn):
   8     parts = hrn.split(".")
   9     return ".".join(parts[-1:])
  10
  11 def get_authority(xrn):
  12     hrn, type = urn_to_hrn(xrn)
  13     if type and type == 'authority':
  14         return hrn
  15
  16     parts = hrn.split(".")
  17     return ".".join(parts[:-1])
  18
  19 def hrn_to_pl_slicename(hrn):
  20     parts = hrn.split(".")
  21     return parts[-2] + "_" + parts[-1]
  22
  23 # assuming hrn is the hrn of an authority, return the plc authority name
  24 def hrn_to_pl_authname(hrn):
  25     parts = hrn.split(".")
  26     return parts[-1]
  27
  28 # assuming hrn is the hrn of an authority, return the plc login_base
  29 def hrn_to_pl_login_base(hrn):
  30     return hrn_to_pl_authname(hrn)
  31
  32 def hostname_to_hrn(auth_hrn, login_base, hostname):
  33     """
  34     Convert hrn to plantelab name.
  35     """
  36     sfa_hostname = ".".join([auth_hrn, login_base, hostname.split(".")[0]])
  37     return sfa_hostname
  38
  39 def slicename_to_hrn(auth_hrn, slicename):
  40     """
  41     Convert hrn to planetlab name.
  42     """
  43     parts = slicename.split("_")
  44     slice_hrn = ".".join([auth_hrn, parts[0]]) + "." + "_".join(parts[1:])
  45
  46     return slice_hrn
  47
  48 def email_to_hrn(auth_hrn, email):
  49     parts = email.split("@")
  50     username = parts[0]
  51     username = username.replace(".", "_").replace("+", "_")
  52     person_hrn = ".".join([auth_hrn, username])
  53
  54     return person_hrn
  55
  56 def urn_to_hrn(urn):
  57     """
  58     convert a urn to hrn
  59     return a tuple (hrn, type)
  60     """
  61
  62     # if this is already a hrn dont do anything
  63     if not urn or not urn.startswith(URN_PREFIX):
  64         return urn, None
  65
  66     name = urn[len(URN_PREFIX):]
  67     hrn_parts = name.split("+")
  68     type = hrn_parts.pop(2)
  69
  70
  71     # Remove the authority name (e.g. '.sa')
  72     if type == 'authority':
  73         hrn_parts = hrn_parts[:-1]
  74
  75     # convert hrn_parts (list) into hrn (str) by doing the following
  76     # dont allow special characters (except ':') in the site login base
  77     # remove blank elements
  78     # replace ':' with '.'
  79     # join list elements using '.'
  80     only_alphanum = re.compile('[^a-zA-Z0-9:]+')
  81     valid_chars = lambda x: only_alphanum.sub('', x).replace(':', '.')
  82     hrn = '.'.join([valid_chars(part) for part in hrn_parts if part])
  83
  84     return str(hrn), str(type)
  85
  86
  87 def hrn_to_urn(hrn, type=None):
  88     """
  89     convert an hrn and type to a urn string
  90     """
  91     # if  this is already a urn dont do anything
  92     if not hrn or hrn.startswith(URN_PREFIX):
  93         return hrn
  94
  95     authority = get_authority(hrn)
  96     name = get_leaf(hrn)
  97
  98     if type == 'authority':
  99         authority = hrn
 100         name = 'sa'
 101
 102     if authority.startswith("plc"):
 103         if type == None:
 104             urn = "+".join(['',authority.replace('.',':'),name])
 105         else:
 106             urn = "+".join(['',authority.replace('.',':'),type,name])
 107
 108     else:
 109         urn = "+".join(['',authority,type,name])
 110
 111     return URN_PREFIX + urn