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$" 
 30   
 31   
32 -class ManholeMixin(object):
33 """ 34 """ 35
36 - def __init__(self, args):
37 38 self.useSSH = False 39 self.authorizedKeysFile = None 40 self.port = None 41 self.requestedPortNum = None 42 self.namespace = {}
43
44 - def start(self, namespace):
45 self.namespace.update(namespace) 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, obj):
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 88
89 -class ManagerManholePlug(base.ManagerPlug, ManholeMixin):
90
91 - def __init__(self, args):
94
95 - def start(self, vishnu):
96 namespace = {'vishnu': vishnu} 97 ManholeMixin.start(self, namespace)
98 99
100 -class ComponentManholePlug(base.ComponentPlug, ManholeMixin):
101
102 - def __init__(self, args):
105
106 - def start(self, component):
107 namespace = {'component': component} 108 ManholeMixin.start(self, namespace)
109