Esempio n. 1
0
    def setRootFolder(self, path):
        ft = FileTrie()
        for top, directories, files in os.walk(path):
            for f in files:
                ft.add(os.path.join(top, f))

        success = False
        for r in self.races:
            m = ft.best_match(r.fileName)
            if m:
                r.fileName = m
                success = True
                self.setChanged()

        return success
Esempio n. 2
0
    def __init__(self):
        self.redirected = not os.isatty(sys.stdin.fileno())
        self.root = FileTrie("", True, None)
        self.current_directory = self.root

        if not os.path.isdir(Constants.DIRECTORY_NAME):
            os.makedirs(Constants.DIRECTORY_NAME)
        else:
            self._build_trie()
Esempio n. 3
0
class FileSystem(object):
    def __init__(self):
        self.redirected = not os.isatty(sys.stdin.fileno())
        self.root = FileTrie("", True, None)
        self.current_directory = self.root

        if not os.path.isdir(Constants.DIRECTORY_NAME):
            os.makedirs(Constants.DIRECTORY_NAME)
        else:
            self._build_trie()

    def _build_trie(self):
        for f in os.listdir(Constants.DIRECTORY_NAME):
            self.root.add_file(f[1:], False)

    def execute(self):
        '''Starts this file system operating'''
        line = self._prompt()
        while line != Constants.QUIT:
            command = line.split()[0]
            line = line[len(command):].lstrip()  # Command is the first option

            if command == Constants.PWD:
                self._pwd()

            elif command == Constants.CD:
                self._cd(line)

            elif command == Constants.LS:
                self._ls(line)

            elif command == Constants.RLS:
                self._rls()

            elif command == Constants.TREE:
                self._tree(line)

            elif command == Constants.CLEAR:
                self._clear()

            elif command == Constants.CREATE:
                self._create(line)

            elif command == Constants.ADD:
                self._add(line)

            elif command == Constants.CAT:
                self._cat(line)

            elif command == Constants.DELETE:
                self._delete(line)

            elif command == Constants.DELETE_DIR:
                self._delete_directory(line)

            line = self._prompt()

    def _prompt(self):
        '''Prompts the user for input and returns the user input'''
        if self.redirected:
            user_input = input()
            print(Constants.PROMPT, user_input)
        else:
            print(Constants.PROMPT, end=" ")
            user_input = input()
        return user_input

    def _pwd(self):
        print(self.current_directory.get_full_name())

    def _cd(self, line):
        if len(line) == 0:
            self.current_directory = self.root
        elif line == Constants.PARENT_DIR:
            self.current_directory = self.current_directory.get_parent()
        elif line[0] == Constants.ROOT:
            self.current_directory = self.root.get_directory(line[1:])
        else:
            self.current_directory = self.current_directory.get_directory(line)

    def _ls(self, line):
        if len(line) == 0:
            self.current_directory.list_files()
        elif line[0] == Constants.ROOT:
            directory = self.root.get_directory(line[1:])
            directory.list_files()
        else:
            directory = self.current_directory.get_directory(line)
            directory.list_files()

    def _rls(self):
        subprocess.call(["ls", "-l", Constants.DIRECTORY_NAME])

    def _tree(self, line):
        if len(line) == 0:
            self.current_directory.tree()
        elif line[0] == Constants.ROOT:
            directory = self.root.get_directory(line[1:])
            directory.tree()
        else:
            directory = self.current_directory.get_directory(line)
            directory.tree()

    def _clear(self):
        self.root.delete()

    def _create(self, line):
        if line[0] == Constants.ROOT:
            self.root.add_file(line[1:])
        else:
            self.current_directory.add_file(line)

    def _add(self, line):
        if line[0] == Constants.ROOT:
            index_of_space = line.find(' ')
            f = self.root.get_file(line[1:index_of_space])
            f.append(line[index_of_space + 1:])
        else:
            index_of_space = line.find(' ')
            f = self.root.get_file(line[:index_of_space])
            f.append(line[index_of_space + 1:])

    def _cat(self, line):
        if line[0] == Constants.ROOT:
            f = self.root.get_file(line[1:])
            f.cat()
        else:
            f = self.current_directory.get_file(line)
            f.cat()

    def _delete(self, line):
        if line[0] == Constants.ROOT:
            f = self.root.get_file(line[1:])
            f.delete()
        else:
            f = self.current_directory.get_file(line)
            f.delete()

    def _delete_directory(self, line):
        if line[0] == Constants.ROOT:
            directory = self.root.get_directory(line[1:])
            directory.delete()
        else:
            directory = self.current_directory.get_directory(line)
            directory.delete()