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.
62 lines
1.8 KiB
Python
62 lines
1.8 KiB
Python
"""
|
|
author: deadc0de6 (https://github.com/deadc0de6)
|
|
Copyright (c) 2022, deadc0de6
|
|
|
|
Class for printing nodes
|
|
"""
|
|
|
|
import sys
|
|
|
|
from catcli.colors import Colors
|
|
from catcli.utils import fix_badchars
|
|
|
|
|
|
class NodePrinter:
|
|
"""a node printer class"""
|
|
|
|
STORAGE = 'storage'
|
|
ARCHIVE = 'archive'
|
|
NBFILES = 'nbfiles'
|
|
|
|
@classmethod
|
|
def print_storage_native(cls, pre, name, args, attr):
|
|
"""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, pre, name, attr):
|
|
"""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, pre, name, depth='', attr=None):
|
|
"""print a directory node"""
|
|
end = []
|
|
if depth != '':
|
|
end.append(f'{cls.NBFILES}:{depth}')
|
|
if attr:
|
|
end.append(' '.join([f'{x}:{y}' for x, y in attr]))
|
|
if end:
|
|
endstring = ', '.join(end)
|
|
end = f' [{endstring}]'
|
|
out = pre + Colors.BLUE + fix_badchars(name) + Colors.RESET
|
|
out += f'{Colors.GRAY}{end}{Colors.RESET}'
|
|
sys.stdout.write(f'{out}\n')
|
|
|
|
@classmethod
|
|
def print_archive_native(cls, pre, name, archive):
|
|
"""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')
|