ingest_into_archive_pyks_results_Sep2022.py

from pipeline import lab, experiment, ephys, shell
from pipeline.ingest import ephys as ephys_ingest


shell.logsetup('DEBUG')

subjects_to_archive = ("SC011", "SC013", "SC015", "SC016", "SC017",
                       "SC022", "SC023", "SC026", "SC027", "SC030",
                       "SC031", "SC032", "SC033", "SC035", "SC038",
                       "SC043", "SC045", "SC048", "SC049", "SC050",
                       "SC052", "SC053", "SC060", "SC061", "SC064",
                       "SC065", "SC066", "SC067")


def main(limit=None):
    subject_keys = (lab.Subject * lab.WaterRestriction.proj('water_restriction_number')
                    & f'water_restriction_number in {subjects_to_archive}').fetch('KEY')

    sessions = (experiment.Session
                - (ephys.ArchivedClustering & {'clustering_method': 'pykilosort2.5'})
                & subject_keys)

    for session_key in sessions.fetch('KEY', limit=limit):
        print(f'\n----------- Archiving for {session_key} ------------')
        try:
            ephys_ingest.do_ephys_ingest(session_key, replace=False,
                                         probe_insertion_exists=True, into_archive=True)
        except Exception as e:
            print(f'\tError... Skipped\n{str(e)}')


if __name__ == '__main__':
    main()