Python package meta config

pcap_anonymize/cli.py → pcap_anonymize/__main__.py

@@ -3,13 +3,16 @@ import argparse
 from .pcap_anonymize import anonymize_pcap
 
 
+### MAIN FUNCTION ###
 def main() -> None:
-    """
-    Main function for the CLI.
-    """
     parser = argparse.ArgumentParser(description="Anonymize a PCAP traffic capture.")
-    parser.add_argument("input", type=os.PathLike, help="Path to the input PCAP file.")
-    parser.add_argument("-o", "--output", type=os.PathLike, help="Path to the output PCAP file.")
+    parser.add_argument("input", type=str, help="Path to the input PCAP file.")
+    parser.add_argument("-o", "--output", type=str, help="Path to the output PCAP file.")
     args = parser.parse_args()
 
     anonymize_pcap(args.input, args.output)
+
+
+### ENTRY POINT ###
+if __name__ == "__main__":
+    main()

pyproject.toml

@@ -52,3 +52,7 @@ dependencies = [
 [project.urls]
 "Homepage" = "https://forge.uclouvain.be/smart-home-network-security/pcap-anonymize"
 "Source" = "https://forge.uclouvain.be/smart-home-network-security/pcap-anonymize"
+
+[project.scripts]
+pcap-anonymize = "pcap_anonymize:__main__.main"
+pcap_anonymize = "pcap_anonymize:__main__.main"