fix bugs
[sfa.git] / sfa / methods / remove.py
1 ### $Id$
2 ### $URL$
3
4 from sfa.util.faults import *
5 from sfa.util.method import Method
6 from sfa.util.parameter import Parameter, Mixed
7 from sfa.trust.auth import Auth
8 from sfa.util.record import GeniRecord
9 from sfa.util.genitable import GeniTable
10 from sfa.util.debug import log
11 from sfa.trust.credential import Credential
12 from sfa.server.registry import Registries
13
14 class remove(Method):
15     """
16     Remove an object from the registry. If the object represents a PLC object,
17     then the PLC records will also be removed.
18     
19     @param cred credential string
20     @param type record type
21     @param hrn human readable name of record to remove
22
23     @return 1 if successful, faults otherwise 
24     """
25
26     interfaces = ['registry']
27     
28     accepts = [
29         Parameter(str, "Credential string"),
30         Parameter(str, "Record type"),
31         Parameter(str, "Human readable name (hrn) of record to be removed")
32         ]
33
34     returns = Parameter(int, "1 if successful")
35     
36     def call(self, cred, type, hrn, caller_cred=None):
37         if caller_cred==None:
38             caller_cred=cred
39         #log the call
40         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))
41
42         self.api.auth.check(cred, "remove")
43         self.api.auth.verify_object_permission(hrn)
44         table = GeniTable()
45         filter = {'hrn': hrn}
46         if type not in ['all', '*']:
47             filter['type'] = type
48         records = table.find(filter)
49         if not records:
50             raise RecordNotFound(hrn)
51         record = records[0]
52         type = record['type']
53
54         credential = self.api.getCredential()
55         registries = Registries(self.api) 
56
57         # Try to remove the object from the PLCDB of federated agg.
58         # This is attempted before removing the object from the local agg's PLCDB and sfa table
59         if hrn.startswith(self.api.hrn) and type in ['user', 'slice', 'authority']:
60             for registry in registries:
61                 if registry not in [self.api.hrn]:
62                     try:
63                         result=registries[registry].remove_peer_object(credential, record)
64                     except:
65                         pass
66         if type == "user":
67             persons = self.api.plshell.GetPersons(self.api.plauth, record['pointer'])
68             # only delete this person if he has site ids. if he doesnt, it probably means 
69             # he was just removed from a site, not actually deleted
70             if persons and persons[0]['site_ids']:
71                 self.api.plshell.DeletePerson(self.api.plauth, record['pointer'])
72         elif type == "slice":
73             if self.api.plshell.GetSlices(self.api.plauth, record['pointer']):
74                 self.api.plshell.DeleteSlice(self.api.plauth, record['pointer'])
75         elif type == "node":
76             if self.api.plshell.GetNodes(self.api.plauth, record['pointer']):
77                 self.api.plshell.DeleteNode(self.api.plauth, record['pointer'])
78         elif type == "authority":
79             if self.api.plshell.GetSites(self.api.plauth, record['pointer']):
80                 self.api.plshell.DeleteSite(self.api.plauth, record['pointer'])
81         else:
82             raise UnknownGeniType(type)
83
84         table.remove(record)
85            
86         # forward the call after replacing the root hrn
87
88         return 1