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