pyson.py

"""
made by StroupBslayen: https://github.com/stroupbslayen/Pyson
"""

import os
import json


class Pyson:
    '''Allows for easier manipulation of json files.
    It will check if a json file already exists with the given file name and open that, otherwise it will create a new one.
    Default datatype is a DICT, but you can pass what you want to it. EX: example=Pyson(file_name,[]) would pass a list to the json.
    To update the data, modify the json using the "data" attribute. EX: example.data.append('test') would add 'test' to the list declared above.

    Commit changes with the "save" attribute. EX: example.save would save the changes to the json file'''

    def __init__(self, file_name, data = {}):
        if not file_name.endswith('.json'):
            file_name = file_name + '.json'
        if not os.path.isfile(file_name):
            pass
        else:
            try:
                with open(file_name) as f:
                    data = json.load(f)
            except ValueError:
                pass
        self.file_name = file_name
        self.data = data

    @property
    def save(self):
        '''Save your json file'''
        if not self.file_name.endswith('.json'):
            self.file_name = self.file_name + '.json'
        with open(self.file_name, "w") as f:
            json.dump(self.data, f, indent = 4, sort_keys = True)