/usr/share/pyshared/celery/db/session.py is in python-celery 2.5.3-4.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
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 | # -*- coding: utf-8 -*-
from __future__ import absolute_import
from collections import defaultdict
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.ext.declarative import declarative_base
ResultModelBase = declarative_base()
_SETUP = defaultdict(lambda: False)
_ENGINES = {}
_SESSIONS = {}
def get_engine(dburi, **kwargs):
if dburi not in _ENGINES:
_ENGINES[dburi] = create_engine(dburi, **kwargs)
return _ENGINES[dburi]
def create_session(dburi, short_lived_sessions=False, **kwargs):
engine = get_engine(dburi, **kwargs)
if short_lived_sessions or dburi not in _SESSIONS:
_SESSIONS[dburi] = sessionmaker(bind=engine)
return engine, _SESSIONS[dburi]
def setup_results(engine):
if not _SETUP["results"]:
ResultModelBase.metadata.create_all(engine)
_SETUP["results"] = True
def ResultSession(dburi, **kwargs):
engine, session = create_session(dburi, **kwargs)
setup_results(engine)
return session()
|