File: stash_responder_blocking_wsh.py

package info (click to toggle)
thunderbird 1%3A91.13.0-1~deb11u1
  • links: PTS, VCS
  • area: main
  • in suites: bullseye
  • size: 2,953,400 kB
  • sloc: cpp: 6,084,049; javascript: 4,790,441; ansic: 3,341,496; python: 862,958; asm: 366,542; xml: 204,277; java: 152,477; sh: 111,436; makefile: 21,388; perl: 15,312; yacc: 4,583; objc: 3,026; lex: 1,720; exp: 762; pascal: 635; awk: 564; sql: 453; php: 436; lisp: 432; ruby: 99; sed: 69; csh: 45
file content (45 lines) | stat: -rwxr-xr-x 1,072 bytes parent folder | download | duplicates (12) 
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
 
#!/usr/bin/python
import json
import threading
import wptserve.stash
from mod_pywebsocket import msgutil

address, authkey = wptserve.stash.load_env_config()
path = "/stash_responder_blocking"
stash = wptserve.stash.Stash(path, address=address, authkey=authkey)
cv = threading.Condition()

def handle_set(key, value):
    with cv:
      stash.put(key, value)
      cv.notify_all()

def handle_get(key):
    with cv:
        while True:
            value = stash.take(key)
            if value is not None:
                return value
            cv.wait()

def web_socket_do_extra_handshake(request):
    pass

def web_socket_transfer_data(request):
    line = request.ws_stream.receive_message()

    query = json.loads(line)
    action = query["action"]
    key = query["key"]

    if action == "set":
        value = query["value"]
        handle_set(key, value)
        response = {}
    elif action == "get":
        value = handle_get(key)
        response = {"value": value}
    else:
        response = {}

    msgutil.send_message(request, json.dumps(response))