stammdaten-formular/edit_data_form.py

#!/usr/bin/env python
# encoding: utf-8

import npyscreen
import json
import datetime
from sys import argv
import pathlib

from base_data_model import base_data_model
from financial_data_model import financial_data_model
from membership_data_model import membership_data_model


class FormApp(npyscreen.NPSAppManaged):
    def __init__(self, filename):
        super().__init__()

        self.dir_path = pathlib.Path(filename).absolute().parent
        self.member = json.load(open(filename, 'r'))

        self.base_data = base_data_model(self.member['stammdaten'])
        self.financial_data = financial_data_model(self.member['finanzdaten']) 
        self.membership_data = membership_data_model(self.member['mitgliederdaten'])
    
    def main(self):
        f = npyscreen.Form(name='Edit Member data')        

        self.base_data.add_edit_fields(f)
        self.financial_data.add_edit_fields(f)
        self.membership_data.add_edit_fields(f)

        f.edit()

    def get_data_from_form(self):
        data_dict = {}
        data_dict['stammdaten'] = self.base_data.get_form_values_dict()
        data_dict['finanzdaten'] = self.financial_data.get_form_values_dict()
        data_dict['mitgliederdaten'] = self.membership_data.get_form_values_dict()
        data_dict['timestamp'] = datetime.datetime.now().replace(microsecond=0).isoformat()
        return data_dict

    def data_has_changed(self):
        return self.base_data.has_changed() | self.financial_data.has_changed() | self.membership_data.has_changed()

    def get_filepath(self):
        name = self.base_data.get_form_values_dict()['fullname']
        date = datetime.datetime.now().replace(microsecond=0).isoformat()
        fn = '{}_{}.json'.format('_'.join(name.lower().split()), date.replace(':',''))
        file_path = pathlib.Path(self.dir_path, fn)
        return file_path


def main(*_args):
    if len(argv) > 1:
        filename = argv[1]
    else:
        filename = pathlib.Path(pathlib.Path(__file__).absolute().parent, 'example.json')

    app = FormApp(filename)

    try:
        app.run()
    except Exception as e:
        print(e)
        return 0    
    
    if app.data_has_changed():
        out_data = app.get_data_from_form()
        outfile_path = app.get_filepath()
        print(f"Writing changed Member Data to {outfile_path}")
        with open(outfile_path, 'w') as outfile:
            json.dump(out_data, outfile, sort_keys=True, indent=4)
    else:
        print("Nothing has changed, skipping writing file")
    

if __name__ == "__main__":
    npyscreen.wrapper_basic(main)