worker.py

import logging
from lib.app import app
from lib import worker
from lib import ml
from lib.db import DB
from time import sleep
from lib.camera import CameraDownError, NightTimeError

if __name__ == "__main__":
    with app.app_context():
        db = DB.get_db()
        db.setup()
        try:
            logging.info("training model")
            # Always try train on first boot.
            ml.train(force=True)
            logging.info("training complete")
        except ml.NoTraingDataError as e:
            logging.warning(e)
        except Exception:
            logging.exception("Unexpected error training model")

        while True:
            try:
                worker.run()
            except NightTimeError as e:
                logging.warning(e)
                logging.info("training model")
                try:
                    ml.train()
                except (ml.TrainingIntervalError, ml.NoTraingDataError) as e:
                    logging.warning(e)
                except Exception:
                    logging.exception("Unexpected error training model")

            except CameraDownError as e:
                logging.warning(e)
            except Exception:
                logging.exception("Unexpected error running worker")

            sleep(app.config["INTERVAL_CAMERA"])