filesystem.py

# -*- coding: utf-8 -*-
from genericpath import isfile
import os
from os.path import join, basename, splitext, isdir


class Node(object):
    def __init__(self, root, path):
        self.path = path
        self.root = root
        self._basename = basename(self.path)

    def __unicode__(self):
        return self.name

class File(Node):

    def __unicode__(self):
        return self.name

    @property
    def extension(self):
        return splitext(self._basename)[1]

    @property
    def name(self):
        return self._basename


class Folder(Node):
    
    def __init__(self, root, path):
        super(Folder, self).__init__(root, path)
        self.files = []
        self.folders = []

    @property
    def name(self):
        return basename(self.path)

    def chunks(self):
        chunk_path = ''
        for chunk in self.path.split(os.sep):
            chunk_path = join(chunk_path, chunk)
            yield {'chunk': chunk, 'path': chunk_path}


    def read(self):
        for node in os.listdir(join(self.root, self.path)):
            full_path = join(self.path, node)
            if isdir(join(self.root, full_path)):
                self.folders.append(Folder(self.root, full_path))
            if isfile(join(self.root, full_path)):
                self.files.append(File(self.root, full_path))