from celery import Celery, Task
from redis import StrictRedis

from charat2.model import sm
from charat2.model.connections import redis_pool

celery = Celery("newparp", include=[
    "charat2.tasks.background",
    "charat2.tasks.matchmaker",
    "charat2.tasks.reaper",
    "charat2.tasks.roulette_matchmaker",
])

celery.config_from_object('charat2.tasks.config')

class WorkerTask(Task):
    abstrct = True
    _db = None
    _redis = None

    @property
    def db(self):
        if self._db is None:
            self._db = sm()

        return self._db

    @property
    def redis(self):
        if self._redis is None:
            self._redis = StrictRedis(connection_pool=redis_pool)

        return self._redis

from celery import Celery, Task
from classtools import reify
from redis import StrictRedis

from charat2.model import sm
from charat2.model.connections import redis_pool

celery = Celery("newparp", include=[
    "charat2.tasks.background",
    "charat2.tasks.matchmaker",
    "charat2.tasks.reaper",
    "charat2.tasks.roulette_matchmaker",
])

celery.config_from_object('charat2.tasks.config')

class WorkerTask(Task):
    abstrct = True

    @reify
    def db(self):
        return sm()

    @reify
    def redis(self):
        return StrictRedis(connection_pool=redis_pool)

    def after_return(self, *args, **kwargs):
        if hasattr(self, "db"):
            self.db.close()
            del self.db

        if hasattr(self, "redis"):
            del self.redis

Use reify instead of properties for the task backend connections.
Use reify instead of properties for the task backend connections.
