Package flumotion :: Package component :: Package plugs :: Module manhole
[hide private]

Source Code for Module flumotion.component.plugs.manhole

 1  # -*- Mode: Python -*- 
 2  # vi:si:et:sw=4:sts=4:ts=4 
 3  # 
 4  # Flumotion - a streaming media server 
 5  # Copyright (C) 2006 Fluendo, S.L. (www.fluendo.com). 
 6  # All rights reserved. 
 7   
 8  # This file may be distributed and/or modified under the terms of 
 9  # the GNU General Public License version 2 as published by 
10  # the Free Software Foundation. 
11  # This file is distributed without any warranty; without even the implied 
12  # warranty of merchantability or fitness for a particular purpose. 
13  # See "LICENSE.GPL" in the source distribution for more information. 
14   
15  # Licensees having purchased or holding a valid Flumotion Advanced 
16  # Streaming Server license may use this file in accordance with the 
17  # Flumotion Advanced Streaming Server Commercial License Agreement. 
18  # See "LICENSE.Flumotion" in the source distribution for more information. 
19   
20  # Headers in this file shall remain intact. 
21   
22  import os 
23  from twisted.internet import defer 
24   
25  from flumotion.component.plugs import base, lifecycle 
26  from flumotion.common.manhole import openSSHManhole 
27  from flumotion.common.manhole import openAnonymousTelnetManhole 
28   
29  __version__ = "$Rev: 6125 $" 
30   
31   
32 -class ManagerManholePlug(lifecycle.ManagerLifecycle):
33 """ 34 """
35 - def __init__(self, args):
36 lifecycle.ManagerLifecycle.__init__(self, args) 37 self.useSSH = False 38 self.authorizedKeysFile = None 39 self.port = None 40 self.requestedPortNum = None
41
42 - def start(self, vishnu):
43 self.namespace = {'vishnu': vishnu} 44 45 props = self.args['properties'] 46 if 'ssh-authorized-keys-file' in props: 47 self.useSSH = True 48 self.authorizedKeysFile = os.path.expanduser( 49 props['ssh-authorized-keys-file']) 50 51 self.requestedPortNum = props.get('port', -1) 52 53 self._insinuate() 54 55 if props.get('initially-open', False): 56 self.openManhole()
57
58 - def stop(self, vishnu):
59 self.closeManhole()
60
61 - def _insinuate(self):
62 # "And I wish you didn't have the devil's curly hair!" 63 from flumotion.manager.admin import AdminAvatar 64 AdminAvatar.perspective_openManhole = self.openManhole 65 AdminAvatar.perspective_closeManhole = self.closeManhole
66
67 - def openManhole(self):
68 if not self.port: 69 if self.useSSH: 70 self.port = openSSHManhole(self.authorizedKeysFile, 71 self.namespace, 72 self.requestedPortNum) 73 else: 74 self.port = openAnonymousTelnetManhole(self.namespace, 75 self.requestedPortNum) 76 77 return self.port.getHost().port
78
79 - def closeManhole(self):
80 if self.port: 81 ret = self.port.loseConnection() 82 else: 83 ret = defer.succeed(None) 84 self.port = None 85 return ret
86