sface/sfirenew.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 sface.config import config
  10 from sface.sfiprocess import SfiProcess
  11
  12 class SfiRenewer(QObject):
  13     def __init__(self, hrn, newExpiration, parent=None):
  14         QObject.__init__(self, parent)
  15         self.hrn = hrn
  16         self.newExpiration = newExpiration
  17
  18         self.renewProcess = SfiProcess(self)
  19         self.connect(self.renewProcess, SIGNAL('finished()'), self.finishedGetRecord)
  20         self.renewProcess.getRecord(hrn=config.getSlice(), filename="/tmp/slicerecord")
  21
  22     def finishedGetRecord(self):
  23         f = open("/tmp/slicerecord", "r")
  24         data = f.read()
  25         f.close()
  26
  27         # find the expiration time
  28         exp = re.compile('expires="[^"]*"')
  29         if exp.search(data)==None:
  30             # didn't find it
  31             self.emitFinished("failure", "failed to find expiration in slice record")
  32             return
  33
  34         # change the expiration time
  35         delta = 24*60*60 # always extend the slice by one extra day to cover slop for time zone differences
  36         data = exp.sub('expires="' + str(calendar.timegm(self.newExpiration.timetuple())+delta) + '"', data)
  37
  38         open("/tmp/slicerecord", "w").write(data)
  39
  40         self.disconnect(self.renewProcess, SIGNAL('finished()'), self.finishedGetRecord)
  41         self.connect(self.renewProcess, SIGNAL('finished()'), self.finishedUpdateRecord)
  42
  43         self.renewProcess.updateRecord("/tmp/slicerecord")
  44
  45     def finishedUpdateRecord(self):
  46         # we have to force sfi.py to download an updated slice credential
  47         sliceCredName = config.fullpath("slice_" + self.hrn.split(".")[-1] + ".cred")
  48         if os.path.exists(sliceCredName):
  49             os.remove(sliceCredName)
  50
  51         open("/tmp/expiration", "w").write(self.newExpiration.strftime("%Y-%m-%dT%H:%M:%SZ"))
  52
  53         # call renewSlivers on the aggregate
  54         self.disconnect(self.renewProcess, SIGNAL('finished()'), self.finishedUpdateRecord)
  55         self.connect(self.renewProcess, SIGNAL('finished()'), self.finishedRenewSlivers)
  56         self.renewProcess.renewSlivers(self.newExpiration.strftime("%Y-%m-%dT%H:%M:%SZ"))
  57
  58     def finishedRenewSlivers(self):
  59         self.emitFinished("success")
  60
  61     def emitFinished(self, status, statusMsg=None):
  62         self.status = status
  63         self.statusMsg = statusMsg
  64         self.emit(SIGNAL("finished()"))
  65