mirror of https://github.com/deadc0de6/catcli
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
74 lines
2.3 KiB
Python
74 lines
2.3 KiB
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2022, deadc0de6
|
|
|
|
Class for printing nodes
|
|
"""
|
|
|
|
import sys
|
|
from typing import TypeVar, Type, Optional, Tuple, List, \
|
|
Dict, Any
|
|
|
|
from catcli.colors import Colors
|
|
from catcli.utils import fix_badchars
|
|
|
|
|
|
CLASSTYPE = TypeVar('CLASSTYPE', bound='NodePrinter')
|
|
|
|
|
|
class NodePrinter:
|
|
"""a node printer class"""
|
|
|
|
STORAGE = 'storage'
|
|
ARCHIVE = 'archive'
|
|
NBFILES = 'nbfiles'
|
|
|
|
@classmethod
|
|
def print_storage_native(cls: Type[CLASSTYPE], pre: str,
|
|
name: str, args: str,
|
|
attr: Dict[str, Any]) -> None:
|
|
"""print a storage node"""
|
|
end = ''
|
|
if attr:
|
|
end = f' {Colors.GRAY}({attr}){Colors.RESET}'
|
|
out = f'{pre}{Colors.UND}{cls.STORAGE}{Colors.RESET}:'
|
|
out += ' ' + Colors.PURPLE + fix_badchars(name) + \
|
|
Colors.RESET + end + '\n'
|
|
out += f' {Colors.GRAY}{args}{Colors.RESET}'
|
|
sys.stdout.write(f'{out}\n')
|
|
|
|
@classmethod
|
|
def print_file_native(cls: Type[CLASSTYPE], pre: str,
|
|
name: str, attr: str) -> None:
|
|
"""print a file node"""
|
|
nobad = fix_badchars(name)
|
|
out = f'{pre}{nobad}'
|
|
out += f' {Colors.GRAY}[{attr}]{Colors.RESET}'
|
|
sys.stdout.write(f'{out}\n')
|
|
|
|
@classmethod
|
|
def print_dir_native(cls: Type[CLASSTYPE], pre: str,
|
|
name: str,
|
|
nbchildren: int = 0,
|
|
attr: Optional[List[Tuple[str, str]]] = None) -> None:
|
|
"""print a directory node"""
|
|
end = []
|
|
if nbchildren > 0:
|
|
end.append(f'{cls.NBFILES}:{nbchildren}')
|
|
if attr:
|
|
end.append(' '.join([f'{x}:{y}' for x, y in attr]))
|
|
end_string = ''
|
|
if end:
|
|
end_string = f' [{", ".join(end)}]'
|
|
out = pre + Colors.BLUE + fix_badchars(name) + Colors.RESET
|
|
out += f'{Colors.GRAY}{end_string}{Colors.RESET}'
|
|
sys.stdout.write(f'{out}\n')
|
|
|
|
@classmethod
|
|
def print_archive_native(cls: Type[CLASSTYPE], pre: str,
|
|
name: str, archive: str) -> None:
|
|
"""archive to stdout"""
|
|
out = pre + Colors.YELLOW + fix_badchars(name) + Colors.RESET
|
|
out += f' {Colors.GRAY}[{cls.ARCHIVE}:{archive}]{Colors.RESET}'
|
|
sys.stdout.write(f'{out}\n')
|