/usr/lib/python3/dist-packages/PyTango/green.py is in python3-pytango 8.1.1-1build3.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 | # ------------------------------------------------------------------------------
# This file is part of PyTango (http://www.tinyurl.com/PyTango)
#
# Copyright 2006-2012 CELLS / ALBA Synchrotron, Bellaterra, Spain
# Copyright 2013-2014 European Synchrotron Radiation Facility, Grenoble, France
#
# Distributed under the terms of the GNU Lesser General Public License,
# either version 3 of the License, or (at your option) any later version.
# See LICENSE.txt for more info.
# ------------------------------------------------------------------------------
__all__ = ["get_green_mode", "set_green_mode",
"get_executor", "submit", "spawn",
"get_synch_executor", "synch_submit",
"get_gevent_executor", "gevent_submit",
"get_futures_executor", "futures_submit",
"result", "submitable", "green"]
__docformat__ = "restructuredtext"
import os
from functools import wraps
from ._PyTango import GreenMode
from .tango_gevent import get_global_executor as get_gevent_executor
from .tango_gevent import submit as gevent_submit
from .tango_futures import get_global_executor as get_futures_executor
from .tango_futures import submit as futures_submit
__default_green_mode = GreenMode.Synchronous
try:
__current_green_mode = getattr(GreenMode,
os.environ.get("PYTANGO_GREEN_MODE",
"Synchronous").capitalize())
except:
__current_green_mode = __default_green_mode
def set_green_mode(green_mode=None):
"""Sets the global default PyTango green mode.
Advice: Use only in your final application. Don't use this in a python library
in order not to interfere with the beavior of other libraries and/or
application where your library is being.
:param green_mode: the new global default PyTango green mode
:type green_mode: GreenMode
"""
global __current_green_mode
if __current_green_mode == green_mode:
return
if green_mode == GreenMode.Gevent:
# check if we can change to gevent mode
import PyTango.gevent
elif green_mode == GreenMode.Futures:
# check if we can change to futures mode
import PyTango.futures
__current_green_mode = green_mode
def get_green_mode():
"""Returns the current global default PyTango green mode.
:returns: the current global default PyTango green mode
:rtype: GreenMode
"""
return __current_green_mode
class SynchExecutor(object):
def submit(self, fn, *args, **kwargs):
return fn(*args, **kwargs)
__synch_executor = SynchExecutor()
def get_synch_executor():
return __synch_executor
def synch_submit(fn, *args, **kwargs):
return get_synch_executor().submit(fn, *args, **kwargs)
__executor_map = {
GreenMode.Synchronous: get_synch_executor,
GreenMode.Futures: get_futures_executor,
GreenMode.Gevent: get_gevent_executor,
}
__submit_map = {
GreenMode.Synchronous: synch_submit,
GreenMode.Futures: futures_submit,
GreenMode.Gevent: gevent_submit,
}
def get_executor(mode):
return __executor_map[mode]()
def get_submitter(mode):
executor = get_executor(mode)
if mode == GreenMode.Gevent:
return executor.spawn
return executor.submit
def submit(mode, fn, *args, **kwargs):
return get_submitter(mode)(fn, *args, **kwargs)
spawn = submit
def result(value, green_mode, wait=True, timeout=None):
if wait and not green_mode is GreenMode.Synchronous:
if green_mode == GreenMode.Futures:
return value.result(timeout=timeout)
elif green_mode == GreenMode.Gevent:
return value.get(timeout=timeout)
return value
def submitable(obj, green_mode=None):
"""Returns a proper submit callable for the given object.
If the object has *_executors* and *_green_mode* members it returns a submit
callable for the executor corresponding to the green_mode.
Otherwise it returns the global submit callable for the given green_mode
:returns: green_mode, submit callable"""
# determine the efective green_mode
if green_mode is None:
if hasattr(obj, "get_green_mode"):
green_mode = obj.get_green_mode()
else:
green_mode = get_green_mode()
if green_mode == GreenMode.Synchronous:
return green_mode, synch_submit
has_executors = hasattr(obj, "_executors")
s_func = __submit_map[green_mode]
if green_mode == GreenMode.Futures:
if has_executors:
executor = obj._executors.get(GreenMode.Futures)
if executor:
s_func = executor.submit
elif green_mode == GreenMode.Gevent:
if has_executors:
executor = obj._executors.get(GreenMode.Gevent)
if executor:
s_func = executor.spawn
else:
raise TypeError("Undefined green_mode '%s' for %s" % (str(green_mode)), str(obj))
return green_mode, s_func
def green(fn):
"""make a method green. Can be used as a decorator"""
@wraps(fn)
def greener(self, *args, **kwargs):
# first take out all green parameters
green_mode = kwargs.pop('green_mode', None)
wait = kwargs.pop('wait', True)
timeout = kwargs.pop('timeout', None)
# get the proper submitable for the given green_mode
green_mode, submit = submitable(self, green_mode)
# submit the method
ret = submit(fn, self, *args, **kwargs)
# return the proper result
return result(ret, green_mode, wait=wait, timeout=timeout)
return greener
|