inital checkin

diff --git a/sfa/methods/remove_peer_object.py b/sfa/methods/remove_peer_object.py
new file mode 100644 (file)
index 0000000..67fa56a
--- /dev/null
+++ b/sfa/methods/remove_peer_object.py
@@ -0,0 +1,99 @@
+from sfa.util.faults import *
+from sfa.util.method import Method
+from sfa.util.parameter import Parameter, Mixed
+from sfa.trust.auth import Auth
+from sfa.util.record import GeniRecord
+from sfa.util.genitable import GeniTable
+from sfa.util.debug import log
+from sfa.trust.credential import Credential
+from sfa.util.misc import *
+from types import StringTypes
+
+class remove_peer_object(Method):
+    """
+    Remove an peer object from the PLC records of a local aggregate. 
+    This method will be called by registry.remove() while removing 
+    a record from the local aggreage's PLCDB and sfa table. This 
+    method need not be directly called by end-user.
+    
+    @param cred credential string
+    @param record record as stored in the local registry
+
+    @return 1 if successful, faults otherwise 
+    """
+
+    interfaces = ['registry']
+    
+    accepts = [
+        Parameter(str, "Credential string"),
+        Parameter(dict, "Record dictionary")
+        ]
+
+    returns = Parameter(int, "1 if successful")
+    
+    def call(self, cred, record, caller_cred=None):
+        if caller_cred==None:
+            caller_cred=cred
+        #log the call
+        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(), hrn, self.name))
+        
+        self.api.auth.check(cred, "remove")
+
+        # Only allow the local interface or record owner to delete peer_records 
+        try: self.api.auth.verify_object_permission(record['hrn'])
+        except: self.api.auth.verify_cred_is_me(cred)
+        
+        table = GeniTable()
+        hrn, type, peer_auth = record['hrn'], record['type']
+        records = table.find({'hrn': hrn, 'type': type })
+        for record in records:
+            self.remove_plc_record(record)
+            table.remove(record)
+            
+        return 1
+
+    def remove_plc_record(self, record):
+        type = record['type']        
+        if type == "user":
+            persons = self.api.plshell.GetPersons(self.api.plauth, {'person_id' : record['pointer']})
+            if not persons:
+                return 1
+            person = persons[0]
+            if person['peer_id']:
+                peer = self.get_peer_name(person['peer_id']) 
+                self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'person', person['person_id'], peer)
+            self.api.plshell.DeletePerson(self.api.plauth, person['person_id'])
+           
+        elif type == "slice":
+            slices=self.api.plshell.GetSlices(self.api.plauth, {'slice_id' : record['pointer']})
+               if not slices:
+                return 1
+               slice=slices[0]
+            if slice['peer_id']:
+                peer = self.get_peer_name(slice['peer_id']) 
+                   self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'slice', slice['slice_id'], peer)
+            self.api.plshell.DeleteSlice(self.api.plauth, slice['slice_id'])
+        elif type == "authority":
+            sites=self.api.plshell.GetSites(self.api.plauth, {'site_id' : record['pointer']})
+            if not sites:
+                return 1
+            site=sites[0]
+            if site['peer_id']:
+                peer = self.get_peer_name(site['peer_id']) 
+                self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'site', site['site_id'], peer)
+            self.api.plshell.DeleteSite(self.api.plauth, site['site_id'])
+           
+        else:
+            raise UnknownGeniType(type)
+
+        return 1
+
+    def get_peer_name(self, peer_id):
+        peers = self.api.plshell.GetPeers(self.api.plauth, [peer_id], ['peername', 'shortname', 'hrn_root'])
+        if not peers:
+            raise GeniInvalidArgument, "No such peer"
+        peer = peers[0]
+        return peer['peername'] 
+
+
+