File: vine_python_no_serialization.py

package info (click to toggle)
cctools 1%3A7.14.5-1
  • links: PTS, VCS
  • area: main
  • in suites: trixie
  • size: 36,956 kB
  • sloc: ansic: 114,614; python: 29,532; cpp: 20,313; sh: 13,675; perl: 4,056; xml: 3,688; makefile: 1,436
file content (40 lines) | stat: -rw-r--r-- 979 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
#! /usr/bin/env python

import sys
import ndcctools.taskvine as vine

port_file = None
try:
    port_file = sys.argv[1]
except IndexError:
    sys.stderr.write("Usage: {} PORTFILE\n".format(sys.argv[0]))
    raise


# Define a function to invoke remotely
def my_fun():
    print("the function")
    return "look ma no serialization".encode('utf8')


# Create a new m
m = vine.Manager(port=[9123, 9130])
print("listening on port {}".format(m.port))
with open(port_file, "w") as f:
    f.write(str(m.port))

# Submit several tasks for execution:
print("submitting tasks...")
t = vine.PythonTask(my_fun)
t.disable_output_serialization()
m.submit(t)

while not m.empty():
    t = m.wait(5)
    if t:
        if t.successful():
            print(t.output)
            assert t.output == "look ma no serialization".encode('utf8')
        else:
            print(f"Something went wrong with the task: {t.result}\nstdout: {t.std_output}")
# vim: set sts=4 sw=4 ts=4 expandtab ft=python: