Release version 0.2.2: Fix root-level file detection in directory scan

Author: rezamardaniDev

codetree/cli.py

@@ -21,51 +21,33 @@ def count_lines_of_code(file_path):
             lines = file.readlines()
         return len(lines)
     except Exception as e:
-        # If there is an error reading the file, print the error and skip the file
         print(f"Error reading {file_path}: {e}")
         return 0
 
 def print_directory_tree(path):
     """Prints the directory structure with files and their line counts."""
     folder_files = defaultdict(list)
-    folder_count = 0  # Initialize folder count
-    checked_files = 0  # Initialize checked files count
-
-    # Process the root directory separately to include files in the root
-    if not os.path.basename(path).startswith('.'):  # Skip hidden files and folders
-        folder_name = os.path.basename(path)
-        folder_count += 1  # Increment folder count
-        for file in sorted(os.listdir(path)):
-            file_path = os.path.join(path, file)
-            if os.path.isfile(file_path) and not file.startswith('.'):
-                lines = count_lines_of_code(file_path)
-                if lines > 0:
-                    folder_files[folder_name].append((file, lines))
-                    checked_files += 1  # Increment checked files count
+    folder_count = 0
+    checked_files = 0
 
     # Walk through the directory and get files with line counts
     for root, dirs, files in os.walk(path):
-        # Skip any directories that start with a dot
+        # Skip hidden directories
         dirs[:] = [d for d in dirs if not d.startswith('.')]
 
-        # Only process the current directory if it does not start with a dot
         rel_root = os.path.relpath(root, path)
-        if rel_root.startswith('.'):
-            continue
-        
-        folder_name = rel_root if rel_root != '.' else os.path.basename(path)
-        folder_count += 1  # Increment folder count
+        folder_name = rel_root if rel_root != '.' else 'root'  # Show root separately
+        folder_count += 1
 
         for file in sorted(files):
-            # Skip files that start with a dot
             if file.startswith('.'):
                 continue
 
             file_path = os.path.join(root, file)
             lines = count_lines_of_code(file_path)
             if lines > 0:
                 folder_files[folder_name].append((file, lines))
-                checked_files += 1  # Increment checked files count
+                checked_files += 1
 
     # Print directory tree with line counts
     for folder in sorted(folder_files.keys()):
@@ -79,29 +61,21 @@ def print_directory_tree(path):
 def scan_directory(path):
     """Scan the directory and prints summary of files and line counts."""
     print(f"\nScanning directory: {path}")
-    total_files = 0
     total_lines = 0
-    folder_count = 0
-    checked_files = 0
 
     folder_count, checked_files = print_directory_tree(path)
 
-    # Count total files and lines, including those outside folders
+    # Count total lines
     for root, dirs, files in os.walk(path):
-        # Skip any directories that start with a dot
         dirs[:] = [d for d in dirs if not d.startswith('.')]
 
         for file in files:
-            # Skip files that start with a dot
             if file.startswith('.'):
                 continue
 
             file_path = os.path.join(root, file)
             lines = count_lines_of_code(file_path)
-            if lines > 0:
-                total_files += 1
-                total_lines += lines
-                checked_files += 1
+            total_lines += lines
 
     print(f"\nTotal files checked: {checked_files}")
     print(f"Total lines of code: {total_lines}")

setup.py

@@ -2,7 +2,7 @@
 
 setup(
     name="codetree-cli",
-    version="0.2.1",
+    version="0.2.2",
     author="Reza Mardani", 
     author_email="mardanireza30@gmail.com",
     description="A tool for counting lines of code in a project.",