mirrors/ArchiveBox
1
0
Fork 0
mirror of https://github.com/ArchiveBox/ArchiveBox.git synced 2025-05-13 22:54:27 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

allow filtering for specific files when getting dir size

Browse source
This commit is contained in:
Nick Sweeting 2019-04-24 11:39:34 -04:00
parent 168d253c17
commit ae782a1a0c
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
archivebox/legacy/util.py
View file

@ -4,6 +4,7 @@ import sys
import json import json
import time import time
import shutil import shutil
import argparse
from string import Template from string import Template
from json import JSONEncoder from json import JSONEncoder
@ -563,9 +564,11 @@ def copy_and_overwrite(from_path: str, to_path: str):
@enforce_types @enforce_types
def get_dir_size(path: str, recursive: bool=True) -> Tuple[int, int, int]: def get_dir_size(path: str, recursive: bool=True, pattern: Optional[str]=None) -> Tuple[int, int, int]:
num_bytes, num_dirs, num_files = 0, 0, 0 num_bytes, num_dirs, num_files = 0, 0, 0
for entry in os.scandir(path): for entry in os.scandir(path):
if (pattern is not None) and (pattern not in entry.path):
continue
if entry.is_dir(follow_symlinks=False): if entry.is_dir(follow_symlinks=False):
if not recursive: if not recursive:
continue continue
@ -649,7 +652,7 @@ class ExtendedEncoder(JSONEncoder):
return JSONEncoder.default(self, obj) return JSONEncoder.default(self, obj)
def to_json(obj: Any, file: IO=None, indent: Optional[int]=4, sort_keys: bool=True, cls=ExtendedEncoder) -> Optional[str]: def to_json(obj: Any, file: IO=None, indent: Optional[int]=4, sort_keys: bool=True, cls=ExtendedEncoder) -> str:
if file: if file:
path = os.path.realpath(file.name) path = os.path.realpath(file.name)
contents = json.dumps(obj, indent=indent, sort_keys=sort_keys, cls=ExtendedEncoder) contents = json.dumps(obj, indent=indent, sort_keys=sort_keys, cls=ExtendedEncoder)
@ -729,3 +732,8 @@ def reject_stdin(caller: str) -> None:
raise SystemExit(1) raise SystemExit(1)
class SmartFormatter(argparse.HelpFormatter):
def _split_lines(self, text, width):
if '\n' in text:
return text.splitlines()
return argparse.HelpFormatter._split_lines(self, text, width)