from __future__ import absolute_import, unicode_literals

import os
from datetime import timedelta
from celery import Celery
from celery.signals import worker_ready

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'src.main_config.settings.all')
# crontab(minute="*/1")
app = Celery('src')
app.config_from_object('django.conf:settings', namespace='CELERY')
app.autodiscover_tasks()

# define periodic tasks here

app.conf.beat_schedule = {
    'update_stats': {
        'task': 'src.apps.copo_core.tasks.update_stats',
        'schedule': timedelta(hours=24),
    },
    'poll_expired_viewlocks': {
        'task': 'src.apps.copo_core.tasks.poll_expired_viewlocks',
        'schedule': timedelta(seconds=60),
    },
    'process_housekeeping': {
        'task': 'src.apps.copo_core.tasks.process_housekeeping',
        'schedule': timedelta(days=1),
    },
    'update_ena_checklist': {
        'task': 'src.apps.copo_core.tasks.update_ena_checklist',
        'schedule': timedelta(days=1),
    },
    'update_ena_read_platform': {
        'task': 'src.apps.copo_core.tasks.update_ena_read_platform',
        'schedule': timedelta(days=1),
    },
    'update_ena_read_checklist': {
        'task': 'src.apps.copo_core.tasks.update_ena_read_checklist',
        'schedule': timedelta(days=1),
    },
    'process_tol_validations': {
        'task': 'src.apps.copo_dtol_upload.tasks.process_tol_validations',
        'schedule': timedelta(seconds=3),
    },
    'process_dtol_sample_submission': {
        'task': 'src.apps.copo_dtol_submission.tasks.process_dtol_sample_submission',
        'schedule': timedelta(seconds=10),
    },
    'process_stale_dtol_sample_submission': {
        'task': 'src.apps.copo_dtol_submission.tasks.process_stale_dtol_sample_submission',
        'schedule': timedelta(seconds=10),
    },
    'process_bioimage_submission': {
        'task': 'src.apps.copo_dtol_submission.tasks.process_bioimage_submission',
        'schedule': timedelta(seconds=30),
    },
    'process_bioimage_housekeeping': {
        'task': 'src.apps.copo_dtol_submission.tasks.process_bioimage_housekeeping',
        'schedule': timedelta(days=1)
    },    
    '''
    'find_incorrectly_rejected_samples': {
        'task': 'src.apps.copo_dtol_submission.tasks.find_incorrectly_rejected_samples',
        'schedule': timedelta(seconds=60),
    },
    '''
    'poll_missing_tolids': {
        'task': 'src.apps.copo_dtol_submission.tasks.poll_missing_tolids',
        'schedule': timedelta(hours=2),  # shortened cause sometimes it doesn't work?
    },
    'poll_asyn_ena_submission': {
        'task': 'src.apps.copo_dtol_submission.tasks.poll_asyn_ena_submission',
        'schedule': timedelta(seconds=30),
    },
    'process_ena_submission': {
        'task': 'src.apps.copo_read_submission.tasks.process_ena_submission',
        'schedule': timedelta(seconds=10),  # execute every n minutes minute="*/n"
    },
    'process_ena_transfers': {
        'task': 'src.apps.copo_read_submission.tasks.process_pending_file_transfers',
        'schedule': timedelta(seconds=10),
    },
    'check_for_stuck_transfers': {
        'task': 'src.apps.copo_read_submission.tasks.check_for_stuck_transfers',
        'schedule': timedelta(seconds=1800),
    },
    'update_assembly_submission_pending': {
        'task': 'src.apps.copo_assembly_submission.tasks.update_assembly_submission_pending',
        'schedule': timedelta(seconds=10),
    },
    'process_assembly_submission': {
        'task': 'src.apps.copo_assembly_submission.tasks.process_assembly_submission',
        'schedule': timedelta(seconds=10),
    },
    'process_seq_annotation_submission': {
        'task': 'src.apps.copo_seq_annotation_submission.tasks.process_seq_annotation_submission',
        'schedule': timedelta(seconds=10),
    },
    'poll_asyn_seq_annotation_submission_receipt': {
        'task': 'src.apps.copo_seq_annotation_submission.tasks.poll_asyn_seq_annotation_submission_receipt',
        'schedule': timedelta(seconds=10),
    },
    'update_seq_annotation_submission_pending': {
        'task': 'src.apps.copo_seq_annotation_submission.tasks.update_seq_annotation_submission_pending',
        'schedule': timedelta(seconds=10),
    },
    'processing_pending_tagged_seq_submission': {
        'task': 'src.apps.copo_barcoding_submission.tasks.processing_pending_tagged_seq_submission',
        'schedule': timedelta(seconds=10),
    },
    'send_fortnightly_pending_manifest_notification': {
        'task': 'src.apps.copo_dtol_submission.tasks.send_fortnightly_pending_manifest_notification',
        'schedule': timedelta(weeks=2)
    },
    'update_singlecell_schema': {
        'task': 'src.apps.copo_single_cell_submission.tasks.update_singlecell_schema',
        'schedule': timedelta(days=1)
    },     
    'process_zenodo_submission': {
        'task': 'src.apps.copo_single_cell_submission.tasks.process_zenodo_submission',
        'schedule': timedelta(seconds=10)
    },
    'process_ena_submission_singlecell': {
        'task': 'src.apps.copo_single_cell_submission.tasks.process_ena_submission',
        'schedule': timedelta(seconds=10)
    },
    'update_zenodo_submission_pending': {
        'task': 'src.apps.copo_single_cell_submission.tasks.update_zenodo_submission_pending',
        'schedule': timedelta(seconds=10)
    },

    'process_pending_sample_submission': {
        'task': 'src.apps.copo_sample.tasks.process_pending_sample_submission',
        'schedule': timedelta(seconds=10)
    },

    'poll_asyn_analysis_submission_receipt': {
        'task': 'src.apps.copo_single_cell_submission.tasks.process_asyn_analysis_submission_receipt',
        'schedule': timedelta(seconds=10)
    },
    'sync_system_records_with_ena': {
        'task': 'src.apps.copo_sample.tasks.sync_system_records_with_ena',
        'schedule': timedelta(weeks=1),
    },
    'process_expired_banners': {
        'task': 'src.apps.copo_core.tasks.process_expired_banners',
        'schedule': timedelta(minutes=5),
    },
}

app.conf.task_routes = {
    'src.apps.copo_read_submission.tasks.process_pending_file_transfers': {
        'queue': 'file_transfers'
    },
}


@app.task(bind=True)
def debug_task(self):
    print('Request: {0!r}'.format(self.request))


@worker_ready.connect
def trigger_startup_lookups(sender, **kwargs):
    """Refresh DB-backed remote lookups (ENA, single-cell schemas) when the worker boots.

    Beat schedules these daily, but a timedelta schedule's first run is a full
    interval after beat starts -- so a freshly deployed environment has no
    platforms/checklists/schemas for up to 24h. Enqueue them on worker_ready so the
    data is populated at T+0 instead. Dispatched async (.delay) so a slow fetch never
    blocks boot; the tasks are idempotent upserts so firing on every restart is safe.
    Imported inside the handler to avoid circular imports at module load.
    """
    from src.apps.copo_core.tasks import (
        update_ena_read_platform,
        update_ena_read_checklist,
        update_ena_checklist,
    )
    from src.apps.copo_single_cell_submission.tasks import update_singlecell_schema
    update_ena_read_platform.delay()
    update_ena_read_checklist.delay()
    update_ena_checklist.delay()
    update_singlecell_schema.delay()