|
@@ -11,7 +11,6 @@ from tqdm import tqdm
|
|
|
|
|
|
from pycs import app
|
|
|
from pycs.database.File import File
|
|
|
-from pycs.database.Project import Project
|
|
|
|
|
|
DEFAULT_JPEG_QUALITY = 80
|
|
|
|
|
@@ -252,7 +251,7 @@ def find_images(folder,
|
|
|
""" walk recursively the folder and find images """
|
|
|
|
|
|
suffixes = suffixes if suffixes is not None else [".jpg", ".jpeg", ".png"]
|
|
|
- images: T.List[Path] = list()
|
|
|
+ images: T.List[Path] = []
|
|
|
for root, _, files in os.walk(folder):
|
|
|
for file in files:
|
|
|
fpath = Path(root, file)
|
|
@@ -264,7 +263,12 @@ def find_images(folder,
|
|
|
return images
|
|
|
|
|
|
|
|
|
-def generate_thumbnails(project: Project, sizes = [Size(200, 200), Size(2000, 1200)]):
|
|
|
+def generate_thumbnails(project: "Project", sizes = None):
|
|
|
+ """ generates thumbnails for all image files in the given """
|
|
|
+
|
|
|
+ if sizes is None:
|
|
|
+ sizes = [Size(200, 200), Size(2000, 1200)]
|
|
|
+
|
|
|
app.logger.info(f"Generating thumbnails for project \"{project.name}\"")
|
|
|
|
|
|
files = list(project.files)
|