- added logging variable 'object_type'

[plcapi.git] / PLC / Methods / DeletePersonFromSite.py

diff --git a/PLC/Methods/DeletePersonFromSite.py b/PLC/Methods/DeletePersonFromSite.py
index c17384d..0b7280c 100644 (file)
--- a/PLC/Methods/DeletePersonFromSite.py
+++ b/PLC/Methods/DeletePersonFromSite.py
@@ -25,6 +25,8 @@ class DeletePersonFromSite(Method):
         ]
 
     returns = Parameter(int, '1 if successful')
+
+    object_type = 'Site'
     
 
     def call(self, auth, person_id_or_email, site_id_or_login_base):
@@ -32,19 +34,25 @@ class DeletePersonFromSite(Method):
         persons = Persons(self.api, [person_id_or_email])
         if not persons:
             raise PLCInvalidArgument, "No such account"
-
         person = persons[0]
 
+        if person['peer_id'] is not None:
+            raise PLCInvalidArgument, "Not a local account"
+
         # Get site information
         sites = Sites(self.api, [site_id_or_login_base])
         if not sites:
             raise PLCInvalidArgument, "No such site"
-
         site = sites[0]
 
+        if site['peer_id'] is not None:
+            raise PLCInvalidArgument, "Not a local site"
+
         if site['site_id'] in person['site_ids']:
             site.remove_person(person)
 
+       # Logging variables
        self.object_ids = [site['site_id']]
-       
+       self.message = 'Person %d deleted from site %d  ' % \
+               (person['person_id'], site['site_id'])
         return 1