trimmed useless imports, unstarred all imports
[sfa.git] / sfa / methods / RemovePeerObject.py
1 from sfa.util.faults import UnknownSfaType, SfaInvalidArgument
2 from sfa.util.method import Method
3 from sfa.util.parameter import Parameter, Mixed
4 from sfa.util.table import SfaTable
5 from sfa.trust.credential import Credential
6
7 class RemovePeerObject(Method):
8     """
9     Remove an peer object from the PLC records of a local aggregate. 
10     This method will be called by registry.remove() while removing 
11     a record from the local aggreage's PLCDB and sfa table. This 
12     method need not be directly called by end-user.
13     
14     @param cred credential string
15     @param record record as stored in the local registry
16
17     @return 1 if successful, faults otherwise 
18     """
19
20     interfaces = ['registry']
21     
22     accepts = [
23         Parameter(str, "Credential string"),
24         Parameter(dict, "Record dictionary"),
25         Mixed(Parameter(str, "Human readable name of the original caller"),
26               Parameter(None, "Origin hrn not specified"))
27         ]
28
29     returns = Parameter(int, "1 if successful")
30     
31     def call(self, cred, record, origin_hrn=None):
32         user_cred = Credential(string=cred)
33
34         #log the call
35         if not origin_hrn:
36             origin_hrn = user_cred.get_gid_caller().get_hrn()
37         self.api.logger.info("interface: %s\tcaller-hrn: %s\ttarget-hrn: %s\tmethod-name: %s"%(self.api.interface, origin_hrn, record['hrn'], self.name))
38
39         self.api.auth.check(cred, "remove")
40
41         # Only allow the local interface or record owner to delete peer_records 
42         try: self.api.auth.verify_object_permission(record['hrn'])
43         except: self.api.auth.verify_cred_is_me(cred)
44         
45         table = SfaTable()
46         hrn, type = record['hrn'], record['type']
47         records = table.find({'hrn': hrn, 'type': type })
48         for record in records:
49             if record['peer_authority']:
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 UnknownSfaType(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 SfaInvalidArgument, "No such peer"
95         peer = peers[0]
96         return peer['shortname'] 
97
98
99