From b0c14895b45fb5ce5c10150f48ab39b4eef0413d Mon Sep 17 00:00:00 2001 From: Tiffany Date: Mon, 6 Nov 2023 18:27:52 -0500 Subject: [PATCH] Create new branch to refactor cli --- main.py | 19 ------------------- requirements.txt | 4 ++++ simpleDDoS/__init__.py | 5 +++++ simpleDDoS/__main__.py | 0 simpleDDoS/cli.py | 28 ++++++++++++++++++++++++++++ {src => simpleDDoS}/ddos.py | 0 6 files changed, 37 insertions(+), 19 deletions(-) delete mode 100644 main.py create mode 100644 requirements.txt create mode 100644 simpleDDoS/__init__.py create mode 100644 simpleDDoS/__main__.py create mode 100644 simpleDDoS/cli.py rename {src => simpleDDoS}/ddos.py (100%) diff --git a/main.py b/main.py deleted file mode 100644 index 3ef7608..0000000 --- a/main.py +++ /dev/null @@ -1,19 +0,0 @@ -#!/usr/bin/python - -import argparse - -from src import ddos - - -def main(): - parser = argparse.ArgumentParser(description="DDoS Proof of concept") - parser.add_argument('-t', '--target', help="Attack target IP address") - parser.add_argument('-p', '--port', help="Port to attack") - parser.add_argument('-i', '--ipaddress', help="Fake ip address") - - args = parser.parse_args(ddos) - - - -if __name__ == "__main__": - main() diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..34c7acc --- /dev/null +++ b/requirements.txt @@ -0,0 +1,4 @@ +typer==0.3.2 +colorama==0.4.4 +shellingham==1.4.0 +pytest==6.2.4 \ No newline at end of file diff --git a/simpleDDoS/__init__.py b/simpleDDoS/__init__.py new file mode 100644 index 0000000..4795a4e --- /dev/null +++ b/simpleDDoS/__init__.py @@ -0,0 +1,5 @@ +"""Top-level package for Simple DDoS.""" +# simpleDDoS/__init__.py + +__app_name__ = 'simple_ddos' +__version__ = "0.1.0" diff --git a/simpleDDoS/__main__.py b/simpleDDoS/__main__.py new file mode 100644 index 0000000..e69de29 diff --git a/simpleDDoS/cli.py b/simpleDDoS/cli.py new file mode 100644 index 0000000..422c8e6 --- /dev/null +++ b/simpleDDoS/cli.py @@ -0,0 +1,28 @@ +"""Top-level package for Simple DDoS.""" +# simpleDDoS/cli.py + +from typing import Optional + +import typer + +from simpleDDoS import __app_name__, __version__ + +app = typer.Typer() + +def _version_callback(value: bool) -> None: + if value: + typer.echo(f"{__app_name__} v{__version__}") + raise typer.Exit() + +@app.callback() +def main( + version: Optional[bool] = typer.Option( + None, + "--version", + "-v", + help="Show the application's version and exit.", + callback=_version_callback, + is_eager=True, + ) +) -> None: + return \ No newline at end of file diff --git a/src/ddos.py b/simpleDDoS/ddos.py similarity index 100% rename from src/ddos.py rename to simpleDDoS/ddos.py