Python WeakSet.add方法代码示例

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_contains(self):
     from transaction.weakset import WeakSet
     w = WeakSet()
     dummy = Dummy()
     w.add(dummy)
     self.assertEqual(dummy in w, True)
     dummy2 = Dummy()
     self.assertEqual(dummy2 in w, False)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_remove(self):
     from transaction.weakset import WeakSet
     w = WeakSet()
     dummy = Dummy()
     w.add(dummy)
     self.assertEqual(dummy in w, True)
     w.remove(dummy)
     self.assertEqual(dummy in w, False)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
    def test_len(self):
        import gc

        w = WeakSet()
        d1 = Dummy()
        d2 = Dummy()
        w.add(d1)
        w.add(d2)
        self.assertEqual(len(w), 2)
        del d1
        gc.collect()
        self.assertEqual(len(w), 1)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_len(self):
     import gc
     from transaction.weakset import WeakSet
     w = WeakSet()
     d1 = Dummy()
     d2 = Dummy()
     w.add(d1)
     w.add(d2)
     self.assertEqual(len(w), 2)
     del d1
     gc.collect()
     self.assertEqual(len(w), 1)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_as_weakref_list(self):
     w = WeakSet()
     dummy = Dummy()
     dummy2 = Dummy()
     dummy3 = Dummy()
     w.add(dummy)
     w.add(dummy2)
     w.add(dummy3)
     del dummy3
     L = [x() for x in w.as_weakref_list()]
     # L is a list, but it does not have a guaranteed order.
     self.assertTrue(list, type(L))
     self.assertEqual(set(L), set([dummy, dummy2]))

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_map(self):
     from transaction.weakset import WeakSet
     w = WeakSet()
     dummy = Dummy()
     dummy2 = Dummy()
     dummy3 = Dummy()
     w.add(dummy)
     w.add(dummy2)
     w.add(dummy3)
     def poker(x):
         x.poked = 1
     w.map(poker)
     for thing in dummy, dummy2, dummy3:
         self.assertEqual(thing.poked, 1)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_len(self):
     import gc
     from transaction.weakset import WeakSet
     w = WeakSet()
     d1 = Dummy()
     d2 = Dummy()
     w.add(d1)
     w.add(d2)
     self.assertEqual(len(w), 2)
     del d1
     gc.collect()
     if not JYTHON:
         # The Jython GC is non deterministic
         self.assertEqual(len(w), 1)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
 def test_as_weakref_list(self):
     import gc
     from transaction.weakset import WeakSet
     w = WeakSet()
     dummy = Dummy()
     dummy2 = Dummy()
     dummy3 = Dummy()
     w.add(dummy)
     w.add(dummy2)
     w.add(dummy3)
     del dummy3
     gc.collect()
     L = [x() for x in w.as_weakref_list()]
     # L is a list, but it does not have a guaranteed order.
     self.assertTrue(list, type(L))
     self.assertEqual(set(L), set([dummy, dummy2]))

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
class TransactionManager(object):

    def __init__(self):
        self._txn = None
        self._synchs = WeakSet()

    def begin(self):
        if self._txn is not None:
            self._txn.abort()
        txn = self._txn = Transaction(self._synchs, self)
        _new_transaction(txn, self._synchs)
        return txn

    def get(self):
        if self._txn is None:
            self._txn = Transaction(self._synchs, self)
        return self._txn

    def free(self, txn):
        assert txn is self._txn
        self._txn = None

    def registerSynch(self, synch):
        self._synchs.add(synch)

    def unregisterSynch(self, synch):
        self._synchs.remove(synch)

    def isDoomed(self):
        return self.get().isDoomed()

    def doom(self):
        return self.get().doom()

    def commit(self):
        return self.get().commit()

    def abort(self):
        return self.get().abort()

    def savepoint(self, optimistic=False):
        return self.get().savepoint(optimistic)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
    def test_map_w_gced_element(self):
        import gc
        from transaction.weakset import WeakSet
        w = WeakSet()
        dummy = Dummy()
        dummy2 = Dummy()
        dummy3 = [Dummy()]
        w.add(dummy)
        w.add(dummy2)
        w.add(dummy3[0])

        _orig = w.as_weakref_list
        def _as_weakref_list():
            # simulate race condition during iteration of list
            # object is collected after being iterated.
            result = _orig()
            del dummy3[:]
            gc.collect()
            return result
        w.as_weakref_list = _as_weakref_list

        def poker(x):
            x.poked = 1
        w.map(poker)
        for thing in dummy, dummy2:
            self.assertEqual(thing.poked, 1)

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
class TransactionManager(object):

    def __init__(self):
        self._txn = None
        self._synchs = WeakSet()

    def begin(self):
        if self._txn is not None:
            self._txn.abort()
        txn = self._txn = Transaction(self._synchs, self)
        _new_transaction(txn, self._synchs)
        return txn

    __enter__ = lambda self: self.begin()

    def get(self):
        if self._txn is None:
            self._txn = Transaction(self._synchs, self)
        return self._txn

    def free(self, txn):
        assert txn is self._txn
        self._txn = None

    def registerSynch(self, synch):
        self._synchs.add(synch)

    def unregisterSynch(self, synch):
        self._synchs.remove(synch)

    def isDoomed(self):
        return self.get().isDoomed()

    def doom(self):
        return self.get().doom()

    def commit(self):
        return self.get().commit()

    def abort(self):
        return self.get().abort()

    def __exit__(self, t, v, tb):
        if v is None:
            self.commit()
        else:
            self.abort()

    def savepoint(self, optimistic=False):
        return self.get().savepoint(optimistic)

    def attempts(self, number=3):
        assert number > 0
        while number:
            number -= 1
            if number:
                yield Attempt(self)
            else:
                yield self

    def _retryable(self, error_type, error):
        if issubclass(error_type, TransientError):
            return True

        for dm in self.get()._resources:
            should_retry = getattr(dm, 'should_retry', None)
            if (should_retry is not None) and should_retry(error):
                return True

