crafty-4/app/classes/minecraft/server_props.py

import pprint
import os


class ServerProps:
    def __init__(self, filepath):
        self.filepath = filepath
        self.props = self._parse()

    def _parse(self):
        # Loads and parses the file specified in self.filepath
        with open(self.filepath, encoding="utf-8") as fp:
            line = fp.readline()
            d = {}
            if os.path.exists(".header"):
                os.remove(".header")
            while line:
                if "#" != line[0]:
                    s = line
                    s1 = s[: s.find("=")]
                    if "\n" in s:
                        s2 = s[s.find("=") + 1 : s.find("\n")]
                    else:
                        s2 = s[s.find("=") + 1 :]
                    d[s1] = s2
                else:
                    with open(".header", "a+", encoding="utf-8") as h:
                        h.write(line)
                line = fp.readline()
        return d

    def print(self):
        # Prints the properties dictionary (using pprint)
        pprint.pprint(self.props)

    def get(self):
        # Returns the properties dictionary
        return self.props

    def update(self, key, val):
        # Updates property in the properties dictionary [ update("pvp", "true") ]
        # and returns boolean condition
        if key in self.props.keys():
            self.props[key] = val
            return True
        else:
            return False

    def save(self):
        # Writes to the new file
        with open(self.filepath, "a+", encoding="utf-8") as f:
            f.truncate(0)
            with open(".header", encoding="utf-8") as header:
                line = header.readline()
                while line:
                    f.write(line)
                    line = header.readline()
                header.close()
            for key, value in self.props.items():
                f.write(key + "=" + value + "\n")
        if os.path.exists(".header"):
            os.remove(".header")

    @staticmethod
    def cleanup():
        if os.path.exists(".header"):
            os.remove(".header")