geni/methods/remove.py

   1 from geni.util.faults import *
   2 from geni.util.excep import *
   3 from geni.util.method import Method
   4 from geni.util.parameter import Parameter, Mixed
   5 from geni.util.auth import Auth
   6 from geni.util.record import GeniRecord
   7 from geni.util.debug import log
   8
   9 class remove(Method):
  10     """
  11     Remove an object from the registry. If the object represents a PLC object,
  12     then the PLC records will also be removed.
  13
  14     @param cred credential string
  15     @param type record type
  16     @param hrn human readable name of record to remove
  17
  18     @return 1 if successful, faults otherwise
  19     """
  20
  21     interfaces = ['registry']
  22
  23     accepts = [
  24         Parameter(str, "Credential string"),
  25         Parameter(str, "Record type"),
  26         Parameter(str, "Human readable name (hrn) of record to be removed")
  27         ]
  28
  29     returns = Parameter(int, "1 if successful")
  30
  31     def call(self, cred, type, hrn):
  32         self.api.auth.check(cred, "remove")
  33         self.api.auth.verify_object_permission(hrn)
  34         auth_name = self.api.auth.get_authority(hrn)
  35         table = self.api.auth.get_auth_table(auth_name)
  36         record_list = table.resolve(type, hrn)
  37         if not record_list:
  38             raise RecordNotFound(hrn)
  39         record = record_list[0]
  40
  41         # TODO: sa, ma
  42         if type == "user":
  43             self.api.plshell.DeletePerson(self.api.plauth, record.get_pointer())
  44         elif type == "slice":
  45             self.api.plshell.DeleteSlice(self.api.plauth, record.get_pointer())
  46         elif type == "node":
  47             self.api.plshell.DeleteNode(self.api.plauth, record.get_pointer())
  48         elif (type == "sa") or (type == "ma"):
  49             if (type == "sa"):
  50                 other_rec = table.resolve("ma", record.get_name())
  51             elif (type == "ma"):
  52                 other_rec = table.resolve("sa", record.get_name())
  53
  54             if other_rec:
  55                 # sa and ma both map to a site, so if we are deleting one
  56                 # but the other still exists, then do not delete the site
  57                 print >> log, "not removing site", record.get_name(), "because either sa or ma still exists"
  58                 pass
  59             else:
  60                 print >> log, "removing site", record.get_name()
  61                 self.api.plshell.DeleteSite(self.api.plauth, record.get_pointer())
  62         else:
  63             raise UnknownGeniType(type)
  64
  65         table.remove(record)
  66
  67         return 1