queue.sqlite -> queue.sa

This commit is contained in:
Martin Zimmermann 2014-07-25 17:57:27 +02:00
parent e706fabb26
commit 08cb3cd324
4 changed files with 7 additions and 10 deletions

View File

@ -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"]

View File

@ -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,

View File

@ -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)

View File

@ -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)