kopia lustrzana https://github.com/glidernet/ogn-python
37 wiersze
730 B
Python
37 wiersze
730 B
Python
from __future__ import absolute_import
|
|
|
|
from celery.utils.log import get_task_logger
|
|
from ogn.collect.celery import app
|
|
|
|
from ogn.model import Device
|
|
from ogn.utils import get_ddb
|
|
|
|
logger = get_task_logger(__name__)
|
|
|
|
|
|
@app.task
|
|
def update_ddb_data():
|
|
logger.info("Update ddb data.")
|
|
|
|
app.session.query(Device).delete()
|
|
|
|
devices = get_ddb()
|
|
logger.debug("New Devices: %s" % str(devices))
|
|
|
|
app.session.bulk_save_objects(devices)
|
|
app.session.commit()
|
|
|
|
return len(devices)
|
|
|
|
|
|
@app.task
|
|
def import_ddb_data(filename='custom_ddb.txt'):
|
|
logger.info("Import ddb data from file.")
|
|
|
|
devices = get_ddb(filename)
|
|
|
|
app.session.bulk_save_objects(devices)
|
|
app.session.commit()
|
|
|
|
return len(devices)
|