A generic class for manage JSON files with python.

jsonDbConnection.py

# coding = utf-8
# using namespace std
import json
from typing import AnyStr

class JSONConnection(object):
    """
    """
    json_file = AnyStr
    got_json = False
    document = dict()
    
    class InvalidFileExt(Exception):
        """ If the selected file is not a JSON file. """
       
    class DatabaseAlreadyLoaded(Exception):
        """ If the database already haves a JSON file connected """
       
    class DatabaseNotLoaded(Exception):
        """ If the class don't have a JSON file connected """
       
    @staticmethod
    def check_file_ext(file_nm: AnyStr) -> bool:
        """
        """
        return str(file_nm).split(".")[-1] == "json"
    
    def connect_file(self, file_json: AnyStr):
        """
        """
        if self.got_json: raise self.DatabaseAlreadyLoaded("The class already have a JSON loaded")
        if not self.check_file_ext(file_json): raise self.InvalidFileExt(f"The file {file_json} is not a JSON file")
        with open(file_json, "r") as json_to:
            self.document = json.loads(json_to.read())
            self.json_file =  file_json
            self.got_json = True
    
    def disconnect_file(self):
        """
        """
        if not self.got_json: raise self.DatabaseNotLoaded("There's no  connection already")
        self.json_file = AnyStr
        self.document = dict()
        self.got_json = False
    
    def commit_changes(self):
        """
        """
        if not self.got_json: raise self.DatabaseNotLoaded("There's no connection with a file")
        dumped = json.dumps(self.document)
        with open(self.json_file, "w+") as json_file: json_file.write(dumped)
        del dumped
    
    def get_doc(self) -> dict:
        """
        """
        return self.document
    
    def get_json(self) -> AnyStr:
        """
        """
        return self.json_file
    
    def get_json_name(self) -> str:
        """
        """
        return str(self.json_file)
    
    def get_got_connection(self) -> bool:
        """
        """
        return self.got_json