File: local.py

package info (click to toggle)
nwsserver 2.0.0-2
  • links: PTS
  • area: main
  • in suites: bullseye, buster, jessie, jessie-kfreebsd, sid, stretch, wheezy
  • size: 304 kB
  • ctags: 510
  • sloc: python: 3,252; sh: 76; makefile: 64
file content (103 lines) | stat: -rw-r--r-- 3,768 bytes parent folder | download | duplicates (2)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
#
# Copyright (c) 2007-2009, REvolution Computing, Inc.
#
# NetWorkSpaces is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License as published
# by the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
# USA
#

"""Implementation of NWS server startup as an interpreter-local thread."""

import threading, time
from twisted.internet import reactor
from nwss.server import NwsService

__all__ = ['NwsLocalServerException', 'NwsLocalServer']

class NwsLocalServerException(Exception):
    """Exception thrown if an error occurs while starting the local server."""
    pass

class NwsLocalServer(threading.Thread):
    """Utility to start the NWS server as a thread within a Python
    interpreter."""

    def __init__(self, port=0, interface='', daemon=True,
                 name='NwsLocalServer', **kw):
        threading.Thread.__init__(self, name=name, **kw)
        self.__desired_port = port
        self.__port = None
        self.__interface = interface
        self.__started = False
        self.__condition = threading.Condition()
        self.setDaemon(daemon)
        self.start()

    def get_port(self):
        """Get the actual port number to which we bound."""
        return self.__port._realPortNumber  #pylint: disable-msg=W0212
    port = property(get_port)

    def shutdown(self, timeout=None):
        """Request the shutdown of the server, waiting at most 'timeout'
        seconds for the server thread to stop."""
        reactor.callFromThread(reactor.stop) #pylint: disable-msg=E1101
        self.join(timeout=timeout)

    def wait_until_started(self, timeout=30):
        """Wait for the server thread to finish initializing.  Wait for at most
        'timeout' seconds for the server to start, throwing
        NwsLocalServerException if it has not started by the time the timer
        expires."""

        start_time = time.time()
        timeout_remain = timeout

        self.__condition.acquire()
        try:
            while not self.__started:
                self.__condition.wait(timeout=timeout_remain)
                if timeout is not None:
                    timeout_remain = start_time + timeout - time.time()
                    if timeout_remain <= 0:
                        break

            if not self.__started:
                raise NwsLocalServerException(
                        'local server timeout expired: %d' % timeout)
        finally:
            self.__condition.release()

    def run(self):
        """Main loop of NWS local server thread."""
        srv = NwsService()
        srv.startService()      #pylint: disable-msg=E1101
        try:
            #pylint: disable-msg=E1101,W0212
            self.__port = reactor.listenTCP(self.__desired_port, srv._factory)
            reactor.callWhenRunning(self.__set_started)
            reactor.run(installSignalHandlers=0)
        finally:
            srv.stopService()   #pylint: disable-msg=E1101

    def __set_started(self):
        """Callback from twisted indicating successful startup of the
        server."""
        self.__condition.acquire()
        try:
            self.__started = True
            self.__condition.notifyAll()
        finally:
            self.__condition.release()