Skip to content

move to use max instead of if else in length, use list comprehension to simplify for loops #95

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
galrock wants to merge 6 commits into
base: master
Choose a base branch
from
Open

move to use max instead of if else in length, use list comprehension to simplify for loops #95

Changes from all commits
Commits
Show all changes
6 commits
Select commit Hold shift + click to select a range
c618db1
some changes
galrock Apr 3, 2023
5f626c6
move to class variables and change cust_args
galrock Apr 3, 2023
524e3ac
change width to max, decrease cyclomatic complexity
galrock Apr 3, 2023
bb4a5a4
revert class variable changes, remove __add__ for now until used, rem…
galrock Apr 4, 2023
8f1dd60
forgot to bring back a line and revert the while loop
galrock Apr 4, 2023
edfbb26
forgot the break
galrock Apr 4, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
43 changes: 17 additions & 26 deletions lshca/lshca.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -158,9 +158,7 @@ def parse_arguments(self, user_args):
# comes to handle comma separated list of choices
cust_user_args = []
for arg in user_args:
result = arg.split(",")
for member in result:
cust_user_args.append(member)
cust_user_args = [x for x in arg.split(",")]

args = parser.parse_args(cust_user_args)
self.process_arguments(args)
Expand Down Expand Up @@ -404,7 +402,6 @@ def get_data(self):
sf_dev.get_data()
mlnx_bdf_devices.append(sf_dev)


if port_count >= len(bdf_dev.port_list):
break

Expand Down Expand Up @@ -500,12 +497,12 @@ def __init__(self, config, data_source):

def append(self, data):
self.output.append(data)

def apply_select_output_filters(self):
# type: () -> None
if len(self.config.output_fields_filter_positive) > 0:
if self.config.output_fields_filter_positive:
self.output_order = self.config.output_fields_filter_positive
elif len(self.config.output_fields_filter_negative) > 0:
elif self.config.output_fields_filter_negative:
decrement_list = self.output_order

output_filter = self.config.output_fields_filter_negative
Expand All @@ -516,7 +513,7 @@ def apply_select_output_filters(self):
self.output_order = decrement_list

data_keys_remove_list = []
if len(self.output) > 0:
if self.output:
output_data_keys = list(self.output[0]) + list(self.output[0]["bdf_devices"][0])
data_keys_remove_list = list(set(output_data_keys) - set(self.output_order))

Expand Down Expand Up @@ -694,22 +691,18 @@ def update_separator_and_column_width(self):
# first pass: collect all of the maximum widths for each of the BDF fields
hca_field_line_width = 0
for hca in self.output:
for key in hca:
if key == "bdf_devices":
for bdf_device in hca["bdf_devices"]:
for bdf_key in bdf_device:
if bdf_key in self.output_order:
# decide what is longer the key name or it's value
if len(bdf_device[bdf_key]) > len(bdf_key):
width = len(bdf_device[bdf_key])
else:
width = len(bdf_key)
bdf_devices = [hca[x] for x in hca if x=="bdf_devices"]
for bdf_device in bdf_devices:
for bdf_key in bdf_device:
if bdf_key in self.output_order:
# decide what is longer the key name or it's value
width = max([len(bdf_device[bdf_key]),len(bdf_key)])

if bdf_key not in self.column_width:
self.column_width[bdf_key] = width
if bdf_key not in self.column_width:
self.column_width[bdf_key] = width

if width > self.column_width[bdf_key]:
self.column_width[bdf_key] = width
if width > self.column_width[bdf_key]:
self.column_width[bdf_key] = width

# second pass: calculate width of BDF and HCA lines
for hca in self.output:
Expand Down Expand Up @@ -779,11 +772,9 @@ def print_hca_header(self, args):
# type: (dict) -> None
order_dict = {}

position = 0
for key in self.output_order:
for idx, key in enumerate(self.output_order):
if key in args:
order_dict[key] = position
position += 1
order_dict[key] = idx

output_list = [""] * len(order_dict)
for key in args:
Expand Down