mirror of
https://github.com/tahoe-lafs/tahoe-lafs.git
synced 2024-12-30 09:48:56 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
|
|
from foolscap import Tub, Referenceable
|
|
from twisted.application import service
|
|
from twisted.python import log
|
|
import os.path
|
|
|
|
class Storage(service.MultiService, Referenceable):
|
|
pass
|
|
|
|
class Client(service.MultiService):
|
|
CERTFILE = "client.pem"
|
|
|
|
def __init__(self, queen_pburl):
|
|
service.MultiService.__init__(self)
|
|
self.queen_pburl = queen_pburl
|
|
if os.path.exists(self.CERTFILE):
|
|
self.tub = Tub(certData=open(self.CERTFILE, "rb").read())
|
|
else:
|
|
self.tub = Tub()
|
|
f = open(self.CERTFILE, "wb")
|
|
f.write(self.tub.getCertData())
|
|
f.close()
|
|
self.queen = None # self.queen is either None or a RemoteReference
|
|
self.urls = {}
|
|
s = Storage()
|
|
s.setServiceParent(self)
|
|
#self.urls["storage"] = self.tub.registerReference(s, "storage")
|
|
|
|
def startService(self):
|
|
service.MultiService.startService(self)
|
|
if self.queen_pburl:
|
|
self.connector = self.tub.connectTo(self.queen_pburl,
|
|
self._got_queen)
|
|
else:
|
|
log.msg("no queen_pburl, cannot connect")
|
|
|
|
def stopService(self):
|
|
if self.queen_pburl:
|
|
self.connector.stopConnecting()
|
|
|
|
def _got_queen(self, queen):
|
|
log.msg("connected to queen")
|
|
self.queen = queen
|
|
queen.notifyOnDisconnect(self._lost_queen)
|
|
queen.callRemote("hello", urls=self.urls)
|
|
|
|
def _lost_queen(self):
|
|
log.msg("lost connection to queen")
|
|
self.queen = None
|