sync_yadisk.py

import logging
import sys
from argparse import ArgumentParser

from m4m_sync import serializers

from database import DatabaseManager
from m4m_sync.encrypt import AesStreamWrapper
from m4m_sync.stores import YaDiskStore, Sensor, Controller

logging.basicConfig(
    stream=sys.stdout,
    level=logging.INFO,
    format="%(asctime)s [%(levelname)s] [%(name)s] %(message)s",
)

logger = logging.getLogger(__name__)


def main():
    parser = ArgumentParser()
    parser.add_argument("--db-uri", required=True)
    parser.add_argument("--serializer", default="CsvRawSerializer")

    args = parser.parse_args()

    logger.info("init")

    db = DatabaseManager(args.db_uri)
    store = YaDiskStore(token=db.get_tokens().yandex_disk)

    for controller in db.get_controllers():
        c = Controller(name=controller.name, mac=controller.mac)
        store.prepare_for_sync_controller(c)

        for sensor in db.get_sensors(controller):
            first_date = db.get_first_sensor_data_date(sensor.id)

            s = Sensor(name=sensor.name, id=sensor.id, controller=c)
            store.prepare_for_sync_sensor(s)
            store.sync(
                sensor=s,
                serializer=getattr(serializers, args.serializer)(),
                stream_wrapper=AesStreamWrapper(key=db.get_encryption_key()),
                first_date=first_date,
                get_data=lambda time_range: db.get_sensor_data(sensor.id, time_range),
            )

    logger.info("done")


if __name__ == "__main__":
    main()