add request_hash argument to more method calls
[sfa.git] / 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         Parameter(str, "Request hash"),
31         ]
32
33     returns = Parameter(int, "1 if successful")
34     
35     def call(self, cred, record, request_hash, caller_cred=None):
36         if caller_cred==None:
37             caller_cred=cred
38         #log the call
39         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))
40         self.api.auth.authenticateCred(cred, [cred], request_hash) 
41         self.api.auth.check(cred, "remove")
42
43         # Only allow the local interface or record owner to delete peer_records 
44         try: self.api.auth.verify_object_permission(record['hrn'])
45         except: self.api.auth.verify_cred_is_me(cred)
46         
47         table = GeniTable()
48         hrn, type = record['hrn'], record['type']
49         records = table.find({'hrn': hrn, 'type': type })
50         for record in records:
51           if record['peer_authority']:
52             self.remove_plc_record(record)
53             table.remove(record)
54             
55         return 1
56
57     def remove_plc_record(self, record):
58         type = record['type']        
59         if type == "user":
60             persons = self.api.plshell.GetPersons(self.api.plauth, {'person_id' : record['pointer']})
61             if not persons:
62                 return 1
63             person = persons[0]
64             if person['peer_id']:
65                 peer = self.get_peer_name(person['peer_id']) 
66                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'person', person['person_id'], peer)
67             self.api.plshell.DeletePerson(self.api.plauth, person['person_id'])
68            
69         elif type == "slice":
70             slices=self.api.plshell.GetSlices(self.api.plauth, {'slice_id' : record['pointer']})
71             if not slices:
72                 return 1
73             slice=slices[0]
74             if slice['peer_id']:
75                 peer = self.get_peer_name(slice['peer_id']) 
76                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'slice', slice['slice_id'], peer)
77             self.api.plshell.DeleteSlice(self.api.plauth, slice['slice_id'])
78         elif type == "authority":
79             sites=self.api.plshell.GetSites(self.api.plauth, {'site_id' : record['pointer']})
80             if not sites:
81                 return 1
82             site=sites[0]
83             if site['peer_id']:
84                 peer = self.get_peer_name(site['peer_id']) 
85                 self.api.plshell.UnBindObjectFromPeer(self.api.plauth, 'site', site['site_id'], peer)
86             self.api.plshell.DeleteSite(self.api.plauth, site['site_id'])
87            
88         else:
89             raise UnknownGeniType(type)
90
91         return 1
92
93     def get_peer_name(self, peer_id):
94         peers = self.api.plshell.GetPeers(self.api.plauth, [peer_id], ['peername', 'shortname', 'hrn_root'])
95         if not peers:
96             raise GeniInvalidArgument, "No such peer"
97         peer = peers[0]
98         return peer['shortname'] 
99
100
101