Fixed linting

Author: nklsw

src/docs2llm/__init__.py

@@ -6,13 +6,13 @@
 """
 
 from docs2llm.main import (
-    extract_documentation,
-    setup_logging,
-    is_documentation_file,
-    markdown_to_text,
-    clone_repository,
-    find_documentation_files,
-    process_documentation_files
+    extract_documentation,  # noqa: F401
+    setup_logging, # noqa: F401
+    is_documentation_file, # noqa: F401
+    markdown_to_text, # noqa: F401
+    clone_repository, # noqa: F401
+    find_documentation_files, # noqa: F401
+    process_documentation_files, # noqa: F401
 )
 
 __version__ = "0.1.0"

src/docs2llm/cli.py

@@ -3,28 +3,37 @@
 import click
 from docs2llm.main import extract_documentation, console
 
+
 @click.command()
 @click.argument("path", required=False)
-@click.option("--git", help="GitHub repository URL (e.g., https://github.com/owner/repo.git)")
+@click.option(
+    "--git", help="GitHub repository URL (e.g., https://github.com/owner/repo.git)"
+)
 @click.option("--output", default="llm_context.txt", help="Output file name")
-@click.option("--max-depth", type=int, default=3, help="Maximum directory depth to search")
+@click.option(
+    "--max-depth", type=int, default=3, help="Maximum directory depth to search"
+)
 @click.option("--branch", help="Specific branch to clone (only used with --git)")
 @click.option("--verbose", "-v", is_flag=True, help="Enable verbose logging")
 @click.option("--log-file", help="Log to this file in addition to console")
 def main(path, git, output, max_depth, branch, verbose, log_file):
     """Generate LLM context from documentation in a directory or GitHub repository.
-    
+
     PATH: Local directory path containing documentation files
-    
+
     You can either provide a local path or use --git with a GitHub repository URL.
     Example: --git https://github.com/owner/repo.git
     """
     if not path and not git:
-        console.print("[bold red]Error: Either a local path or --git option must be provided.[/bold red]")
+        console.print(
+            "[bold red]Error: Either a local path or --git option must be provided.[/bold red]"
+        )
         sys.exit(1)
-    
+
     if path and git:
-        console.print("[bold red]Error: Cannot specify both a local path and --git. Choose one input source.[/bold red]")
+        console.print(
+            "[bold red]Error: Cannot specify both a local path and --git. Choose one input source.[/bold red]"
+        )
         sys.exit(1)
 
     success = extract_documentation(
@@ -34,12 +43,12 @@ def main(path, git, output, max_depth, branch, verbose, log_file):
         max_depth=max_depth,
         branch=branch,
         verbose=verbose,
-        log_file=log_file
+        log_file=log_file,
     )
-    
+
     # Exit without any success/failure message
     sys.exit(0 if success else 1)
 
 
 if __name__ == "__main__":
-    main()
+    main()

src/docs2llm/main.py

@@ -32,8 +32,9 @@ def setup_logging(verbose=False, log_file=None, use_rich=True):
 
     # Create console handler, using Rich if available
     if use_rich:
-        console_handler = RichHandler(console=console, rich_tracebacks=True, 
-                                    show_time=False, show_path=False)
+        console_handler = RichHandler(
+            console=console, rich_tracebacks=True, show_time=False, show_path=False
+        )
         root_logger.addHandler(console_handler)
     else:
         console_handler = logging.StreamHandler(sys.stdout)
@@ -126,7 +127,7 @@ def markdown_to_text(markdown_content):
 
 def clone_repository(repo_path, temp_dir, branch=None):
     """Clone a GitHub repository to a local directory.
-    
+
     Args:
         repo_path: GitHub repository URL or owner/repo format
         temp_dir: Directory to clone the repository into
@@ -149,7 +150,7 @@ def clone_repository(repo_path, temp_dir, branch=None):
         # Traditional owner/repo format (for backward compatibility)
         clone_url = f"https://github.com/{repo_path}.git"
         owner_repo = repo_path
-    
+
     logging.info(f"Cloning repository {clone_url} to temporary directory")
 
     start_time = time.time()
@@ -207,7 +208,7 @@ def find_documentation_files(repo_dir, max_depth=3):
     logging.info(f"Searching for documentation files (max depth: {max_depth})")
 
     # Skip .git directory
