diff --git a/src/viv/viv.py b/src/viv/viv.py index 6e88689..5e3a8b4 100755 --- a/src/viv/viv.py +++ b/src/viv/viv.py @@ -51,7 +51,7 @@ from typing import ( Generator, ) -__version__ = "22.12a3-53-g6d98e2a-dev" +__version__ = "22.12a3-54-g79d7978-dev" @dataclass @@ -859,9 +859,11 @@ class Viv: self.local_source = self.running_source else: try: - self.local_source = __import__("viv").__file__ + _local_viv = __import__("viv") + self.local_source = _local_viv.__file__ + self.local_version = _local_viv.__version__ except ImportError: - self.local_source = "Not Found" + self.local_source = self.local_version = "Not Found" self.git = self.local_source != "Not Found" and str(self.local_source).endswith( "src/viv/__init__.py" @@ -1008,7 +1010,7 @@ class Viv: q = ( "Update source at: " + a.style(self.running_source, "bold") - + f" \n from version {__version__} to {next_version}?" + + f" \n from version {self.local_version} to {next_version}?" ) if confirm(q):