planner/webservice/file_io.py

import os
from pathlib import Path
from flask import Blueprint, Response, request
from flask_cors import CORS
from datetime import datetime
import json
import hashlib
import time
from webservice.auth import Auth
from webservice.planner_load import PlannerLoad
from webservice.config_load import ConfigLoad

bp = Blueprint('file_io', __name__)
# cors = CORS(app, resources={r"/*": {"origins": "http://localhost:4200/"}})
CORS(bp)

script_dir = Path(__file__).parent
save_dir = script_dir.parent.joinpath('save')
planner_dir = script_dir.parent.joinpath('export')
config_dir = script_dir.parent.joinpath('config')
# save_dir = Path('C:\\Projekte\\Python\\Planung\\save')
user_info = None
timestamp_keywords = ('accounts', 'plan', 'marketing')


@bp.route('/login/', methods=['POST'])
def login():
    user = request.get_json()['data']['user']
    password = request.get_json()['data']['password']
    user_info = Auth().get_user(user, password)

    return Response(
        response=json.dumps(user_info),
        mimetype='application/json'
    )


@bp.route('/list', methods=['GET'])
def list_json():
    return json.dumps(list_dict(), indent=2)


def list_dict():
    result = {'list': [], 'tree': {}}
    for currentFile in save_dir.iterdir():
        if currentFile.is_file() and currentFile.name[-5:] == '.json':
            year, version, timestamp = currentFile.name[:-5].split("_")
            if year not in result['tree']:
                result['tree'][year] = {}
            if version not in result['tree'][year]:
                result['tree'][year][version] = []
            result['tree'][year][version].append(timestamp)
            result['list'].append({'year': year, 'version': version, 'timestamp': timestamp})
            result['list'].sort(key=lambda x: x['timestamp'])
    return result


@bp.route('/load/<year>/<version>/<timestamp>', methods=['GET'])
def load(year, version, timestamp):
    file = full_filename(year, version, timestamp)

    if timestamp == 'new':
        return Response(
            response=new_file(year),
            mimetype='application/json'
        )

    if timestamp == 'current' or not Path(file).exists():
        file_list = list_dict()
        timestamp2 = file_list['tree'][year][version][-1]
        file = full_filename(year, version, timestamp2)

    print(file)

    with open(file, 'r') as frh:
        structure = json.loads(frh.read())
        if 'options' not in structure[0]:
            p_load = PlannerLoad(planner_dir)
            structure = p_load.convert_file(structure)
        elif timestamp in timestamp_keywords:
            p_load = PlannerLoad(planner_dir)
            p_load.load_values(year)
            p_load.set_structure(structure)
            structure = p_load.update_values(timestamp)

        return Response(
            response=json.dumps(structure),
            mimetype='application/json'
        )


def full_filename(year, version, timestamp):
    return f'{str(save_dir)}/{year}_{version}_{timestamp}.json'


@bp.route('/new/<year>', methods=['GET'])
def new_file(year):
    p_load = PlannerLoad(planner_dir)
    structure = p_load.new_file(year)
    return json.dumps(structure)


@bp.route('/load/<filename>', methods=['GET'])
def load_file(filename):
    full_filename = f'{str(save_dir)}/{filename}.json'
    if not Path(full_filename).exists():
        file_list = list_dict()
        timestamp = file_list[filename][-1]
        full_filename = f'{full_filename[:-5]}_{timestamp}.json'

    print(full_filename)
    with open(full_filename, 'r') as frh:
        return frh.read()


@bp.route('/save/<year>/<version>', methods=['POST'])
def save_version(year, version):
    return save(year + '_' + version)


@bp.route('/save/<filename>', methods=['POST'])
def save(filename):
    if request.method != 'POST':
        return 'File is missing!'

    new_filename = str(save_dir) + '/' + filename + '_' + datetime.now().strftime('%Y%m%d%H%M%S') + '.json'
    data = request.get_json()['data']
    with open(new_filename, 'w') as fwh:
        json.dump(data, fwh, indent=2)
    print(new_filename)

    try:
        file_list = list_dict()
        timestamp = file_list[filename][-2]
        old_filename = str(save_dir) + '/' + filename + "_" + timestamp + '.json'

        if old_filename != new_filename and hash(old_filename) == hash(new_filename):
            Path(old_filename).unlink()
            return 'File saved with no changes!'
    except KeyError:
        pass
    return 'File saved with new data!'


@bp.route('/config', methods=['GET'])
def config():
    cfg = ConfigLoad(str(config_dir))
    return Response(
        response=json.dumps(cfg.load_file('reisacher', '2023')),
        mimetype='application/json'
    )


@bp.route('/accounts/<period>', methods=['GET'])
def accounts(period):
    with open(planner_dir.joinpath(f'accounts_{period}.json'), 'r') as frh:
        return frh.read()


def hash(filename):
    BUF_SIZE = 65536
    sha1 = hashlib.sha1()
    with open(filename, 'rb') as g:
        while True:
            hashdata = g.read(BUF_SIZE)
            if not hashdata:
                break
            sha1.update(hashdata)
    return sha1.hexdigest()