sfa/methods/remove_peer_object.py

   1 from sfa.util.faults import *
   2 from sfa.util.method import Method
   3 from sfa.util.parameter import Parameter, Mixed
   4 from sfa.trust.auth import Auth
   5 from sfa.util.record import GeniRecord
   6 from sfa.util.genitable import GeniTable
   7 from sfa.util.debug import log
   8 from sfa.trust.credential import Credential
   9 from sfa.util.misc import *
  10 from types import StringTypes
  11
  12 class remove_peer_object(Method):
  13     """
  14     Remove an peer object from the PLC records of a local aggregate.
  15     This method will be called by registry.remove() while removing
  16     a record from the local aggreage's PLCDB and sfa table. This
  17     method need not be directly called by end-user.
  18
  19     @param cred credential string
  20     @param record record as stored in the local registry
  21
  22     @return 1 if successful, faults otherwise
  23     """
  24
  25     interfaces = ['registry']
  26
  27     accepts = [
  28         Parameter(str, "Credential string"),
  29         Parameter(dict, "Record dictionary")
  30         ]
  31
  32     returns = Parameter(int, "1 if successful")
  33
  34     def call(self, cred, record, caller_cred=None):
  35         if caller_cred==None:
  36             caller_cred=cred
  37         #log the call
  38         self.api.logger.info("interface: %s\tcaller-hrn: %s\ttarget-hrn: %s\tmethod-name: %s"%(self.api.interface, Credential(string=caller_cred).get_gid_caller().get_hrn(), record['hrn'], self.name))
  39
  40         self.api.auth.check(cred, "remove")
  41
  42         # Only allow the local interface or record owner to delete peer_records
  43         try: self.api.auth.verify_object_permission(record['hrn'])
  44         except: self.api.auth.verify_cred_is_me(cred)
  45
  46         table = GeniTable()
  47         hrn, type = record['hrn'], record['type']
  48         records = table.find({'hrn': hrn, 'type': type })
  49         for record in records:
  50             self.remove_plc_record(record)
  51             table.remove(record)
  52
  53         return 1
  54
  55     def remove_plc_record(self, record):
  56         type = record['type']
  57         if type == "user":
  58             persons = self.api.plshell.GetPersons(self.api.plauth, {'person_id' : record['pointer']})
  59             if not persons:
  60                 return 1
  61             person = persons[0]
  62             if person['peer_id']:
  63                 peer = self.get_peer_name(person['peer_id'])
  64                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'person', person['person_id'], peer)
  65             self.api.plshell.DeletePerson(self.api.plauth, person['person_id'])
  66
  67         elif type == "slice":
  68             slices=self.api.plshell.GetSlices(self.api.plauth, {'slice_id' : record['pointer']})
  69             if not slices:
  70                 return 1
  71             slice=slices[0]
  72             if slice['peer_id']:
  73                 peer = self.get_peer_name(slice['peer_id'])
  74                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'slice', slice['slice_id'], peer)
  75             self.api.plshell.DeleteSlice(self.api.plauth, slice['slice_id'])
  76         elif type == "authority":
  77             sites=self.api.plshell.GetSites(self.api.plauth, {'site_id' : record['pointer']})
  78             if not sites:
  79                 return 1
  80             site=sites[0]
  81             if site['peer_id']:
  82                 peer = self.get_peer_name(site['peer_id'])
  83                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'site', site['site_id'], peer)
  84             self.api.plshell.DeleteSite(self.api.plauth, site['site_id'])
  85
  86         else:
  87             raise UnknownGeniType(type)
  88
  89         return 1
  90
  91     def get_peer_name(self, peer_id):
  92         peers = self.api.plshell.GetPeers(self.api.plauth, [peer_id], ['peername', 'shortname', 'hrn_root'])
  93         if not peers:
  94             raise GeniInvalidArgument, "No such peer"
  95         peer = peers[0]
  96         return peer['peername']
  97
  98
  99