diff --git a/main.py b/main.py index a9b8509..1433eff 100644 --- a/main.py +++ b/main.py @@ -15,6 +15,9 @@ from source.jetsort import JetSortSimulator from source.pelican import PelicanSimulator from source.selex import SelexSimulator +DEFAULT_BAUDRATE = 115200 +DEFAULT_PORT = "/dev/ttyUSB0" + def main(): parser = argparse.ArgumentParser( @@ -54,11 +57,15 @@ Examples: pelican_parser.add_argument( "--port", "-p", - default="/dev/ttyUSB0", - help="Serial port (default: /dev/ttyUSB0)", + default=DEFAULT_PORT, + help=f"Serial port (default: {DEFAULT_PORT})", ) pelican_parser.add_argument( - "--baudrate", "-b", type=int, default=115200, help="Baud rate (default: 115200)" + "--baudrate", + "-b", + type=int, + default=DEFAULT_BAUDRATE, + help=f"Baud rate (default: {DEFAULT_BAUDRATE})", ) # Glory simulator subcommand @@ -68,11 +75,15 @@ Examples: glory_parser.add_argument( "--port", "-p", - default="/dev/ttyUSB0", - help="Serial port (default: /dev/ttyUSB0)", + default=DEFAULT_PORT, + help=f"Serial port (default: {DEFAULT_PORT})", ) glory_parser.add_argument( - "--baudrate", "-b", type=int, default=115200, help="Baud rate (default: 115200)" + "--baudrate", + "-b", + type=int, + default=DEFAULT_BAUDRATE, + help=f"Baud rate (default: {DEFAULT_BAUDRATE})", ) # Selex simulator subcommand @@ -82,11 +93,15 @@ Examples: selex_parser.add_argument( "--port", "-p", - default="/dev/ttyUSB0", - help="Serial port (default: /dev/ttyUSB0)", + default=DEFAULT_PORT, + help=f"Serial port (default: {DEFAULT_PORT})", ) selex_parser.add_argument( - "--baudrate", "-b", type=int, default=115200, help="Baud rate (default: 115200)" + "--baudrate", + "-b", + type=int, + default=DEFAULT_BAUDRATE, + help=f"Baud rate (default: {DEFAULT_BAUDRATE})", ) # JetSort simulator subcommand @@ -96,11 +111,15 @@ Examples: jetsort_parser.add_argument( "--port", "-p", - default="/dev/ttyUSB0", - help="Serial port (default: /dev/ttyUSB0)", + default=DEFAULT_PORT, + help=f"Serial port (default: {DEFAULT_PORT})", ) jetsort_parser.add_argument( - "--baudrate", "-b", type=int, default=115200, help="Baud rate (default: 115200)" + "--baudrate", + "-b", + type=int, + default=DEFAULT_BAUDRATE, + help=f"Baud rate (default: {DEFAULT_BAUDRATE})", ) args = parser.parse_args()