sessionIterator.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jun  1 10:13:38 2021

@author: Gia.Jordan
to be ran in the data server ICR folder to iterate thorugh all rats/sessions
"""
import os
#move to repo
os.chdir(r'G:\GitHub\Rodent-Turns')
from processVideo import lightCorr
import pandas as pd

sessList=pd.read_excel(r'Rotation_Sessions.xlsx',engine='openpyxl')
roi=list(sessList.Rat)
soi=list(sessList.File)

#move to ICR directory
os.chdir(r'\\DATA-SERVER\ICR_Behavior\BehaviorPilot')
#os.chdir(r'Z:\BehaviorPilot')
#save directory name
dataFolder=os.getcwd()
#get list of rats in directory
rats=os.listdir()



#iterate through rats
for rat in rats:
    
    if 'r'+rat+"'" not in roi:
        continue
    
    #display rat to be analyzed
    print(os.path.join(dataFolder,rat))
    
    #move to rat's directory
    os.chdir(os.path.join(dataFolder,rat))
    #load sessions for that rat
    sessions=os.listdir()
    
    #iterate thorugh sesions and files
    for session in sessions:
        if session+"'" not in soi:
            continue
        
        """
        Stopped at rat 0140 session ending with 9-34-16
        """
        
        
        #if the 'file' is a valid ICR session, move to directory and process/analyze video
        if os.path.isdir(os.path.join(dataFolder,rat,session)):
            os.chdir(os.path.join(dataFolder,rat,session))
            
            print(os.getcwd(),'\n')
            lightCorr()