refactor(GitStatus): Use dataclasses

Author: tony

libvcs/projects/git.py

@@ -18,7 +18,7 @@
 import logging
 import pathlib
 import re
-from typing import Dict, NamedTuple, Optional, TypedDict, Union
+from typing import Dict, Optional, TypedDict, Union
 from urllib import parse as urlparse
 
 from .. import exc
@@ -54,14 +54,21 @@ def to_tuple(self):
 GitRemotesArgs = Union[None, GitFullRemoteDict, Dict[str, str]]
 
 
-class GitStatus(NamedTuple):
+@dataclasses.dataclass
+class GitStatus:
     branch_oid: Optional[str]
     branch_head: Optional[str]
     branch_upstream: Optional[str]
     branch_ab: Optional[str]
     branch_ahead: Optional[str]
     branch_behind: Optional[str]
 
+    def to_dict(self):
+        return dataclasses.asdict(self)
+
+    def to_tuple(self):
+        return dataclasses.astuple(self)
+
 
 def extract_status(value) -> GitStatus:
     """Returns ``git status -sb --porcelain=2`` extracted to a dict

tests/projects/test_git.py

@@ -628,7 +628,7 @@ def test_extract_status():
     assert {
         "branch_oid": "d4ccd4d6af04b53949f89fbf0cdae13719dc5a08",
         "branch_head": "fix-current-remote-name",
-    }.items() <= extract_status(FIXTURE_A)._asdict().items()
+    }.items() <= extract_status(FIXTURE_A).to_dict().items()
 
 
 @pytest.mark.parametrize(
@@ -676,7 +676,7 @@ def test_extract_status():
 )
 def test_extract_status_b(fixture: str, expected_result: dict):
     assert (
-        extract_status(textwrap.dedent(fixture))._asdict().items()
+        extract_status(textwrap.dedent(fixture)).to_dict().items()
         >= expected_result.items()
     )
 
@@ -727,7 +727,7 @@ def test_extract_status_b(fixture: str, expected_result: dict):
 def test_extract_status_c(fixture: str, expected_result: dict):
     assert (
         expected_result.items()
-        <= extract_status(textwrap.dedent(fixture))._asdict().items()
+        <= extract_status(textwrap.dedent(fixture)).to_dict().items()
     )