# ...PremiumMod/libs/database.py

# =============================================================================
# >> IMPORT
# =============================================================================
from PremiumMod.libs.constant import DATA_PATH
from cPickle import load, dump
from path import path

# =============================================================================
# >> LOAD & UNLOAD
# =============================================================================


class database(dict):
    def __init__(self):
        dict.__init__(self)
        self.path = DATA_PATH
        self.load()

    def load(self):
        self.clear()
        if self.path.isfile():
            v = open(self.path, 'rb')
            try:
                self.update(load(v))
            except:
                pass
            v.close()

    def save(self):
        if self.path.isfile():
            v = open(self.path, 'wb')
            dump(self, v, 'wb')
            v.close()

setup_dict = database()


'''
TypeError: argument must have 'write' attribute

'''