Replace StrictVersion with AwesomeVersion (#46331)

This commit is contained in:
Joakim Sørensen 2021-02-10 15:25:44 +01:00 committed by GitHub
parent 7928cda080
commit ea4ad85488
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,10 +1,10 @@
"""Support to check for available updates.""" """Support to check for available updates."""
import asyncio import asyncio
from datetime import timedelta from datetime import timedelta
from distutils.version import StrictVersion
import logging import logging
import async_timeout import async_timeout
from awesomeversion import AwesomeVersion
from distro import linux_distribution # pylint: disable=import-error from distro import linux_distribution # pylint: disable=import-error
import voluptuous as vol import voluptuous as vol
@ -83,16 +83,16 @@ async def async_setup(hass, config):
# Validate version # Validate version
update_available = False update_available = False
if StrictVersion(newest) > StrictVersion(current_version): if AwesomeVersion(newest) > AwesomeVersion(current_version):
_LOGGER.debug( _LOGGER.debug(
"The latest available version of Home Assistant is %s", newest "The latest available version of Home Assistant is %s", newest
) )
update_available = True update_available = True
elif StrictVersion(newest) == StrictVersion(current_version): elif AwesomeVersion(newest) == AwesomeVersion(current_version):
_LOGGER.debug( _LOGGER.debug(
"You are on the latest version (%s) of Home Assistant", newest "You are on the latest version (%s) of Home Assistant", newest
) )
elif StrictVersion(newest) < StrictVersion(current_version): elif AwesomeVersion(newest) < AwesomeVersion(current_version):
_LOGGER.debug( _LOGGER.debug(
"Local version (%s) is newer than the latest available version (%s)", "Local version (%s) is newer than the latest available version (%s)",
current_version, current_version,