1

#!/usr/bin/env python

     2

# -*- coding: utf-8 -*-

     3

     4

# See COPYING file for copyrights details.

     5

     6

from __future__ import absolute_import

     7

import os

     8

import time

     9

import json

    10

from zipfile import ZipFile

    11

    12

# PSK Management Data model :

    13

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

    16

    17

def _pskpath(project_path):

    18

    return os.path.join(project_path, 'psk')

    19

    20

def _mgtpath(project_path):

    21

    return os.path.join(_pskpath(project_path), 'management.json')

    22

    23

def _default(ID):

    24

    return [ID,

    25

            '', # default description

    26

            None, # last known URI

    27

            None]  # last connection date

    28

    29

def _dataByID(data):

    30

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

    31

    32

def _LoadData(project_path):

    33

    """ load known keys metadata """

    34

    if os.path.isdir(_pskpath(project_path)):

    35

        _path = _mgtpath(project_path)

    36

        if os.path.exists(_path):

    37

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

    38

    return []

    39

    40

def _filterData(psk_files, data_input):

    41

    input_by_ID = _dataByID(data_input)

    42

    output = []

    43

    # go through all secret files available an build data

    44

    # out of data recoverd from json and list of secret.

    45

    # this implicitly filters IDs out of metadata who's

    46

    # secret is missing

    47

    for filename in psk_files:

    48

       if filename.endswith('.secret'):

    49

           ID = filename[:-7]  # strip filename extension

    50

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

    51

    return output

    52

    53

def GetData(project_path):

    54

    loaded_data = _LoadData(project_path)

    55

    psk_files = os.listdir(_pskpath(project_path))

    56

    return _filterData(psk_files, loaded_data)

    57

    58

def DeleteID(project_path, ID):

    59

    secret_path = os.path.join(_pskpath(project_path), ID+'.secret')

    60

    os.remove(secret_path)

    61

    62

def SaveData(project_path, data):

    63

    pskpath = _pskpath(project_path)

    64

    if not os.path.isdir(pskpath):

    65

        os.mkdir(pskpath)

    66

    with open(_mgtpath(project_path), 'w') as f:

    67

        f.write(json.dumps(data))

    68

    69

    70

def UpdateID(project_path, ID, secret, URI):

    71

    pskpath = _pskpath(project_path)

    72

    if not os.path.exists(pskpath):

    73

        os.mkdir(pskpath)

    74

    75

    secpath = os.path.join(pskpath, ID+'.secret')

    76

    with open(secpath, 'w') as f:

    77

        f.write(ID+":"+secret)

    78

    79

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

    80

    data = _LoadData(project_path)

    81

    idata = _dataByID(data)

    82

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

    83

    dataForID[COL_URI] = URI

    84

    # FIXME : could store time instead os a string and use DVC model's cmp 

    85

    # then date display could be smarter, etc - sortable sting hack for now

    86

    dataForID[COL_LAST] = time.strftime('%y/%M/%d-%H:%M:%S')

    87

    SaveData(project_path, data)

    88

    89

def ExportIDs(project_path, export_zip):

    90

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

    91

        path = _pskpath(project_path)

    92

        for nm in os.listdir(path):

    93

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

    94

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

    95

    96

def ImportIDs(project_path, import_zip, should_I_replace_callback):

    98

    data = GetData(project_path)

    99

   100

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

   101

    name_list = zf.namelist()

   103

   104

    idata = _dataByID(data)

   105

   109

    for imported_row in zip_filtered_data:

   110

        ID = imported_row[COL_ID]

   111

        existing_row = idata.get(ID, None)

   112

        if existing_row is None:

   113

            data.append(imported_row)

   114

        else:

   115

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

   121

   123

                # replace with imported

   124

                existing_row[:] = imported_row

   125

                # copy the key of selected

   130

   131

    SaveData(project_path, data)

   132

   134