Add typedDict

Author: Yobmod

git/types.py

@@ -4,12 +4,12 @@
 
 import os
 import sys
-from typing import Union, Any
+from typing import Dict, Union, Any
 
 if sys.version_info[:2] >= (3, 8):
-    from typing import Final, Literal, SupportsIndex  # noqa: F401
+    from typing import Final, Literal, SupportsIndex, TypedDict  # noqa: F401
 else:
-    from typing_extensions import Final, Literal, SupportsIndex  # noqa: F401
+    from typing_extensions import Final, Literal, SupportsIndex, TypedDict  # noqa: F401
 
 
 if sys.version_info[:2] < (3, 9):
@@ -22,3 +22,21 @@
 TBD = Any
 
 Lit_config_levels = Literal['system', 'global', 'user', 'repository']
+
+
+class Files_TD(TypedDict):
+    insertions: int
+    deletions: int
+    lines: int
+
+
+class Total_TD(TypedDict):
+    insertions: int
+    deletions: int
+    lines: int
+    files: int
+
+
+class HSH_TD(TypedDict):
+    total: Total_TD
+    files: Dict[str, Files_TD]

git/util.py

@@ -32,7 +32,7 @@
     from git.repo.base import Repo
     from git.config import GitConfigParser, SectionConstraint
 
-from .types import PathLike, TBD, Literal, SupportsIndex
+from .types import PathLike, Literal, SupportsIndex, HSH_TD, Files_TD
 
 # ---------------------------------------------------------------------
 
@@ -746,7 +746,9 @@ class Stats(object):
      files = number of changed files as int"""
     __slots__ = ("total", "files")
 
-    def __init__(self, total: Dict[str, Dict[str, int]], files: Dict[str, Dict[str, int]]):
+    from git.types import Total_TD, Files_TD
+
+    def __init__(self, total: Total_TD, files: Dict[str, Files_TD]):
         self.total = total
         self.files = files
 
@@ -756,13 +758,12 @@ def _list_from_string(cls, repo: 'Repo', text: str) -> 'Stats':
 
         :return: git.Stat"""
 
-        # hsh: Dict[str, Dict[str, Union[int, Dict[str, int]]]]
-        hsh: Dict[str, Dict[str, TBD]] = {'total': {'insertions': 0,
-                                                    'deletions': 0,
-                                                    'lines': 0,
-                                                    'files': 0},
-                                          'files': {}
-                                          }   # need typeddict?
+        hsh: HSH_TD = {'total': {'insertions': 0,
+                                 'deletions': 0,
+                                 'lines': 0,
+                                 'files': 0},
+                       'files': {}
+                       }
         for line in text.splitlines():
             (raw_insertions, raw_deletions, filename) = line.split("\t")
             insertions = raw_insertions != '-' and int(raw_insertions) or 0
@@ -771,9 +772,9 @@ def _list_from_string(cls, repo: 'Repo', text: str) -> 'Stats':
             hsh['total']['deletions'] += deletions
             hsh['total']['lines'] += insertions + deletions
             hsh['total']['files'] += 1
-            files_dict = {'insertions': insertions,
-                          'deletions': deletions,
-                          'lines': insertions + deletions}
+            files_dict: Files_TD = {'insertions': insertions,
+                                    'deletions': deletions,
+                                    'lines': insertions + deletions}
             hsh['files'][filename.strip()] = files_dict
         return Stats(hsh['total'], hsh['files'])