F5/unison
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

style: apply black and isort

Browse Source
This commit is contained in:
Gabriel Ferreira
2023-11-19 20:07:19 -03:00
parent af98671fbe
commit 8f6a3413a4
68 changed files with 7848 additions and 4299 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
utils.py
View File

@@ -8,53 +8,51 @@ import os
def get_list_from_file(file_path, list_name):
'''Looks for a Python list called list_name in the file specified
"""Looks for a Python list called list_name in the file specified
by file_path and returns it.
If the file or list name aren't found, this function will return
an empty list.
'''
"""
# Read in the file if it exists.
if not os.path.exists(file_path):
return []
with open(file_path, "r", encoding="utf-8") as file_in:
# Look for the list.
list_string = ""
parsing_multiline_list = False
for line in file_in:
# Remove any comments.
if '#' in line:
(line, comment) = line.split('#', 1)
if "#" in line:
(line, comment) = line.split("#", 1)
# Parse the line.
if list_name in line or parsing_multiline_list:
list_string += line
# Handle multiline lists.
if ']' not in list_string:
if "]" not in list_string:
parsing_multiline_list = True
else:
# Evaluate the list once its end is reached.
# Make the split function only split it once.
return eval(list_string.split('=', 1)[1].strip())
return eval(list_string.split("=", 1)[1].strip())
# List name was not found
return []
def get_bool_from_file(file_path, bool_name, value_if_missing):
'''Looks for a Python boolean variable called bool_name in the
"""Looks for a Python boolean variable called bool_name in the
file specified by file_path and returns its value.
If the file or boolean variable aren't found, this function will
return value_if_missing.
'''
"""
# Read in the file if it exists.
if not os.path.exists(file_path):
@@ -63,16 +61,15 @@ def get_bool_from_file(file_path, bool_name, value_if_missing):
with open(file_path, "r", encoding="utf-8") as file_in:
# Look for the boolean variable.
for line in file_in:
# Remove any comments.
if '#' in line:
(line, comment) = line.split('#', 1)
if "#" in line:
(line, comment) = line.split("#", 1)
# Parse the line.
if bool_name in line:
# Evaluate the variable's line once it is found. Make
# the split function only split it once.
return eval(line.split('=', 1)[1].strip())
return eval(line.split("=", 1)[1].strip())
# Boolean variable was not found
return value_if_missing
@@ -85,17 +82,17 @@ def get_bool_from_file(file_path, bool_name, value_if_missing):
def read_config_file():
# By default, all modules will be enabled, examples will be disabled,
# and tests will be disabled.
modules_enabled = ['all_modules']
modules_enabled = ["all_modules"]
examples_enabled = False
tests_enabled = False
tests_enabled = False
# See if the ns3 configuration file exists in the current working
# directory and then look for it in the ~ directory.
config_file_exists = False
dot_ns3rc_name = '.ns3rc'
dot_ns3rc_name = ".ns3rc"
dot_ns3rc_path = dot_ns3rc_name
if not os.path.exists(dot_ns3rc_path):
dot_ns3rc_path = os.path.expanduser('~/') + dot_ns3rc_name
dot_ns3rc_path = os.path.expanduser("~/") + dot_ns3rc_name
if not os.path.exists(dot_ns3rc_path):
# Return all of the default values if the .ns3rc file can't be found.
return (config_file_exists, modules_enabled, examples_enabled, tests_enabled)
@@ -103,17 +100,17 @@ def read_config_file():
config_file_exists = True
# Read in the enabled modules.
modules_enabled = get_list_from_file(dot_ns3rc_path, 'modules_enabled')
modules_enabled = get_list_from_file(dot_ns3rc_path, "modules_enabled")
if not modules_enabled:
# Enable all modules if the modules_enabled line can't be found.
modules_enabled = ['all_modules']
modules_enabled = ["all_modules"]
# Read in whether examples should be enabled or not.
value_if_missing = False
examples_enabled = get_bool_from_file(dot_ns3rc_path, 'examples_enabled', value_if_missing)
examples_enabled = get_bool_from_file(dot_ns3rc_path, "examples_enabled", value_if_missing)
# Read in whether tests should be enabled or not.
value_if_missing = False
tests_enabled = get_bool_from_file(dot_ns3rc_path, 'tests_enabled', value_if_missing)
tests_enabled = get_bool_from_file(dot_ns3rc_path, "tests_enabled", value_if_missing)
return (config_file_exists, modules_enabled, examples_enabled, tests_enabled)