fileutils-dir 0.1.2__py3-none-any.whl → 0.1.3__py3-none-any.whl

fileutils/files.py

@@ -1,52 +1,33 @@
-from pathlib import Path
-from typing import Iterable
 from collections.abc import Iterable
+from pathlib import Path
 
 FILE_TYPES = {
-    "image": [".jpg", ".jpeg", ".png", ".webp", ".bmp", ".gif", ".tiff"],
-    "text": [".txt", ".md", ".rst", ".log"],
-    "pdf": [".pdf"],
-    "doc": [".doc", ".docx", ".odt"],
-    "sheet": [".xls", ".xlsx", ".ods", ".csv"],
-    "presentation": [".ppt", ".pptx", ".odp"],
-    "code": [
-        ".py", ".js", ".ts", ".java", ".c", ".cpp", ".h",
-        ".go", ".rs", ".rb", ".php", ".sh"
-    ],
-    "data": [".json", ".yaml", ".yml", ".xml", ".toml"],
-    "audio": [".mp3", ".wav", ".flac", ".ogg", ".aac", ".m4a"],
-    "video": [".mp4", ".mkv", ".avi", ".mov", ".webm"],
-    "archive": [".zip", ".tar", ".gz", ".bz2", ".7z", ".rar"],
+              "image": [".jpg", ".jpeg", ".png", ".webp", ".bmp", ".gif", ".tiff"],
+              "text": [".txt", ".md", ".rst", ".log"], "pdf": [".pdf"], "doc": [".doc", ".docx", ".odt"],
+              "sheet": [".xls", ".xlsx", ".ods", ".csv"], "presentation": [".ppt", ".pptx", ".odp"],
+              "code": [".py", ".js", ".ts", ".java", ".c", ".cpp", ".h", ".go", ".rs", ".rb", ".php", ".sh"],
+              "data": [".json", ".yaml", ".yml", ".xml", ".toml"],
+              "audio": [".mp3", ".wav", ".flac", ".ogg", ".aac", ".m4a"],
+              "video": [".mp4", ".mkv", ".avi", ".mov", ".webm"],
+              "archive": [".zip", ".tar", ".gz", ".bz2", ".7z", ".rar"],
 }
 
 
 def in_dir(
-    directories: Iterable[str] | None = None,
-    extensions: Iterable[str] | None = None,
-    types: Iterable[str] | None = None,
+        paths: Iterable[str] | None = None,
+        ext: Iterable[str] | None = None,
+        dtype: Iterable[str] | None = None,
 ) -> list[str]:
-    """
-    Return files from the given directories:\n.
-    - If 'directories' are not given, only the PWD is considered.
-    - If `extensions` or `types` are provided, only matching files are returned.
-    - If neither is provided, all files in the PWD are returned.
-    """
-    if isinstance(directories, str):
-        types = [directories]
-        directories = None
-    if directories is None:
-        directories = ["."]
-    if extensions is None:
-        extensions = []
-    if types is None:
-        types = []
+    paths = paths or ["."]
+    ext = ext or []
+    dtype = dtype or []
 
     normalized_extensions = {
-        ext.lower() if ext.startswith(".") else f".{ext.lower()}"
-        for ext in extensions
+        e.lower() if e.startswith(".") else f".{e.lower()}"
+        for e in ext
     }
 
-    for file_type in types:
+    for file_type in dtype:
         key = file_type.strip().lower()
         try:
             normalized_extensions.update(FILE_TYPES[key])
@@ -55,18 +36,12 @@ def in_dir(
 
     files_to_return: list[str] = []
 
-    for path in directories:
+    for path in paths:
         for item in Path(path).iterdir():
             if not item.is_file():
                 continue
 
-            # If no filters, accept everything
-            if not normalized_extensions:
-                files_to_return.append(str(item))
-                continue
-
-            # Otherwise, filter by extension
-            if item.suffix.lower() in normalized_extensions:
+            if not normalized_extensions or item.suffix.lower() in normalized_extensions:
                 files_to_return.append(str(item))
 
     return files_to_return

{fileutils_dir-0.1.2.dist-info → fileutils_dir-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: fileutils-dir
-Version: 0.1.2
+Version: 0.1.3
 Summary: Small utilities for listing files in directories
 Author: Jatavallabhula Sarat Anirudh
 License-Expression: MIT

fileutils_dir-0.1.3.dist-info/RECORD

@@ -0,0 +1,6 @@
+fileutils/__init__.py,sha256=3u-pHKVodcCPECMcj95lrkk8nr7tl6_HRqiQOw6YbBM,78
+fileutils/files.py,sha256=pSHHNR8oZyCtSi7ZL73wIqywyntwe7mKVogaAdhLh_8,1718
+fileutils_dir-0.1.3.dist-info/METADATA,sha256=f4HpV1wJjFH_7pcDiCqP0Yd8Mo6ArSNKvXLWLJg4v8M,1239
+fileutils_dir-0.1.3.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+fileutils_dir-0.1.3.dist-info/top_level.txt,sha256=kZPCAZpYo7OkmwY_R2a4ifuMINdW5AZd2JaNHDSXQYQ,10
+fileutils_dir-0.1.3.dist-info/RECORD,,

fileutils_dir-0.1.2.dist-info/RECORD

@@ -1,6 +0,0 @@
-fileutils/__init__.py,sha256=3u-pHKVodcCPECMcj95lrkk8nr7tl6_HRqiQOw6YbBM,78
-fileutils/files.py,sha256=j4knqqqwS8iOywxapZjJHx8kQee1SB0u6dzrVYdIhIA,2375
-fileutils_dir-0.1.2.dist-info/METADATA,sha256=_yHXInJ_ccb18DhVFmBjVp20UezeofZy5KqVhgwWAyQ,1239
-fileutils_dir-0.1.2.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
-fileutils_dir-0.1.2.dist-info/top_level.txt,sha256=kZPCAZpYo7OkmwY_R2a4ifuMINdW5AZd2JaNHDSXQYQ,10
-fileutils_dir-0.1.2.dist-info/RECORD,,