closes #4

davidpeckham · Feb 25, 2024 · a169e14 · a169e14
1 parent 43ed3e1
commit a169e14
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 2 deletions.
diff --git a/src/vin/__init__.py b/src/vin/__init__.py
@@ -18,7 +18,7 @@
 from vin.constants import VIN_LENGTH
 from vin.constants import VIN_MODEL_YEAR_CHARACTERS
 from vin.constants import VIN_POSITION_WEIGHTS
-from vin.database import decode_vin
+from vin.database import decode_vin, get_vpic_version
 
 
 class DecodingError(Exception):
@@ -588,8 +588,12 @@ def _decode_vin(self) -> None:
         self._body_class = vehicle.get("body_class", "")
         self._electrification_level = vehicle.get("electrification_level", "")
 
+    @classmethod
+    def version(cls):
+        return get_vpic_version()
+
     def __repr__(self) -> str:
         return f"VIN({self!s})"
 
     def __str__(self) -> str:
-        return self._vin
+        return self.description
diff --git a/src/vin/database.py b/src/vin/database.py
@@ -155,3 +155,18 @@ def get_make_from_wmi(wmi: str) -> str:
 where
     wmi.code == ?;
 """
+
+
+def get_vpic_version() -> dict:
+    return dict(query(sql=GET_VPIC_VERSION_SQL)[0])
+
+
+GET_VPIC_VERSION_SQL = """
+select
+    version,
+    released,
+    effective,
+    url
+from
+    vpic_version;
+"""
diff --git a/src/vin/vehicle.db b/src/vin/vehicle.db
diff --git a/tests/test_version.py b/tests/test_version.py
@@ -0,0 +1,7 @@
+from vin import VIN
+
+
+def test_version():
+    version = VIN.version()
+    assert version["version"] == "3.44"
+    assert version["released"] == "2024-02-17"