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
|
r"""
# -----------------------------------------------------------------------------
# This WebServer aims to provide a good starting point when creating
# wslink based service.
#
# $ python ./server/server.py --content ./www
# -----------------------------------------------------------------------------
"""
import os
import sys
# Try handle virtual env if provided
if "--virtual-env" in sys.argv:
virtualEnvPath = sys.argv[sys.argv.index("--virtual-env") + 1]
virtualEnv = virtualEnvPath + "/bin/activate_this.py"
execfile(virtualEnv, dict(__file__=virtualEnv))
import argparse
from wslink.websocket import ServerProtocol
from wslink import server
from api import PubSubAPI
# -----------------------------------------------------------------------------
class WebServer(ServerProtocol):
authKey = "wslink-secret"
@staticmethod
def add_arguments(parser):
parser.add_argument(
"--virtual-env", default=None, help="Path to virtual environment to use"
)
@staticmethod
def configure(options):
WebServer.authKey = options.authKey
def initialize(self):
self.registerLinkProtocol(PubSubAPI())
# Update authentication key to use
self.updateSecret(WebServer.authKey)
# -----------------------------------------------------------------------------
# Main
# -----------------------------------------------------------------------------
if __name__ == "__main__":
# Create argument parser
parser = argparse.ArgumentParser(description="WSLink sample application")
# Add arguments
server.add_arguments(parser)
WebServer.add_arguments(parser)
args = parser.parse_args()
WebServer.configure(args)
# Start web server
server.start(None, WebServer)
|
