directory.py

import os
import shutil
import time


class Folder:
    def __init__(self):
        self.os_type = os.name                          # Get type of os
        self.root = os.getcwd() + os.sep + "asset"      # Location of root folder
        self.current = self.root                        # Pointer to current folder
        if not os.path.exists(self.root):               # If root does not exit
            os.mkdir(self.root)                         # Create asset as root

    # Trimming current path name & returning it
    def curr(self):
        return self.current.replace(self.root, os.sep)

    # Making directory
    def make_dir(self, name):
        dir_path = self.current + os.sep + name
        if os.path.exists(dir_path):
            print(f"{name} Directory already exit...")
        else:
            try:
                os.mkdir(dir_path)
                print(f"Directory {name} is successfully created...")
            except OSError:
                print("Invalid path...")

    # Removing directory
    def remove_dir(self, name):
        dir_path = self.current + os.sep + name
        # If directory with name exit
        if os.path.exists(dir_path) and os.path.isdir(dir_path):
            if (name != "..") and (name != "."):
                shutil.rmtree(dir_path)
                print(f"Directory {name} is successfully removed...")
        else:
            print("Invalid path.......")

    # Changing current directory
    def change_dir(self, name):
        temp = self.current     # Making a temporary copy
        if os.path.exists(self.current + os.sep + name) and os.path.isdir(self.current + os.sep + name):
            if os.sep in name:     # If path contain multilevel of directories
                name = name.split(os.sep)  # Getting individual folder name
                if "" in name:          # Remove "" after last / in present
                    name.remove("")
                for i in name:          # Moving in and out of directory
                    if (i != "..") and (i != "."):
                        temp += os.sep + i
                    elif i == "..":
                        temp = temp[:temp.rfind(os.sep)]
            else:
                if name == "..":    # exiting current directory
                    temp = temp[:temp.rfind(os.sep)]
                elif name != ".":   # . is same as current level
                    temp += os.sep + name
            if self.root in temp:   # If you are inside root directory
                self.current = temp
                return self.curr()
        return "Invalid path...."

    # List content of current directory
    def list_folder(self, show):
        # for showing hidden file
        if show == 0:
            lis = [i for i in os.listdir(self.current) if i[0] != '.']
            return lis
        # For showing details
        elif show == 2:
            lis = os.listdir(self.current)
            for i in range(len(lis)):
                lis[i] = self.current + os.sep + lis[i]
                s = str(os.path.getsize(lis[i]))
                s += " "*(8-len(s))
                t = time.ctime(os.path.getctime(lis[i]))[4:]
                lis[i] = s + t + "   " + lis[i].replace(self.root, os.sep)
            return lis
        # for not showing hidden file
        else:
            return os.listdir(self.current)

    # Moving directory or file to other
    def move(self, orignal, target):
        temp = orignal     # making copy of orignal
        orignal = self.current + os.sep + orignal
        target = self.current + os.sep + target
        # Can't make change with current or parent file
        if temp != "./" and temp != "../":
            if os.path.exists(orignal) or os.path.isfile(orignal):
                if os.path.exists(target):
                    # Can't go outside root directory
                    if (self.root in orignal) and (self.root in target):
                        # if you want to move file, append filename in target
                        if os.path.isfile(orignal):
                            target += os.sep + orignal[orignal.rfind(os.sep)+1:]
                        target = r'{}'.format(target)       # Encoding
                        orignal = r'{}'.format(orignal)     # Encoding
                        try:
                            shutil.move(orignal, target)
                        except IsADirectoryError:
                            return "Can't move a directory to file....."
                        except shutil.Error:
                            return f"File or directory is already exist...."
                        return "Move successful........."
        return "Invalid path......."

    # To copy one directory to other
    def copy(self, orignal, target):
        temp = orignal     # making copy of orignal
        orignal = self.current + os.sep + orignal
        target = self.current + os.sep + target
        # Can't make change with current or parent file
        if temp != "./" and temp != "../":
            if os.path.exists(orignal) or os.path.isfile(orignal):
                if os.path.exists(target):
                    # Can't go outside root directory
                    if (self.root in orignal) and (self.root in target):
                        orignal = r'{}'.format(orignal)     # Encoding
                        target = r'{}'.format(target)       # Encoding
                        try:
                            if os.path.isdir(self.current + os.sep + temp):  # IF src is directory
                                shutil.copytree(orignal, os.path.join(target, temp))
                            else:                                            # IF src is file
                                shutil.copy(orignal, target)
                        except IsADirectoryError:
                            return "Can't copy a directory in a file"
                        except FileExistsError:
                            return "File or directory already exist......"
                        return "Copy successfully......."
        return "Invalid path...."

    # Renaming file and directory
    def rename(self, original, target):
        if os.sep not in target:
            original = self.current + os.sep + original
            target = self.current + os.sep + target
            if os.path.exists(original) or os.path.isfile(original):
                if self.root in original:
                    os.rename(original, target)
                    return "Rename successful....."
        return "Invalid path......"

    # To clear screen
    def clear(self):
        os.system("clear" if self.os_type == "posix" else "cls")