from bootstrapvz.base.fs.volume import Volume


class Folder(Volume):

    # Override the states this volume can be in (i.e. we can't "format" or "attach" it)
    events = [{'name': 'create', 'src': 'nonexistent', 'dst': 'attached'},
              {'name': 'delete', 'src': 'attached', 'dst': 'deleted'},
              ]

    extension = 'chroot'

    def create(self, path):
        self.fsm.create(path=path)

    def _before_create(self, e):
        import os
        self.path = e.path
        os.mkdir(self.path)

    def _before_delete(self, e):
        from shutil import rmtree
        rmtree(self.path)
        del self.path