10

from zipfile import ZipFile

    12

# PSK Management Data model :

    13

# [[ID,Desc, LastKnownURI, LastConnect]]

    22

def _default(ID):

    23

    return [ID,

    24

            '', # default description

    28

def _dataByID(data):

    29

    return {row[COL_ID]:row for row in data}

    30

    32

    """ load known keys metadata """

    35

        if os.path.exists(_path):

    36

            return json.loads(open(_path).read())

    37

    return []

    39

def _filterData(psk_files, data_input):

    40

    input_by_ID = _dataByID(data_input)

    41

    output = []

    49

           output.append(input_by_ID.get(ID,_default(ID)))

    50

    return output

    52

def GetData(project_path):

    53

    loaded_data = _LoadData(project_path)

    54

    psk_files = os.listdir(_pskpath(project_path))

    55

    return _filterData(psk_files, loaded_data)

    61

def SaveData(project_path, data):

    78

    # here we directly use _LoadData, avoiding filtering that could be long

    80

    idata = _dataByID(data)

    81

    dataForID = idata.get(ID, _default(ID)) if data else _default(ID)

    86

    SaveData(project_path, data)

    87

    88

def ExportIDs(project_path, export_zip):

    89

    with ZipFile(export_zip, 'w') as zf:

    90

        path = _pskpath(project_path)

    91

        for nm in os.listdir(path):

    92

            if nm.endswith('.secret') or nm == 'management.json':

    93

                zf.write(os.path.join(path, nm), nm)

    94

    95

def ImportIDs(project_path, import_zip, should_I_replace_callback):

    96

    data = GetData(project_path)

    97

    98

    zip_loaded_data = json.loads(zf.open('management.json').read())

    99

    name_list = zf.namelist()

   100

    zip_filtered_data = _filterData(name_list, loaded_data)

   101

   102

    idata = _dataByID(data)

   103

   104

    for imported_row in zip_filtered_data:

   105

        ID = imported_row[COL_ID]

   106

        existing_row = idata.get(ID, None)

   107

        if existing_row is None:

   108

            data.append(imported_row)

   109

        else:

   110

            # callback returns the selected list for merge or none if canceled

   111

            result = should_I_replace_callback(existing_row, imported_row)

   112

            if result is None:

   113

                break

   114

   115

            if result: 

   116

                # replace with imported

   117

                existing_row[:] = imported_row

   118

                # copy the key of selected

   119

                self.extract(ID+".secret", _pskpath(project_path))

   120

   121

    SaveData(project_path, data)

   122

   123