queue.sqlite -> queue.sa

pull/108/head
Martin Zimmermann 10 years ago
parent e706fabb26
commit 08cb3cd324

@ -188,7 +188,6 @@ class Worker(threading.Thread):
time.sleep(f * Worker.interval) time.sleep(f * Worker.interval)
from .sqlite import SQLite3Queue from .sa import SAQueue
__all__ = ["Full", "Empty", "Retry", "Timeout", "Message", "Queue", __all__ = ["Full", "Empty", "Retry", "Timeout", "Message", "Queue", "SAQueue"]
"SQLite3Queue"]

@ -16,12 +16,12 @@ pickle = lambda val: json.dumps(val).encode("utf-8")
unpickle = lambda val: json.loads(val.decode("utf-8")) unpickle = lambda val: json.loads(val.decode("utf-8"))
class SQLite3Queue(Queue): class SAQueue(Queue):
"""Implements a shared queue using SQLAlchemy Core """Implements a shared queue using SQLAlchemy Core
""" """
def __init__(self, db, maxlen=-1, timeout=2**10): def __init__(self, db, maxlen=-1, timeout=2**10):
super(SQLite3Queue, self).__init__(maxlen, timeout) super(SAQueue, self).__init__(maxlen, timeout)
self.metadata = MetaData() self.metadata = MetaData()
self.engine = create_engine(db) self.engine = create_engine(db)
self.queue = Table("queue", self.metadata, self.queue = Table("queue", self.metadata,

@ -6,7 +6,6 @@ import unittest
from isso.compat import text_type as str from isso.compat import text_type as str
from isso.db import Adapter
from isso.cache import Cache, SACache from isso.cache import Cache, SACache
ns = "test" ns = "test"
@ -56,4 +55,4 @@ class TestCache(unittest.TestCase):
class TestSQLite3Cache(TestCache): class TestSQLite3Cache(TestCache):
def setUp(self): def setUp(self):
self.cache = SACache(Adapter("sqlite:///:memory:"), threshold=8) self.cache = SACache("sqlite:///:memory:", threshold=8)

@ -5,8 +5,7 @@ from __future__ import unicode_literals
import unittest import unittest
import datetime import datetime
from isso.db import SQLite3 from isso.queue import Message, Queue, Full, Empty, Timeout, SAQueue
from isso.queue import Message, Queue, Full, Empty, Timeout, SQLite3Queue
class TestMessage(unittest.TestCase): class TestMessage(unittest.TestCase):
@ -100,4 +99,4 @@ class TestQueue(unittest.TestCase):
class TestSQLite3Queue(TestQueue): class TestSQLite3Queue(TestQueue):
def setUp(self): def setUp(self):
self.cls = lambda *x, **z: SQLite3Queue("sqlite:///:memory:", *x, **z) self.cls = lambda *x, **z: SAQueue("sqlite:///:memory:", *x, **z)

Loading…
Cancel
Save