diskset.py

import os

from collections.abc import MutableSet


class DiskSet(MutableSet):
    def __init__(self, path, isRelative=False):
        if isRelative is True:
            self.path = path
        else:
            self.path = os.path.abspath(os.path.normpath(os.path.expanduser(path)))
        if not os.path.exists(self.path):
            os.makedirs(self.path)

    def __contains__(self, name):
        return os.path.exists(os.path.join(self.path, name))

    def __iter__(self):
        return iter(os.listdir(self.path))

    def __len__(self):
        return len(os.listdir(self.path))

    def add(self, name):
        with open(os.path.join(self.path, name), 'w') as f:
            f.write('')

    def discard(self, name):
        try:
            os.unlink(os.path.join(self.path, name))
        except IOError:
            pass