Fix TypeError in clean_nutrition function (#1043)

* Fix Null error

* Better fix

* Black fixes

* extend-ignore should be used for adding ignore
This commit is contained in:
zackbcom 2022-03-15 15:44:46 -05:00 committed by GitHub
parent 73c537ed36
commit 6c549ff993
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 2 deletions

View File

@ -1,5 +1,5 @@
[flake8]
ignore = [
extend-ignore = [
E501 # Line Length - See Black Config in pyproject.toml
E402 # Import Not at Top of File
]

View File

@ -93,7 +93,12 @@ def clean_nutrition(nutrition: Optional[dict]) -> dict[str, str]:
output_nutrition = {key: val.replace(",", ".") for key, val in output_nutrition.items()}
if "sodiumContent" in nutrition and "m" not in nutrition["sodiumContent"] and "g" in nutrition["sodiumContent"]:
if (
"sodiumContent" in nutrition
and type(nutrition["sodiumContent"]) == str
and "m" not in nutrition["sodiumContent"]
and "g" in nutrition["sodiumContent"]
):
# Sodium is in grams. Parse its value, multiple by 1k and return to string.
try:
output_nutrition["sodiumContent"] = str(float(output_nutrition["sodiumContent"]) * 1000)