summaryrefslogtreecommitdiff
path: root/sleekxmpp/util/__init__.py
blob: 47a935af4d9f039a3a05a70f9ca645730839d655 (plain)
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
# -*- coding: utf-8 -*-
"""
    sleekxmpp.util
    ~~~~~~~~~~~~~~

    Part of SleekXMPP: The Sleek XMPP Library

    :copyright: (c) 2012 Nathanael C. Fritz, Lance J.T. Stout
    :license: MIT, see LICENSE for more details
"""


from sleekxmpp.util.misc_ops import bytes, unicode, hashes, hash, \
                                    num_to_bytes, bytes_to_num, quote, \
                                    XOR, safedict


# =====================================================================
# Standardize import of Queue class:

import sys

def _gevent_threads_enabled():
    if not 'gevent' in sys.modules:
        return False
    try:
        from gevent import thread as green_thread
        thread = __import__('thread')
        return thread.LockType is green_thread.LockType
    except ImportError:
        return False

if _gevent_threads_enabled():
    import gevent.queue as queue
    _queue = queue.JoinableQueue
else:
    try:
        import queue
    except ImportError:
        import Queue as queue
    _queue = queue.Queue
class Queue(_queue):
    def put(self, item, block=True, timeout=None):
        if _queue.full(self):
            _queue.get(self)
        return _queue.put(self, item, block, timeout)

QueueEmpty = queue.Empty