diff --git a/code/get_behaviour.py b/code/get_behaviour.py index 74dcc9d..36311ca 100644 --- a/code/get_behaviour.py +++ b/code/get_behaviour.py @@ -2,6 +2,7 @@ import os from paramiko import SSHClient from scp import SCPClient from IPython import embed +from pandas import read_csv ssh = SSHClient() ssh.load_system_host_keys() @@ -10,22 +11,25 @@ ssh.connect(hostname='kraken', username='efish', password='fwNix4U', ) -embed() + # SCPCLient takes a paramiko transport as its only argument scp = SCPClient(ssh.get_transport()) -foldername = '2020-03-16-10_00' +data = read_csv('../recs.csv') +foldernames = data['recording'].values + directory = f'/Users/acfw/Documents/uni_tuebingen/chirpdetection/GP2023_chirp_detection/data/mount_data/' +for foldername in foldernames: -if not os.path.exists(directory+foldername): - os.makedirs(directory+foldername) + if not os.path.exists(directory+foldername): + os.makedirs(directory+foldername) -files = [('-').join(foldername.split('-')[:3])+'.csv','chirp_ids.npy', 'chirps.npy', 'fund_v.npy', 'ident_v.npy', 'idx_v.npy', 'times.npy', 'spec.npy', 'LED_on_time.npy', 'sign_v.npy'] + files = [('-').join(foldername.split('-')[:3])+'.csv','chirp_ids.npy', 'chirps.npy', 'fund_v.npy', 'ident_v.npy', 'idx_v.npy', 'times.npy', 'spec.npy', 'LED_on_time.npy', 'sign_v.npy'] -for f in files: - scp.get(f'/home/efish/behavior/2019_tube_competition/{foldername}/{f}', - directory+foldername) + for f in files: + scp.get(f'/home/efish/behavior/2019_tube_competition/{foldername}/{f}', + directory+foldername) scp.close()