我在Python中有以下函数:
def _extract_grp_entitlements(self,saml_authentication_attributes,groups):
result = []
input_length = len(saml_authentication_attributes[groups])
if input_length == 0:
log.error(self.empty_entitlements_message)
raise RuntimeError(self.empty_entitlements_message)
if input_length == 1:
result = [t.strip() for t in saml_authentication_attributes[groups][0].split(',')]
elif input_length:
result = saml_authentication_attributes[groups]
return result
用else
子句替换那里的elif
是否有任何优点/缺点(除了逻辑控制流(——速度、内存等?
这是否更可取:
def _extract_grp_entitlements(self,saml_authentication_attributes,groups):
input_length = len(saml_authentication_attributes[groups])
if input_length == 0:
log.error(self.empty_entitlements_message)
raise RuntimeError(self.empty_entitlements_message)
return [t.strip() for t in saml_authentication_attributes[groups][0].split(',')]
if len(saml_authentication_attributes[groups]) == 1
else saml_authentication_attributes[groups]
else
会更清楚。 您的elif
将始终运行,因此没有必要对其设置条件。
您的第一个函数已经足够可读,并且性能不太可能成为问题。为了在短函数中获得最大的可读性,我会这样写:
def _extract_grp_entitlements(self,saml_authentication_attributes,groups):
inp = saml_authentication_attributes[groups]
if inp:
if len(inp) == 1:
return [t.strip() for t in inp[0].split(',')]
else:
return inp
else:
log.error(self.empty_entitlements_message)
raise RuntimeError(self.empty_entitlements_message)
在我看来,这让流程一目了然。else
s 都是不必要的(因为如果条件为真,函数已经return
了(,并且只是为了使其更明确。有些人喜欢在return
之后省略else
。
对于较长的函数,将所有主逻辑都嵌套在一起可能会变得很痛苦,并且将不再清楚末尾的尾随else
指的是什么条件,因此处理顶部参数的基本问题更方便。
def _extract_grp_entitlements(self,saml_authentication_attributes,groups):
inp = saml_authentication_attributes[groups]
# get the no input error out of the way
if not inp:
log.error(self.empty_entitlements_message)
raise RuntimeError(self.empty_entitlements_message)
# now do everything else (no need for else)
if len(inp) == 1:
# etc.
我设法缩短了函数的第二个版本,使其既可读又简洁:
def _extract_grp_entitlements(self,groups):
groups_length = len(groups)
if groups_length == 0:
log.error(self.empty_entitlements_message)
raise RuntimeError(self.empty_entitlements_message)
return [t.strip() for t in groups[0].split(',')]
if groups_length == 1
else groups