X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=sface%2Fscreens%2Fuserscreen.py;h=92089d14d2a5efa5776ecc956d10f2560806f4bd;hb=deb56ec8ef1adb3c2a5b4180623452bb9dcc1b48;hp=9eb2e6c3c9756a0e2832f5098cf3d054065e6322;hpb=908ac46c4760618de030e0cad35479acc06dd22a;p=sface.git

diff --git a/sface/screens/userscreen.py b/sface/screens/userscreen.py
index 9eb2e6c..92089d1 100644
--- a/sface/screens/userscreen.py
+++ b/sface/screens/userscreen.py
@@ -59,6 +59,9 @@ class UserView(QTableView):
         node_data = node_index.data().toString()
         self.emit(SIGNAL('hostnameClicked(QString)'), node_data)
 
+    def hideUnusableColumns(self):
+        self.hideColumn(SERVER_MEMBERSHIP_STATUS_COLUMN)
+
 class UserModel(QStandardItemModel):
     def __init__(self, rows=0, columns=4, parent=None):
          QStandardItemModel.__init__(self, rows, columns, parent)
@@ -82,7 +85,7 @@ class UserModel(QStandardItemModel):
                      slice_persons.append({"name": name, "role": "researcher", "member": user_status["in"]})
                      added_persons.append(name)
 
-        i=1
+        i=0
         while (os.path.exists(config.getAuthorityListFile(i))):
             rec = self.readUserRecord(i)
             if rec:
@@ -126,6 +129,19 @@ class UserModel(QStandardItemModel):
 
         return change
 
+    def getResearchers(self):
+        researchers = []
+        item = self.invisibleRootItem()
+        children = item.rowCount()
+        for row in range(0, children):
+            childName = str(item.child(row, NAME_COLUMN).data(Qt.DisplayRole).toString())
+            childStatus = str(item.child(row, MEMBERSHIP_STATUS_COLUMN).data(Qt.DisplayRole).toString())
+
+            if (childStatus == user_status['add']) or (childStatus == user_status['in']):
+                researchers.append(childName)
+
+        return researchers
+
     def readUserRecord(self, i):
         rec_file = config.getAuthorityListFile(i)
         if os.path.exists(rec_file):