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 
26  from flumotion.common.manhole import openSSHManhole 
27  from flumotion.common.manhole import openAnonymousTelnetManhole 
28   
29  __version__ = "$Rev: 7363 $" 
30   
31   
32 -class ManagerManholePlug(base.ManagerPlug):
33 """ 34 """ 35
36 - def __init__(self, args):
37 base.ManagerPlug.__init__(self, args) 38 self.useSSH = False 39 self.authorizedKeysFile = None 40 self.port = None 41 self.requestedPortNum = None
42
43 - def start(self, vishnu):
44 self.namespace = {'vishnu': vishnu} 45 46 props = self.args['properties'] 47 if 'ssh-authorized-keys-file' in props: 48 self.useSSH = True 49 self.authorizedKeysFile = os.path.expanduser( 50 props['ssh-authorized-keys-file']) 51 52 self.requestedPortNum = props.get('port', -1) 53 54 self._insinuate() 55 56 if props.get('initially-open', False): 57 self.openManhole()
58
59 - def stop(self, vishnu):
60 self.closeManhole()
61
62 - def _insinuate(self):
63 # "And I wish you didn't have the devil's curly hair!" 64 from flumotion.manager.admin import AdminAvatar 65 AdminAvatar.perspective_openManhole = self.openManhole 66 AdminAvatar.perspective_closeManhole = self.closeManhole
67
68 - def openManhole(self):
69 if not self.port: 70 if self.useSSH: 71 self.port = openSSHManhole(self.authorizedKeysFile, 72 self.namespace, 73 self.requestedPortNum) 74 else: 75 self.port = openAnonymousTelnetManhole(self.namespace, 76 self.requestedPortNum) 77 78 return self.port.getHost().port
79
80 - def closeManhole(self):
81 if self.port: 82 ret = self.port.loseConnection() 83 else: 84 ret = defer.succeed(None) 85 self.port = None 86 return ret
87