yacloud/shadowmire
1
0
Fork 0
mirror of https://github.com/taoky/shadowmire.git synced 2025-07-08 01:02:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace glob by fast_iterdir in verification step 4

Browse Source 
To avoid stat() every files in package/.
Related: #4
This commit is contained in:
taoky 2024-09-03 22:47:43 +08:00
parent 6e4aa61119
commit a5dbd61e95
1 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
shadowmire.py
View File

@ -2,7 +2,7 @@
import sys
from types import FrameType
from typing import IO, Any, Callable, Generator, Literal, NoReturn, Optional
from typing import IO, Any, Callable, Generator, Literal, NoReturn, Optional, Set
import xmlrpc.client
from dataclasses import dataclass
import re
@ -192,7 +192,7 @@ def remove_dir_with_files(directory: Path) -> None:
def fast_iterdir(
directory: Path, filter_type: Literal["dir", "file"]
directory: Path | str, filter_type: Literal["dir", "file"]
) -> Generator[os.DirEntry[str], Any, None]:
"""
iterdir() in pathlib would ignore file type information from getdents64(),
@ -1217,7 +1217,7 @@ def verify(
logger.info(
"====== Step 4. Remove any unreferenced files in `packages` folder ======"
)
ref_set = set()
ref_set: Set[str] = set()
with ThreadPoolExecutor(max_workers=IOWORKERS) as executor:
# Part 1: iterate simple/
def iterate_simple(sname: str) -> list[str]:
@ -1258,15 +1258,15 @@ def verify(
# Part 2: iterate packages
def unlink_not_in_set(first_dirname: str, position: int) -> None:
for file in tqdm(
(basedir / "packages" / first_dirname).glob("*/*/*"),
desc=f"Iterating packages/{first_dirname}/*/*/*",
position=position,
):
logger.debug("find file %s", file)
if str(file) not in ref_set:
logger.info("removing unreferenced file %s", file)
file.unlink()
with tqdm(desc=f"Iterating packages/{first_dirname}/*/*/*", position=position) as pb:
for d1 in fast_iterdir(basedir / "packages" / first_dirname, "dir"):
for d2 in fast_iterdir(d1.path, "dir"):
for file in fast_iterdir(d2.path, "file"):
pb.update(1)
logger.debug("find file %s", file)
if file.path not in ref_set:
logger.info("removing unreferenced file %s", file.path)
Path(file.path).unlink()
# MyPy does not enjoy same variable name with different types, even when --allow-redefinition
# Ignore here to make mypy happy