imporve types accoding to test/test_attributes.py

Author: BenediktSeidl

pygit2/_pygit2.pyi

@@ -1,10 +1,12 @@
 from typing import Iterator, Literal, Optional, overload, Type, TypedDict
 from io import IOBase, DEFAULT_BUFFER_SIZE
+from pathlib import Path
 from queue import Queue
 import tarfile
 from threading import Event
 from . import Index
 from .enums import (
+    AttrCheck,
     ApplyLocation,
     BranchType,
     BlobFilter,
@@ -793,6 +795,13 @@ class Repository:
     def descendant_of(self, oid1: _OidArg, oid2: _OidArg) -> bool: ...
     def expand_id(self, hex: str) -> Oid: ...
     def free(self) -> None: ...
+    def get_attr(
+        self,
+        path: str | bytes | Path,
+        name: str | bytes,
+        flags: AttrCheck = AttrCheck.FILE_THEN_INDEX,
+        commit: _OidArg | None = None,
+    ) -> bool | None | str: ...
     def git_object_lookup_prefix(self, oid: _OidArg) -> Object: ...
     def list_worktrees(self) -> list[str]: ...
     def listall_branches(self, flag: BranchType = BranchType.LOCAL) -> list[str]: ...

test/test_attributes.py

@@ -26,8 +26,10 @@
 # Standard Library
 from pathlib import Path
 
+from pygit2 import Repository
 
-def test_no_attr(testrepo):
+
+def test_no_attr(testrepo: Repository) -> None:
     assert testrepo.get_attr('file', 'foo') is None
 
     with (Path(testrepo.workdir) / '.gitattributes').open('w+') as f:
@@ -41,7 +43,7 @@ def test_no_attr(testrepo):
     assert 'lf' == testrepo.get_attr('file.sh', 'eol')
 
 
-def test_no_attr_aspath(testrepo):
+def test_no_attr_aspath(testrepo: Repository) -> None:
     with (Path(testrepo.workdir) / '.gitattributes').open('w+') as f:
         print('*.py  text\n', file=f)