import os 
from paramiko import SSHClient
from scp import SCPClient
from IPython import embed

ssh = SSHClient()
ssh.load_system_host_keys()

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'
directory = f'/Users/acfw/Documents/uni_tuebingen/chirpdetection/GP2023_chirp_detection/data/mount_data/'

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']


for f in files:
    scp.get(f'/home/efish/behavior/2019_tube_competition/{foldername}/{f}',
            directory+foldername)

scp.close()