kopia lustrzana https://github.com/glidernet/ogn-python
35 wiersze
921 B
Python
35 wiersze
921 B
Python
import os
|
|
import importlib
|
|
from sqlalchemy import create_engine
|
|
from sqlalchemy.orm import sessionmaker
|
|
|
|
from celery import Celery
|
|
from celery.signals import worker_init, worker_shutdown
|
|
|
|
os.environ.setdefault('OGN_CONFIG_MODULE', 'config.default')
|
|
config = importlib.import_module(os.environ['OGN_CONFIG_MODULE'])
|
|
|
|
|
|
@worker_init.connect
|
|
def connect_db(signal, sender):
|
|
# Load settings like DB_URI...
|
|
engine = create_engine(config.SQLALCHEMY_DATABASE_URI, echo=False)
|
|
|
|
Session = sessionmaker(bind=engine)
|
|
sender.app.session = Session()
|
|
|
|
|
|
@worker_shutdown.connect
|
|
def close_db(signal, sender):
|
|
sender.app.session.close()
|
|
|
|
|
|
app = Celery('ogn.collect',
|
|
include=["ogn.collect.database",
|
|
"ogn.collect.heatmap",
|
|
"ogn.collect.logbook",
|
|
"ogn.collect.receiver"
|
|
])
|
|
|
|
app.config_from_envvar("OGN_CONFIG_MODULE")
|