diff --git a/pcap_anonymize/cli.py b/pcap_anonymize/__main__.py similarity index 50% rename from pcap_anonymize/cli.py rename to pcap_anonymize/__main__.py index dcdbb3394acdb095074c6d12c8ce21035a9c7fa2..9b740f61c139f3764d32a5d2fe931cb8280995b0 100644 --- a/pcap_anonymize/cli.py +++ b/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() diff --git a/pyproject.toml b/pyproject.toml index 8d0e43d229f308b41ac7f39134e3e661ccd8f863..3e2eb78af8f3351db5aaa0a88c1756539fd3f658 100644 --- a/pyproject.toml +++ b/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"