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()

ssh.connect(
    hostname="kraken",
    username="efish",
    password="fwNix4U",
)


# SCPCLient takes a paramiko transport as its only argument
scp = SCPClient(ssh.get_transport())

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)

    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()