From: Tony Mack <tmack@cs.princeton.edu>
Date: Thu, 3 Sep 2009 17:01:17 +0000 (+0000)
Subject: add sfa upcalls to AddPersontToSite and DeletPersonFromSite
X-Git-Tag: PLCAPI-4.3-23~2
X-Git-Url: http://git.onelab.eu/?p=plcapi.git;a=commitdiff_plain;h=27cd56b5b4a797a1f730753901b972235224f16d

add sfa upcalls to AddPersontToSite and DeletPersonFromSite
---

diff --git a/PLC/Methods/AddPersonToSite.py b/PLC/Methods/AddPersonToSite.py
index a6699cf..fce91f1 100644
--- a/PLC/Methods/AddPersonToSite.py
+++ b/PLC/Methods/AddPersonToSite.py
@@ -57,7 +57,9 @@ class AddPersonToSite(Method):
         self.message = 'Person %d added to site %d' % \
                        (person['person_id'], site['site_id'])
 
+        
         sfa = SFA(self.api)
+        person['site_ids'] = [site['site_id']]
         sfa.update_record(person, 'user')
 
         return 1
diff --git a/PLC/Methods/DeletePersonFromSite.py b/PLC/Methods/DeletePersonFromSite.py
index 0345100..7400dc3 100644
--- a/PLC/Methods/DeletePersonFromSite.py
+++ b/PLC/Methods/DeletePersonFromSite.py
@@ -6,6 +6,7 @@ from PLC.Parameter import Parameter, Mixed
 from PLC.Persons import Person, Persons
 from PLC.Sites import Site, Sites
 from PLC.Auth import Auth
+from PLC.SFA import SFA
 
 class DeletePersonFromSite(Method):
     """
@@ -56,4 +57,7 @@ class DeletePersonFromSite(Method):
         self.message = 'Person %d deleted from site %d	' % \
                         (person['person_id'], site['site_id'])
 
+        sfa = SFA(self.api)
+        sfa.delete_record(person, 'user')
+
         return 1