ogn-python/ogn/commands/stats.py

44 wiersze
1.4 KiB
Python

from datetime import datetime
from tqdm import tqdm
from manager import Manager
from ogn.commands.dbutils import session
from ogn.commands.database import get_database_days
from ogn.collect.stats import create_device_stats, create_receiver_stats, create_relation_stats,\
update_qualities, update_receivers as update_receivers_command, update_devices as update_devices_command,\
update_device_stats_jumps
manager = Manager()
@manager.command
def create(start=None, end=None):
"""Create DeviceStats, ReceiverStats and RelationStats."""
days = get_database_days(start, end)
pbar = tqdm(days)
for single_date in pbar:
pbar.set_description(datetime.strftime(single_date, '%Y-%m-%d'))
result = create_device_stats(session=session, date=single_date)
result = update_device_stats_jumps(session=session, date=single_date)
result = create_receiver_stats(session=session, date=single_date)
result = create_relation_stats(session=session, date=single_date)
result = update_qualities(session=session, date=single_date)
@manager.command
def update_receivers():
"""Update receivers with data from stats."""
result = update_receivers_command(session=session)
print(result)
@manager.command
def update_devices():
"""Update devices with data from stats."""
result = update_devices_command(session=session)
print(result)