def setUp(self): flask_app = create_flask(FLASK_TESTING_CONFIG, DATABASE_TESTING_CONFIG) self.flask_app = flask_app.test_client() self.db_engine = get_engine(DATABASE_TESTING_CONFIG.URI) self.db = get_session_from_engine(self.db_engine) create_base(Base, self.db_engine)
#!bin/python
"""Starts a flask app, loads initial directory info into the database, and starts a watchdog observer that will kick-off celery tasks in response to directoy
events."""
import os, time
from celery import states, Celery
from celery.exceptions import TimeoutError
from app.setup.database import get_engine, get_session_from_engine, create_base, initialize_base
from app.setup.config import DATABASE_CONFIG, FLASK_CONFIG
from app.setup.create_flask import create_flask
from app.tasks import celery, create_file, update_file, delete_file, discover_files, verify_files
from app.file_monitor import schedule_observer
from app.models import Archive, Base
from multiprocessing import Process
# DATABASE
db_engine = get_engine(DATABASE_CONFIG.URI)
db = get_session_from_engine(db_engine)
create_base(Base, db_engine)
if not os.path.exists(DATABASE_CONFIG.LOCATION):
initialize_base(db)
# INITIAL DIRECTORY DATA
def hard_check_files():
try:
discover_files(immediate=db)
verify_files().apply_async(timeout=10)
except TimeoutError:
pass
hard_check_files()