Allow more strategies for major,minor,patch (#162)

awesomeversion/awesomeversion.py

@@ -277,22 +277,51 @@ def sections(self) -> int:
 
     @property
     def major(self) -> AwesomeVersion | None:
-        """Return a AwesomeVersion representation of the major version."""
-        if self.strategy != AwesomeVersionStrategy.SEMVER:
+        """
+        Return a AwesomeVersion representation of the major version.
+
+        Will return None if the versions is not semver/buildver/simplever.
+        """
+        if self.strategy not in (
+            AwesomeVersionStrategy.SEMVER,
+            AwesomeVersionStrategy.BUILDVER,
+            AwesomeVersionStrategy.SIMPLEVER,
+        ):
             return None
         return AwesomeVersion(self.section(0))
 
     @property
     def minor(self) -> AwesomeVersion | None:
-        """Return a AwesomeVersion representation of the minor version."""
-        if self.strategy != AwesomeVersionStrategy.SEMVER:
+        """
+        Return a AwesomeVersion representation of the minor version.
+
+        Will return None if the versions is not semver/simplever
+        Will return None if the version does not have at least 2 sections.
+        """
+        if self.strategy not in (
+            AwesomeVersionStrategy.SEMVER,
+            AwesomeVersionStrategy.SIMPLEVER,
+        ):
             return None
+
         return AwesomeVersion(self.section(1))
 
     @property
     def patch(self) -> AwesomeVersion | None:
-        """Return a AwesomeVersion representation of the patch version."""
-        if self.strategy != AwesomeVersionStrategy.SEMVER:
+        """
+        Return a AwesomeVersion representation of the patch version.
+
+        Will return None if the versions is not semver/simplever
+        Will return None if the version does not have at least 3 sections.
+        """
+        if (
+            self.strategy
+            not in (
+                AwesomeVersionStrategy.SEMVER,
+                AwesomeVersionStrategy.SIMPLEVER,
+            )
+            or self.sections < 3
+        ):
             return None
         return AwesomeVersion(self.section(2))
 

tests/test_awesomeversion.py

@@ -59,6 +59,30 @@ def test_awesomeversion() -> None:
             assert upstream > current
 
 
+def test_major() -> None:
+    """Test major."""
+    assert AwesomeVersion("1.0.0.0").major == 1
+    assert AwesomeVersion("1.0.0").major == 1
+    assert AwesomeVersion("1.0").major == 1
+    assert AwesomeVersion("1").major == 1
+
+
+def test_minor() -> None:
+    """Test minor."""
+    assert AwesomeVersion("1.1.0.0").minor == 1
+    assert AwesomeVersion("1.1.0").minor == 1
+    assert AwesomeVersion("1.1").minor == 1
+    assert AwesomeVersion("1").minor is None
+
+
+def test_patch() -> None:
+    """Test patch."""
+    assert AwesomeVersion("1.1.1.1").patch == 1
+    assert AwesomeVersion("1.1.1").patch == 1
+    assert AwesomeVersion("1.1").patch is None
+    assert AwesomeVersion("1").patch is None
+
+
 def test_serialization() -> None:
     """Test to and from JSON serialization."""
     version = AwesomeVersion("20.12.1")