Refactored for Typer compatability

Author: nicoleman0

main.py

@@ -1,41 +1,60 @@
+import typer
+from typing_extensions import Annotated
 from scan import scan_url, get_url_report, get_api_key
 from workbook import save_to_excel
 from generate import generate_html_report
-import argparse
 
+app = typer.Typer()
 
-def main():
-    api_key = get_api_key()  # Get the API key from user input
 
-    parser = argparse.ArgumentParser(description="Scan URLs with VirusTotal.")
-    parser.add_argument(
-        "urls",
-        metavar="URL",
-        type=str,
-        nargs="+",
-        help="URLs to scan (space-separated)",
-    )
-    args = parser.parse_args()
-    urls = args.urls
+@app.command()
+def main(
+    urls: Annotated[
+        list[str],
+        typer.Argument(help="URLs to scan (space-separated)"),
+    ],
+    api_key: Annotated[
+        str,
+        typer.Option(
+            "--api-key",
+            "-k",
+            help="Your VirusTotal API key. If not provided, you will be prompted.",
+        ),
+    ] = None,
+):
+    """
+    Scan URLs with VirusTotal and generate reports.
+    """
+    if not api_key:
+        api_key = get_api_key()
 
     results = []
     for url in urls:
-        scan_id = scan_url(url, api_key)  # Pass the API key to scan_url
+        scan_id = scan_url(url, api_key)
         if scan_id:
-            # Pass the API key to get_url_report
             report = get_url_report(scan_id, api_key)
             if report:
-                stats = report.results.get('attributes', {}).get('stats', {})
-                results.append({
-                    'url': url,
-                    'malicious': stats.get('malicious', 0),
-                    'harmless': stats.get('harmless', 0),
-                    'undetected': stats.get('undetected', 0)
-                })
+                attributes = report.get("attributes", {})
+                stats = attributes.get("stats", {})
+                results.append(
+                    {
+                        "url": url,
+                        "malicious": stats.get("malicious", 0),
+                        "harmless": stats.get("harmless", 0),
+                        "undetected": stats.get("undetected", 0),
+                    }
+                )
+            else:
+                print(f"No report found for {url}")
+        else:
+            print(f"Scan failed for {url}")
 
-    save_to_excel(results)
-    generate_html_report(results)
+    if results:
+        save_to_excel(results)
+        generate_html_report(results)
+    else:
+        print("No results to report.")
 
 
 if __name__ == "__main__":
-    main()
+    app()

requirements.txt

@@ -0,0 +1,34 @@
+aiofiles==24.1.0
+aiohappyeyeballs==2.6.1
+aiohttp==3.11.14
+aiosignal==1.3.2
+attrs==25.3.0
+certifi==2025.1.31
+charset-normalizer==3.4.1
+click==8.1.8
+et_xmlfile==2.0.0
+frozenlist==1.5.0
+idna==3.10
+Jinja2==3.1.6
+markdown-it-py==3.0.0
+MarkupSafe==3.0.2
+mdurl==0.1.2
+multidict==6.2.0
+numpy==2.2.4
+openpyxl==3.1.5
+pandas==2.2.3
+propcache==0.3.0
+Pygments==2.19.1
+python-dateutil==2.9.0.post0
+python-dotenv==1.1.0
+pytz==2025.2
+requests==2.32.3
+rich==13.9.4
+shellingham==1.5.4
+six==1.17.0
+typer==0.15.2
+typing_extensions==4.13.0
+tzdata==2025.2
+urllib3==2.3.0
+vt-py==0.19.0
+yarl==1.18.3