Gecko [ goons4good.org ]

Name	Size	Permission	Date
__pycache__	[ DIR ]	drwxr-xr-x	2022-03-29 05:07
__init__.py	616 B	-rw-r--r--	2018-02-10 08:02
_cdefs.h	1.94 KB	-rw-r--r--	2018-02-10 08:02
_cffi.py	3.79 KB	-rw-r--r--	2018-02-10 08:02
_poll.py	2.77 KB	-rw-r--r--	2018-02-10 08:02
_verify.c	208 B	-rw-r--r--	2018-02-10 08:02
constants.py	357 B	-rw-r--r--	2018-02-10 08:02
context.py	2.52 KB	-rw-r--r--	2018-02-10 08:02
devices.py	580 B	-rw-r--r--	2018-02-10 08:02
error.py	353 B	-rw-r--r--	2018-02-10 08:02
message.py	1.36 KB	-rw-r--r--	2018-02-10 08:02
socket.py	8.49 KB	-rw-r--r--	2018-02-10 08:02
utils.py	1.45 KB	-rw-r--r--	2018-02-10 08:02

Rename

# coding: utf-8
"""zmq Context class"""

# Copyright (C) PyZMQ Developers
# Distributed under the terms of the Modified BSD License.

import weakref

from ._cffi import C, ffi

from .constants import EINVAL, IO_THREADS, LINGER

from zmq.error import ZMQError, InterruptedSystemCall, _check_rc

class Context(object):
    _zmq_ctx = None
    _iothreads = None
    _closed = None
    _sockets = None
    _shadow = False

def __init__(self, io_threads=1, shadow=None):
        
        if shadow:
            self._zmq_ctx = ffi.cast("void *", shadow)
            self._shadow = True
        else:
            self._shadow = False
            if not io_threads >= 0:
                raise ZMQError(EINVAL)
        
            self._zmq_ctx = C.zmq_ctx_new()
        if self._zmq_ctx == ffi.NULL:
            raise ZMQError(C.zmq_errno())
        if not shadow:
            C.zmq_ctx_set(self._zmq_ctx, IO_THREADS, io_threads)
        self._closed = False
        self._sockets = set()
    
    @property
    def underlying(self):
        """The address of the underlying libzmq context"""
        return int(ffi.cast('size_t', self._zmq_ctx))
    
    @property
    def closed(self):
        return self._closed

def _add_socket(self, socket):
        ref = weakref.ref(socket)
        self._sockets.add(ref)
        return ref

def _rm_socket(self, ref):
        if ref in self._sockets:
            self._sockets.remove(ref)

def set(self, option, value):
        """set a context option
        
        see zmq_ctx_set
        """
        rc = C.zmq_ctx_set(self._zmq_ctx, option, value)
        _check_rc(rc)

def get(self, option):
        """get context option
        
        see zmq_ctx_get
        """
        rc = C.zmq_ctx_get(self._zmq_ctx, option)
        _check_rc(rc)
        return rc

def term(self):
        if self.closed:
            return
        
        rc = C.zmq_ctx_destroy(self._zmq_ctx)
        try:
            _check_rc(rc)
        except InterruptedSystemCall:
            # ignore interrupted term
            # see PEP 475 notes about close & EINTR for why
            pass

self._zmq_ctx = None
        self._closed = True

def destroy(self, linger=None):
        if self.closed:
            return

sockets = self._sockets
        self._sockets = set()
        for s in sockets:
            s = s()
            if s and not s.closed:
                if linger is not None:
                    s.setsockopt(LINGER, linger)
                s.close()
        
        self.term()

__all__ = ['Context']