gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2026-02-04 11:07:53 -08:00
Code Issues Packages Projects Releases Wiki Activity

pep8

Browse Source
This commit is contained in:
William Ballenthin
2021-06-14 10:25:37 -06:00
parent dba250ca86
commit 6842b92ca2
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
capa/rules.py
View File

@@ -148,10 +148,12 @@ class InvalidRuleSet(ValueError):
def ensure_feature_valid_for_scope(scope: str, feature: Feature):
# if the given feature is a characteristic,
# check that is a valid characteristic for the given scope.
if (isinstance(feature, capa.features.common.Characteristic)
if (
isinstance(feature, capa.features.common.Characteristic)
and isinstance(feature.value, str)
and capa.features.common.Characteristic(feature.value) not in SUPPORTED_FEATURES[scope]):
raise InvalidRule("feature %s not support for scope %s" % (feature, scope))
and capa.features.common.Characteristic(feature.value) not in SUPPORTED_FEATURES[scope]
):
raise InvalidRule("feature %s not support for scope %s" % (feature, scope))
# features of this scope that are not Characteristics will be Type instances.
# check that the given feature is one of these types.
@@ -309,8 +311,7 @@ def parse_description(s: Union[str, int, bytes], value_type: str, description=No
# the string "10" that needs to become the number 10.
if value_type == "bytes":
value = parse_bytes(value)
elif (value_type in ("number", "offset")
or value_type.startswith(("number/", "offset/"))):
elif value_type in ("number", "offset") or value_type.startswith(("number/", "offset/")):
try:
value = parse_int(value)
except ValueError: