mirror of https://github.com/deadc0de6/catcli
refactoring
deadc0de6
4 months ago
parent
691396c96a
commit
0dcbfa94bd
@ -1,73 +0,0 @@
|
||||
"""
|
||||
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')
|
||||
@ -0,0 +1,84 @@
|
||||
"""
|
||||
author: deadc0de6 (https://github.com/deadc0de6)
|
||||
Copyright (c) 2024, deadc0de6
|
||||
|
||||
Class for printing nodes in csv format
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
from typing import List
|
||||
|
||||
from catcli.nodes import NodeAny, NodeStorage, TYPE_DIR
|
||||
from catcli.utils import size_to_str, epoch_to_str, \
|
||||
has_attr
|
||||
|
||||
|
||||
class CsvPrinter:
|
||||
"""a node printer class"""
|
||||
|
||||
DEFSEP = ','
|
||||
CSV_HEADER = ('name,type,path,size,indexed_at,'
|
||||
'maccess,md5,nbfiles,free_space,'
|
||||
'total_space,meta')
|
||||
|
||||
def _print_entries(self, entries: List[str], sep: str = DEFSEP) -> None:
|
||||
line = sep.join(['"' + o + '"' for o in entries])
|
||||
if len(line) > 0:
|
||||
sys.stdout.write(f'{line}\n')
|
||||
|
||||
def print_header(self) -> None:
|
||||
"""print csv header"""
|
||||
sys.stdout.write(f'{self.CSV_HEADER}\n')
|
||||
|
||||
def print_storage(self, node: NodeStorage,
|
||||
sep: str = DEFSEP,
|
||||
raw: bool = False) -> None:
|
||||
"""print a storage node"""
|
||||
out = []
|
||||
out.append(node.name) # name
|
||||
out.append(node.type) # type
|
||||
out.append('') # fake full path
|
||||
size = node.get_rec_size()
|
||||
out.append(size_to_str(size, raw=raw)) # size
|
||||
out.append(epoch_to_str(node.ts)) # indexed_at
|
||||
out.append('') # fake maccess
|
||||
out.append('') # fake md5
|
||||
out.append(str(len(node.children))) # nbfiles
|
||||
# fake free_space
|
||||
out.append(size_to_str(node.free, raw=raw))
|
||||
# fake total_space
|
||||
out.append(size_to_str(node.total, raw=raw))
|
||||
out.append(node.attr) # meta
|
||||
self._print_entries(out, sep=sep)
|
||||
|
||||
def print_node(self, node: NodeAny,
|
||||
sep: str = DEFSEP,
|
||||
raw: bool = False) -> None:
|
||||
"""print other nodes"""
|
||||
out = []
|
||||
out.append(node.name.replace('"', '""')) # name
|
||||
out.append(node.type) # type
|
||||
parents = node.get_parent_hierarchy()
|
||||
storage = node.get_storage_node()
|
||||
fullpath = os.path.join(storage.name, parents)
|
||||
out.append(fullpath.replace('"', '""')) # full path
|
||||
|
||||
out.append(size_to_str(node.nodesize, raw=raw)) # size
|
||||
out.append(epoch_to_str(storage.ts)) # indexed_at
|
||||
if has_attr(node, 'maccess'):
|
||||
out.append(epoch_to_str(node.maccess)) # maccess
|
||||
else:
|
||||
out.append('') # fake maccess
|
||||
if has_attr(node, 'md5'):
|
||||
out.append(node.md5) # md5
|
||||
else:
|
||||
out.append('') # fake md5
|
||||
if node.type == TYPE_DIR:
|
||||
out.append(str(len(node.children))) # nbfiles
|
||||
else:
|
||||
out.append('') # fake nbfiles
|
||||
out.append('') # fake free_space
|
||||
out.append('') # fake total_space
|
||||
out.append('') # fake meta
|
||||
self._print_entries(out, sep=sep)
|
||||
@ -0,0 +1,151 @@
|
||||
"""
|
||||
author: deadc0de6 (https://github.com/deadc0de6)
|
||||
Copyright (c) 2022, deadc0de6
|
||||
|
||||
Class for printing nodes in native format
|
||||
"""
|
||||
|
||||
import sys
|
||||
import os
|
||||
|
||||
from catcli.nodes import NodeFile, NodeDir, NodeStorage
|
||||
from catcli.colors import Colors
|
||||
from catcli.logger import Logger
|
||||
from catcli.utils import fix_badchars, size_to_str, \
|
||||
has_attr, epoch_to_ls_str
|
||||
|
||||
|
||||
TS_LJUST = 13
|
||||
SIZE_LJUST = 6
|
||||
NAME_LJUST = 20
|
||||
|
||||
class NativePrinter:
|
||||
"""a node printer class"""
|
||||
|
||||
STORAGE = 'storage'
|
||||
ARCHIVE = 'archive'
|
||||
NBFILES = 'nbfiles'
|
||||
|
||||
def print_top(self, pre: str, name: str) -> None:
|
||||
"""print top node"""
|
||||
sys.stdout.write(f'{pre}{name}\n')
|
||||
|
||||
def print_storage(self, pre: str,
|
||||
node: NodeStorage,
|
||||
raw: bool = False) -> None:
|
||||
"""print a storage node"""
|
||||
# construct name
|
||||
name = node.name
|
||||
name = fix_badchars(name)
|
||||
# construct attrs
|
||||
attrs = []
|
||||
# nb files
|
||||
attrs.append(f'nbfiles:{len(node.children)}')
|
||||
# the children size
|
||||
recsize = node.get_rec_size()
|
||||
sizestr = size_to_str(recsize, raw=raw)
|
||||
attrs.append(f'totsize:{sizestr}')
|
||||
# free
|
||||
pcent = 0.0
|
||||
if node.total > 0:
|
||||
pcent = node.free * 100 / node.total
|
||||
attrs.append(f'free:{pcent:.1f}%')
|
||||
# du
|
||||
sztotal = size_to_str(node.total, raw=raw)
|
||||
szused = size_to_str(node.total - node.free, raw=raw)
|
||||
attrs.append(f'du:{szused}/{sztotal}')
|
||||
# timestamp
|
||||
if has_attr(node, 'ts'):
|
||||
attrs.append(f'date:{epoch_to_ls_str(node.ts)}')
|
||||
|
||||
# print
|
||||
out = f'{pre}{Colors.UND}{self.STORAGE}{Colors.RESET}: '
|
||||
out += f'{Colors.PURPLE}{name}{Colors.RESET}'
|
||||
if attrs:
|
||||
out += f'\n{" "*len(name)}{Colors.GRAY}{"|".join(attrs)}{Colors.RESET}'
|
||||
sys.stdout.write(f'{out}\n')
|
||||
|
||||
def print_file(self, pre: str,
|
||||
node: NodeFile,
|
||||
withpath: bool = False,
|
||||
withstorage: bool = False,
|
||||
raw: bool = False) -> None:
|
||||
"""print a file node"""
|
||||
# construct name
|
||||
name = node.name
|
||||
storage = node.get_storage_node()
|
||||
if withpath:
|
||||
name = os.sep.join([
|
||||
storage.name,
|
||||
node.parent.get_parent_hierarchy(),
|
||||
name])
|
||||
name = fix_badchars(name)
|
||||
# construct attributes
|
||||
attrs = []
|
||||
if node.md5:
|
||||
attrs.append(f'md5:{node.md5}')
|
||||
if withstorage:
|
||||
content = Logger.get_bold_text(storage.name)
|
||||
attrs.append(f', storage:{content}')
|
||||
# print
|
||||
out = []
|
||||
out .append(f'{pre}')
|
||||
line = name.ljust(NAME_LJUST, ' ')
|
||||
out.append(f'{line}')
|
||||
size = 0
|
||||
if node.nodesize:
|
||||
size = node.nodesize
|
||||
line = size_to_str(size, raw=raw).ljust(SIZE_LJUST, ' ')
|
||||
out.append(f'{Colors.BLUE}{line}{Colors.RESET}')
|
||||
line = epoch_to_ls_str(node.maccess).ljust(TS_LJUST, ' ')
|
||||
out.append(f'{Colors.PURPLE}{line}{Colors.RESET}')
|
||||
if attrs:
|
||||
out.append(f'{Colors.GRAY}[{",".join(attrs)}]{Colors.RESET}')
|
||||
sys.stdout.write(f'{" ".join(out)}\n')
|
||||
|
||||
def print_dir(self, pre: str,
|
||||
node: NodeDir,
|
||||
withpath: bool = False,
|
||||
withstorage: bool = False,
|
||||
withnbchildren: bool = False,
|
||||
raw: bool = False) -> None:
|
||||
"""print a directory node"""
|
||||
# construct name
|
||||
name = node.name
|
||||
storage = node.get_storage_node()
|
||||
if withpath:
|
||||
name = os.sep.join([
|
||||
storage.name,
|
||||
node.parent.get_parent_hierarchy(),
|
||||
name])
|
||||
name = fix_badchars(name)
|
||||
# construct attrs
|
||||
attrs = []
|
||||
if withnbchildren:
|
||||
nbchildren = len(node.children)
|
||||
attrs.append(f'{self.NBFILES}:{nbchildren}')
|
||||
if withstorage:
|
||||
attrs.append(f'storage:{Logger.get_bold_text(storage.name)}')
|
||||
# print
|
||||
out = []
|
||||
out.append(f'{pre}')
|
||||
line = name.ljust(NAME_LJUST, ' ')
|
||||
out.append(f'{Colors.BLUE}{line}{Colors.RESET}')
|
||||
size = 0
|
||||
if node.nodesize:
|
||||
size = node.nodesize
|
||||
line = size_to_str(size, raw=raw).ljust(SIZE_LJUST, ' ')
|
||||
out.append(f'{Colors.GRAY}{line}{Colors.RESET}')
|
||||
line = epoch_to_ls_str(node.maccess).ljust(TS_LJUST, ' ')
|
||||
out.append(f'{Colors.GRAY}{line}{Colors.RESET}')
|
||||
if attrs:
|
||||
out.append(f'{Colors.GRAY}[{",".join(attrs)}]{Colors.RESET}')
|
||||
sys.stdout.write(f'{" ".join(out)}\n')
|
||||
|
||||
def print_archive(self, pre: str,
|
||||
name: str, archive: str) -> None:
|
||||
"""print an archive"""
|
||||
name = fix_badchars(name)
|
||||
out = f'{pre}{Colors.YELLOW}{name}{Colors.RESET} '
|
||||
out += f'{Colors.GRAY}[{self.ARCHIVE}:{archive}]{Colors.RESET}'
|
||||
sys.stdout.write(f'{out}\n')
|
||||
Loading…
Reference in New Issue