-    with console.status(f"[bold blue]Scanning repository...", spinner="dots") as status:
+    with console.status("[bold blue]Scanning repository...", spinner="dots") as status:
         # Walk the directory tree
         for root, dirs, files in os.walk(repo_dir):
             # Remove .git directory from traversal
@@ -269,14 +270,18 @@ def process_documentation_files(repo_dir, doc_files):
 
     logging.info(f"Processing {len(doc_files)} documentation files")
 
-    with console.status(f"[bold blue]Processing documentation files...", spinner="dots") as status:
+    with console.status(
+        "[bold blue]Processing documentation files...", spinner="dots"
+    ) as status:
         file_count = 0
-        
+
         for file_path in sorted_files:
             full_path = os.path.join(repo_dir, file_path)
             try:
                 file_count += 1
-                status.update(f"[cyan]Processing: {file_path} ({file_count}/{len(doc_files)})")
+                status.update(
+                    f"[cyan]Processing: {file_path} ({file_count}/{len(doc_files)})"
+                )
                 logging.debug(f"Processing {file_path}...")
 
                 # Read file content
@@ -302,7 +307,7 @@ def process_documentation_files(repo_dir, doc_files):
 
             except Exception as e:
                 logging.error(f"Error processing {file_path}: {e}", exc_info=True)
-        
+
         status.update("[bold green]Processing complete")
 
     total_length = sum(len(text) for text in docs_text)
@@ -311,16 +316,16 @@ def process_documentation_files(repo_dir, doc_files):
 
 
 def extract_documentation(
-    local_path=None, 
-    git_repo=None, 
-    output_file="llm_context.txt", 
-    max_depth=3, 
-    branch=None, 
-    verbose=False, 
-    log_file=None
+    local_path=None,
+    git_repo=None,
+    output_file="llm_context.txt",
+    max_depth=3,
+    branch=None,
+    verbose=False,
+    log_file=None,
 ):
     """Extract documentation from a local directory or a GitHub repository.
-    
+
     Args:
         local_path: Path to a local directory containing documentation
         git_repo: GitHub repository in the format 'owner/repo'
@@ -329,40 +334,39 @@ def extract_documentation(
         branch: Specific branch to clone (only used with git_repo)
         verbose: Enable verbose logging
         log_file: Path to a file where logs will be written in addition to the console
-        
+
     Returns:
         bool: True if the extraction was successful, False otherwise
     """
     # Setup logging
     setup_logging(verbose=verbose, log_file=log_file)
-    
+
     # Determine if we're using a local path or git repo
     is_local = local_path is not None
-    source_path = local_path if is_local else git_repo
-    
+
     try:
         if is_local:
             # Using local directory
             if not exists(local_path):
                 logging.error(f"Local path does not exist: {local_path}")
                 return False
-                
+
             if not isdir(local_path):
                 logging.error(f"Path is not a directory: {local_path}")
                 return False
-                
+
             logging.info(f"Using local directory: {local_path}")
-            
+
             # Process the local directory
             doc_files = find_documentation_files(local_path, max_depth)
-            
+
             if not doc_files:
                 logging.warning("No documentation files found in the directory.")
                 return False
-                
+
             # Process the documentation files
             docs_text = process_documentation_files(local_path, doc_files)
-            
+
             # Create header for local directory
             dir_name = os.path.basename(os.path.abspath(local_path))
             header = f"""# Documentation from local directory: {local_path}
@@ -421,7 +425,7 @@ def extract_documentation(
             return False
 
         file_size_kb = os.path.getsize(output_file) / 1024
-        
+
         console.print(f"[bold green]Documentation saved to [blue]{output_file}[/blue]")
         console.print(f"[bold green]Total size: [yellow]{file_size_kb:.1f}[/yellow] KB")
 
@@ -430,9 +434,9 @@ def extract_documentation(
                 f"[bold yellow]Warning: The output file is large ({file_size_kb:.1f} KB). "
                 f"This may exceed context limits for some LLMs.[/bold yellow]"
             )
-        
+
         return True
 
     except Exception as e:
         logging.error(f"An error occurred: {e}", exc_info=True)
-        return False
+        return False