• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• Discord-Selfbot-master
  • run_linuxmac.sh
  • run.bat
  • Procfile
  • appuselfbot.py
  • anims
    • example.txt
  • cogs
    • debugger.py
    • replacements.py
    • misc.py
    • serverinfo.py
    • fc.py
    • utils
      • mal_char_find.py
      • menu.py
      • credit.txt
      • config.py
      • allmsgs.py
      • LICENSE
      • image_dump.py
      • webhooks.py
      • dataIO.py
      • common.py
      • notify.py
      • save
        • server_role_count.txt
        • mem_info.txt
        • same_discrim.txt
        • server_share.txt
        • stalker_percent.txt
        • server_share_long.txt
        • largest_server.txt
        • hidden_channels.txt
        • total_users.txt
      • checks.py
    • userinfo.py
    • imagedump.py
    • todo.py
    • customcmds.py
    • lockdown.py
    • cog_download.py
    • utility.py
    • fun.py
    • mod.py
    • track.py
    • google.py
    • help.py
    • translate.py
    • repl.py
    • keywordlog.py
    • emoji.py
  • LICENSE
  • loopself.py
  • issue_template.md
  • .travis.yml
  • README.md
  • requirements.txt
  • Dockerfile
  • settings
    • fc.json.sample
    • log.json.sample
    • config.json.sample
    • optional_config.json.sample
    • notify.json.sample
    • replacements.json.sample
    • ignored.json.sample
    • moderation.json.sample
    • github.json.sample
    • commands.json.sample
    • todo.json.sample
    • avatars.json.sample
  • .gitignore

from random import randint
from json import decoder, dump, load
from os import replace
from os.path import splitext

class DataIO():

    def save_json(self, filename, data):
        """Atomically save a JSON file given a filename and a dictionary."""
        path, ext = splitext(filename)
        tmp_file = "{}.{}.tmp".format(path, randint(1000, 9999))
        with open(tmp_file, 'w', encoding='utf-8') as f:
            dump(data, f, indent=4,sort_keys=True,separators=(',',' : '))
        try:
            with open(tmp_file, 'r', encoding='utf-8') as f:
                data = load(f)
        except decoder.JSONDecodeError:
            print("Attempted to write file {} but JSON "
                                  "integrity check on tmp file has failed. "
                                  "The original file is unaltered."
                                  "".format(filename))
            return False
        except Exception as e:
            print('A issue has occured saving ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return False

        replace(tmp_file, filename)
        return True

    def load_json(self, filename):
        """Load a JSON file and return a dictionary."""
        try:
            with open(filename, 'r', encoding='utf-8') as f:
                data = load(f)
            return data
        except Exception as e:
            print('A issue has occured loading ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return {}

    def append_json(self, filename, data):
        """Append a value to a JSON file."""
        try:
            with open(filename, 'r', encoding='utf-8') as f:
                file = load(f)
        except Exception as e:
            print('A issue has occured loading ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return False
        try:
            file.append(data)
        except Exception as e:
            print('A issue has occured updating ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return False
        path, ext = splitext(filename)
        tmp_file = "{}.{}.tmp".format(path, randint(1000, 9999))
        with open(tmp_file, 'w', encoding='utf-8') as f:
            dump(file, f, indent=4,sort_keys=True,separators=(',',' : '))
        try:
            with open(tmp_file, 'r', encoding='utf-8') as f:
                data = load(f)
        except decoder.JSONDecodeError:
            print("Attempted to write file {} but JSON "
                                  "integrity check on tmp file has failed. "
                                  "The original file is unaltered."
                                  "".format(filename))
            return False
        except Exception as e:
            print('A issue has occured saving ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return False

        replace(tmp_file, filename)
        return True

    def is_valid_json(self, filename):
        """Verify that a JSON file exists and is readable. Take in a filename and return a boolean."""
        try:
            with open(filename, 'r', encoding='utf-8') as f:
                data = load(f)
            return True
        except (FileNotFoundError, decoder.JSONDecodeError):
            return False
        except Exception as e:
            print('A issue has occured validating ' + filename + '.\n'
                  'Traceback:\n'
                  '{0} {1}'.format(str(e), e.args))
            return False

dataIO = DataIO()