Skip to content

Commit

Permalink
Simplify argparse usage (#380)
Browse files Browse the repository at this point in the history
Refactor argument parsing to use argparse Namespace object for
structured access.
  • Loading branch information
William-Brown5515 authored Nov 28, 2024
1 parent 37d8623 commit 76baabc
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 14 deletions.
16 changes: 8 additions & 8 deletions bin/receiver.py
Original file line number Diff line number Diff line change
Expand Up @@ -51,23 +51,23 @@ def main():
action='version',
version=ver)

# Using the vars function to output a dict-like view rather than Namespace object.
options = vars(arg_parser.parse_args())
# Parsing arguments into an argparse.Namespace object for structured access.
options = arg_parser.parse_args()

# Deprecating functionality.
old_log_config_default_path = '/etc/apel/logging.cfg'
if (os.path.exists(old_log_config_default_path) or options['log_config'] is not None):
if (os.path.exists(old_log_config_default_path) or options.log_config is not None):
logging.warning('Separate logging config file option has been deprecated.')

# Absolute file path required when refreshing dn_file, relative path resulted in an error.
options['dn_file'] = os.path.abspath(options['dn_file'])
options.dn_file = os.path.abspath(options.dn_file)

# Check if config file exists using os.path.isfile function.
if os.path.isfile(options['config']):
if os.path.isfile(options.config):
cp = configparser.ConfigParser({'use_ssl': 'true'})
cp.read(options['config'])
cp.read(options.config)
else:
print("Config file not found at", options['config'])
print("Config file not found at", options.config)
sys.exit(1)

# Check for pidfile
Expand All @@ -90,7 +90,7 @@ def main():
brokers, project, token = ssm.agents.get_ssm_args(protocol, cp, log)

ssm.agents.run_receiver(protocol, brokers, project, token,
cp, log, options['dn_file'])
cp, log, options.dn_file)


if __name__ == '__main__':
Expand Down
12 changes: 6 additions & 6 deletions bin/sender.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,20 +46,20 @@ def main():
action='version',
version=ver)

# Using the vars function to output a dict-like view rather than Namespace object.
options = vars(arg_parser.parse_args())
# Parsing arguments into an argparse.Namespace object for structured access.
options = arg_parser.parse_args()

# Deprecating functionality.
old_log_config_default_path = '/etc/apel/logging.cfg'
if (os.path.exists(old_log_config_default_path) or options['log_config'] is not None):
if (os.path.exists(old_log_config_default_path) or options.log_config is not None):
logging.warning('Separate logging config file option has been deprecated.')

# Check if config file exists using os.path.isfile function.
if os.path.isfile(options['config']):
if os.path.isfile(options.config):
cp = configparser.ConfigParser({'use_ssl': 'true'})
cp.read(options['config'])
cp.read(options.config)
else:
print("Config file not found at", options['config'])
print("Config file not found at", options.config)
sys.exit(1)

ssm.agents.logging_helper(cp)
Expand Down

0 comments on commit 76baabc

Please sign in to comment.