pferd/PFERD/utils.py

import os
import sys
import pathlib
from colorama import Style
from colorama import Fore

__all__ = [
    "get_base_dir",
    "move",
    "rename",
    "stream_to_path",
    "ContentTypeException",
    "FileNotFoundException",
    "PrettyLogger",
]

def get_base_dir(script_file):
    return pathlib.Path(os.path.dirname(os.path.abspath(script_file)))

def move(path, from_folders, to_folders):
    l = len(from_folders)
    if path.parts[:l] == from_folders:
        return pathlib.PurePath(*to_folders, *path.parts[l:])

def rename(path, to_name):
    return pathlib.PurePath(*path.parts[:-1], to_name)

def stream_to_path(response, to_path, chunk_size=1024**2):
    with open(to_path, 'wb') as fd:
        for chunk in response.iter_content(chunk_size=chunk_size):
            fd.write(chunk)

def isOutputPipe():
    """Returns whether this program's output is attached to a pipe.
    """
    return sys.stdout.isatty

class ContentTypeException(Exception):
    pass

class FileNotFoundException(Exception):
    pass

class PrettyLogger:

    def __init__(self, logger):
        self.logger = logger

    def modified_file(self, file_name):
        self.logger.info(f"{Fore.MAGENTA}{Style.BRIGHT}Modified {file_name}.{Style.RESET_ALL}")

    def new_file(self, file_name):
        self.logger.info(f"{Fore.GREEN}{Style.BRIGHT}Created {file_name}.{Style.RESET_ALL}")

    def ignored_file(self, file_name):
        self.logger.info(f"{Style.DIM}Ignored {file_name}.{Style.RESET_ALL}")

    def starting_synchronizer(self, target_directory, synchronizer_name, subject=None):
        subject_str = f"{subject} " if subject else ""
        self.logger.info("")
        self.logger.info((
            f"{Fore.CYAN}{Style.BRIGHT}Synchronizing {subject_str}to {target_directory}"
            f" using the {synchronizer_name} synchronizer.{Style.RESET_ALL}"
        ))