# 需要导入模块: from transaction.weakset import WeakSet [as 别名]
# 或者: from transaction.weakset.WeakSet import add [as 别名]
class TransactionManager(object):

    def __init__(self):
        self._txn = None
        self._synchs = WeakSet()

    def begin(self):
        """ See ITransactionManager.
        """
        if self._txn is not None:
            self._txn.abort()
        txn = self._txn = Transaction(self._synchs, self)
        _new_transaction(txn, self._synchs)
        return txn

    __enter__ = lambda self: self.begin()

    def get(self):
        """ See ITransactionManager.
        """
        if self._txn is None:
            self._txn = Transaction(self._synchs, self)
        return self._txn

    def free(self, txn):
        if txn is not self._txn:
            raise ValueError("Foreign transaction")
        self._txn = None

    def registerSynch(self, synch):
        """ See ITransactionManager.
        """
        self._synchs.add(synch)
        if self._txn is not None:
            synch.newTransaction(self._txn)

    def unregisterSynch(self, synch):
        """ See ITransactionManager.
        """
        self._synchs.remove(synch)

    def clearSynchs(self):
        """ See ITransactionManager.
        """
        self._synchs.clear()

    def registeredSynchs(self):
        """ See ITransactionManager.
        """
        return bool(self._synchs)

    def isDoomed(self):
        """ See ITransactionManager.
        """
        return self.get().isDoomed()

    def doom(self):
        """ See ITransactionManager.
        """
        return self.get().doom()

    def commit(self):
        """ See ITransactionManager.
        """
        return self.get().commit()

    def abort(self):
        """ See ITransactionManager.
        """
        return self.get().abort()

    def __exit__(self, t, v, tb):
        if v is None:
            self.commit()
        else:
            self.abort()

    def savepoint(self, optimistic=False):
        """ See ITransactionManager.
        """
        return self.get().savepoint(optimistic)

    def attempts(self, number=3):
        if number <= 0:
            raise ValueError("number must be positive")
        while number:
            number -= 1
            if number:
                yield Attempt(self)
            else:
                yield self

    def _retryable(self, error_type, error):
        if issubclass(error_type, TransientError):
            return True

        for dm in self.get()._resources:
            should_retry = getattr(dm, 'should_retry', None)
            if (should_retry is not None) and should_retry(error):
                return True