sface/sficreate.py

   1 import calendar
   2 import datetime
   3 import os
   4 import re
   5 import sys
   6 import time
   7
   8 from PyQt4.QtCore import *
   9 from PyQt4.QtGui import *
  10 from sface.config import config
  11 from sface.sfiprocess import SfiProcess
  12 from sface.screens.userscreen import UserView, UserModel
  13
  14 class CreateWindow(QDialog):
  15     def __init__(self, parent=None):
  16         super(CreateWindow, self).__init__(parent)
  17         self.setWindowTitle("Create new Slice")
  18
  19         self.renewProcess = None
  20
  21         self.sliceWasCreated = False
  22
  23         hrnLabel = QLabel("Slice HRN:")
  24         self.hrnEdit = QLineEdit()
  25         urlLabel = QLabel("Project URL:")
  26         self.urlEdit = QLineEdit()
  27         descLabel = QLabel("Description:")
  28         self.descEdit = QTextEdit()
  29         self.descEdit.setMinimumWidth(400)
  30         self.descEdit.setMaximumHeight(100)
  31
  32         self.userLabel = QLabel("Researchers:")
  33         self.userView = UserView()
  34
  35         self.status = QLabel("")
  36
  37         self.buttonBox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)
  38         self.buttonBox.addButton("refresh", QDialogButtonBox.ActionRole)
  39         self.buttonBox.button(QDialogButtonBox.Ok).setDefault(True)
  40
  41         layout = QVBoxLayout()
  42         layout.addWidget(hrnLabel)
  43         layout.addWidget(self.hrnEdit)
  44         layout.addWidget(urlLabel)
  45         layout.addWidget(self.urlEdit)
  46         layout.addWidget(descLabel)
  47         layout.addWidget(self.descEdit)
  48         layout.addWidget(self.userLabel)
  49         layout.addWidget(self.userView)
  50         layout.addWidget(self.status)
  51         layout.addWidget(self.buttonBox)
  52         self.setLayout(layout)
  53
  54         self.userModel = UserModel(parent=self)
  55
  56         self.connect(self.buttonBox, SIGNAL("accepted()"), self, SLOT("accept()"))
  57         self.connect(self.buttonBox, SIGNAL("rejected()"), self, SLOT("reject()"))
  58         self.connect(self.buttonBox, SIGNAL("clicked(QAbstractButton *)"), self.clicked)
  59
  60         self.updateView()
  61
  62         if (self.userModel.rowCount() == 0):
  63             self.refreshAuthority()
  64
  65     def clicked(self, button):
  66         if button.text() == "refresh":
  67             self.refreshAuthority()
  68
  69     def refreshAuthority(self):
  70         self.refreshProcess = SfiProcess(self)
  71         self.connect(self.refreshProcess, SIGNAL('finished()'), self.getAuthorityRecordFinished)
  72
  73         self.refreshProcess.listRecords(config.getAuthority(), "user", config.getAuthorityListFile())
  74         self.setStatus("Refreshing user list...")
  75
  76     def accept(self):
  77         auth = config.getAuthority()
  78         desc = str(self.descEdit.toPlainText())
  79         hrn = str(self.hrnEdit.text())
  80         type = "slice"
  81         url = str(self.urlEdit.text())
  82
  83         researchers = self.userModel.getResearchers()
  84
  85         if not hrn.startswith(auth):
  86             QMessageBox.warning(self, "Invalid HRN", "HRN must be within your current authority (%s)" % auth)
  87             return
  88
  89         if not (url.startswith("http://") or url.startswith("https://")):
  90             QMessageBox.warning(self, "Invalid URL", "URL must start with http:// or https://")
  91             return
  92
  93         if not desc:
  94             QMessageBox.warning(self, "Invalid Description", "Description is too short")
  95             return
  96
  97         if not researchers:
  98             QMessageBox.warning(self, "Invalid Researchers", "Please add at least one researcher by double-clicking on a researcher name")
  99             return
 100
 101         self.setStatus("Registering Slice...")
 102
 103         self.createProcess = SfiProcess(self)
 104         self.connect(self.createProcess, SIGNAL('finished()'), self.createFinished)
 105
 106         self.buttonBox.setEnabled(False)
 107
 108         researcherXml = "<researcher>" + "</researcher><researcher>".join(researchers) + "</researcher>"
 109
 110         newSliceRecord = os.path.expanduser("~/.sfi/newslice.record")
 111         file(newSliceRecord, "w").write('<record authority="%s" description="%s" hrn="%s" type="%s" url="%s">%s</record>' % (auth, desc, hrn, type, url, researcherXml))
 112         self.createProcess.addRecord(newSliceRecord)
 113
 114     def getAuthorityRecordFinished(self):
 115         self.setStatus("<font color='green'>User list refreshed.</font>")
 116         self.updateView()
 117
 118     def setStatus(self, x):
 119         self.status.setText(x)
 120
 121     def createFinished(self):
 122         faultString = self.createProcess.getFaultString()
 123         if not faultString:
 124             self.setStatus("<font color='green'>Slice created.</font>")
 125             self.sliceWasCreated = True
 126             self.buttonBox.setEnabled(True)
 127             self.buttonBox.clear()
 128             self.buttonBox.addButton(QDialogButtonBox.Close)
 129         else:
 130             self.setStatus("<font color='red'>Slice creation failed: %s</font>" % (faultString))
 131             self.sliceWasCreated = False
 132             self.buttonBox.setEnabled(True)
 133
 134         self.disconnect(self.createProcess, SIGNAL('finished()'), self.createFinished)
 135         self.createProcess = None
 136
 137     def getHrn(self):
 138         return self.hrnEdit.text()
 139
 140     def updateView(self):
 141         self.userModel.updateModel(None)
 142
 143         self.userView.setModel(self.userModel)
 144         self.userView.hideUnusableColumns()
 145         self.userView.resizeColumnToContents(0)
 146