import json
import os

patients_path = '/PHShome/ju601/crt/data/json/'
#patients_path = '/PHShome/ju601/crt/data/MGHpallcare/json/'

# Deprecated
#loads old annonomyzed data from JSON file into nested dictionary
def get_old_data():
    with open('data/anon_data_OLD.json', 'r') as f:
        data = json.load(f)
    return data

# Gets a specific patient by their EMPI
def get_patient_by_EMPI(empi):
    file_path = patients_path + empi + '.json'
    if os.path.isfile(file_path):
        with open(file_path, 'r') as f:
            patient = json.load(f)
        return patient
    else:
        raise ValueError("Patient with EMPI: " + empi + " does not exist")

def save(patient):
    file_path = patients_path + patient['EMPI'] + '.json'
    if os.path.isfile(file_path):
        with open(file_path, 'w') as f:
            json.dump(patient, f)
        return patient
    else:
        raise ValueError("Patient with EMPI: " + empi + " does not exist")

def get_data(patient_range=range(0,10)):
    if len(patient_range) > 50:
        raise ValueError("You've attempted to load too many patients into memory at once.")

    files = os.listdir(patients_path)
    # Sort by number order
    # Changed for pal care
    #files = sorted(files, key=lambda x: int(x.split('.json')[0].split('_')[2]))
    files = sorted(files, key=lambda x: int(x.split('.json')[0]))

    # Get all patients in patient range
    patients = []
    for i in patient_range:
        if i < len(files):
            file_path = patients_path + files[i]
            with open(file_path, 'r') as f:
                try:
                    patient = json.load(f)
                    patients.append(patient)
                except Exception as e:
                    print "Error loading file: " + file_path
                    raise e
    return patients

def get_dummy_non_anonymized_patient():
    return {'First_Name':'Josh', 'Last_Name':'Haimson', 'EMPI':'1234emPI', 'NEW_EMPI':'FAKE_EMPI_1', 'MRNS':['mrn12','mrn34']}

#Removes empty fields from data
#Recursive, modifies object, returns nothing
def clean_data(data):
    if type(data) == type(dict()):
        for k in data.keys():
            clean_data(data[k])
            if data[k] in [u'', [], "", dict(), None]:
                data.pop(k)
    elif type(data) == type([]):
        for i in range(len(data)):
            clean_data(data[i])    

#little function to help with exploring the data from commandline
def explore(data):
    if type(data) == type(dict()) or (type(data) ==  type([]) and len(data)>1):
        again = True
        
        while again:
            if type(data) == type(dict()):    
                keys = data.keys()
                for i in range(len(keys)):
                    size = ""
                    if type(data[keys[i]]) == type([]):
                        if len(data[keys[i]]) > 1:
                            size = " (" + str(len(data[keys[i]])) + ")"
                    print str(i+1) + ".\t", keys[i], size
                inp = unicode("-1")
                while not(inp == ""  or  inp in keys or (inp.isnumeric() and ( int(inp) > 0 and int(inp) <= len(keys)))):
                    inp = unicode(raw_input("Select a key: "))
                print ""
                if inp != "" and inp.isnumeric():
                    print keys[int(inp)-1], ": "
                    explore(data[keys[int(inp)-1]])
                elif inp == "":
                    again = False
                else:
                    print inp, ": "
                    explore(data[inp])
            else:
                inp = unicode("-1")
                while not(inp == "" or (inp.isnumeric() and int(inp) > 0 and int(inp) <= len(data))):
                    inp = unicode(raw_input("Select a file from 1-" + str(len(data)) +": "))
                if inp != "":
                    explore(data[int(inp)-1])
                else:
                    again = False 
    elif type(data) == type([]):
        explore(data[0])  
    else:
        print data.strip("\n")
        print
        inp = raw_input("[ Press enter to continue ]")
        print