scmrepo 3.3.4__py3-none-any.whl → 3.3.5__py3-none-any.whl

scmrepo/git/backend/pygit2/__init__.py

@@ -34,6 +34,7 @@ logger = logging.getLogger(__name__)
 if TYPE_CHECKING:
     from pygit2 import Commit, Oid, Signature
     from pygit2.config import Config as _Pygit2Config
+    from pygit2.enums import CheckoutStrategy
     from pygit2.remotes import Remote
     from pygit2.repository import Repository
 
@@ -246,17 +247,15 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         )
 
     @staticmethod
-    def _get_checkout_strategy(strategy: Optional[int] = None):
-        from pygit2 import (
-            GIT_CHECKOUT_RECREATE_MISSING,
-            GIT_CHECKOUT_SAFE,
-            GIT_CHECKOUT_SKIP_LOCKED_DIRECTORIES,
-        )
+    def _get_checkout_strategy(
+        strategy: Optional["CheckoutStrategy"] = None,
+    ) -> "CheckoutStrategy":
+        from pygit2.enums import CheckoutStrategy
 
         if strategy is None:
-            strategy = GIT_CHECKOUT_SAFE | GIT_CHECKOUT_RECREATE_MISSING
+            strategy = CheckoutStrategy.SAFE | CheckoutStrategy.RECREATE_MISSING
         if os.name == "nt":
-            strategy |= GIT_CHECKOUT_SKIP_LOCKED_DIRECTORIES
+            strategy |= CheckoutStrategy.SKIP_LOCKED_DIRECTORIES
         return strategy
 
     # Workaround to force git_backend_odb_pack to release open file handles
@@ -343,9 +342,12 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         force: bool = False,
         **kwargs,
     ):
-        from pygit2 import GIT_CHECKOUT_FORCE, GitError
+        from pygit2 import GitError
+        from pygit2.enums import CheckoutStrategy
 
-        strategy = self._get_checkout_strategy(GIT_CHECKOUT_FORCE if force else None)
+        strategy = self._get_checkout_strategy(
+            CheckoutStrategy.FORCE if force else None
+        )
 
         with self.release_odb_handles():
             if create_new:
@@ -613,7 +615,7 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         force: bool = False,
         on_diverged: Optional[Callable[[str, str], bool]] = None,
     ) -> SyncStatus:
-        import pygit2
+        from pygit2.enums import MergeAnalysis
 
         rh_rev = self.resolve_rev(rh)
 
@@ -627,16 +629,16 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
             self.set_ref(lh, rh_rev)
             return SyncStatus.SUCCESS
 
-        if merge_result & pygit2.GIT_MERGE_ANALYSIS_UP_TO_DATE:
+        if merge_result & MergeAnalysis.UP_TO_DATE:
             return SyncStatus.UP_TO_DATE
-        if merge_result & pygit2.GIT_MERGE_ANALYSIS_FASTFORWARD:
+        if merge_result & MergeAnalysis.FASTFORWARD:
             self.set_ref(lh, rh_rev)
             return SyncStatus.SUCCESS
-        if merge_result & pygit2.GIT_MERGE_ANALYSIS_NORMAL:
+        if merge_result & MergeAnalysis.NORMAL:
             if on_diverged and on_diverged(lh, rh_rev):
                 return SyncStatus.SUCCESS
             return SyncStatus.DIVERGED
-        logger.debug("Unexpected merge result: %s", pygit2.GIT_MERGE_ANALYSIS_NORMAL)
+        logger.debug("Unexpected merge result: %s", MergeAnalysis.NORMAL)
         raise SCMError("Unknown merge analysis result")
 
     @contextmanager
@@ -779,7 +781,8 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         skip_conflicts: bool = False,
         **kwargs,
     ):
-        from pygit2 import GIT_CHECKOUT_ALLOW_CONFLICTS, GitError
+        from pygit2 import GitError
+        from pygit2.enums import CheckoutStrategy
 
         from scmrepo.git import Stash
 
@@ -788,7 +791,7 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
                 self.repo.index.read(False)
                 strategy = self._get_checkout_strategy()
                 if skip_conflicts:
-                    strategy |= GIT_CHECKOUT_ALLOW_CONFLICTS
+                    strategy |= CheckoutStrategy.ALLOW_CONFLICTS
                 self.repo.stash_apply(
                     index, strategy=strategy, reinstate_index=reinstate_index
                 )
@@ -834,7 +837,8 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         raise NotImplementedError
 
     def reset(self, hard: bool = False, paths: Optional[Iterable[str]] = None):
-        from pygit2 import GIT_RESET_HARD, GIT_RESET_MIXED, IndexEntry
+        from pygit2 import IndexEntry
+        from pygit2.enums import ResetMode
 
         self.repo.index.read(False)
         if paths is not None:
@@ -847,9 +851,9 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
                 self.repo.index.add(IndexEntry(rel, obj.id, obj.filemode))
             self.repo.index.write()
         elif hard:
-            self.repo.reset(self.repo.head.target, GIT_RESET_HARD)
+            self.repo.reset(self.repo.head.target, ResetMode.HARD)
         else:
-            self.repo.reset(self.repo.head.target, GIT_RESET_MIXED)
+            self.repo.reset(self.repo.head.target, ResetMode.MIXED)
 
     def checkout_index(
         self,
@@ -858,22 +862,17 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         ours: bool = False,
         theirs: bool = False,
     ):
-        from pygit2 import (
-            GIT_CHECKOUT_ALLOW_CONFLICTS,
-            GIT_CHECKOUT_FORCE,
-            GIT_CHECKOUT_RECREATE_MISSING,
-            GIT_CHECKOUT_SAFE,
-        )
+        from pygit2.enums import CheckoutStrategy
 
         assert not (ours and theirs)
-        strategy = GIT_CHECKOUT_RECREATE_MISSING
+        strategy = CheckoutStrategy.RECREATE_MISSING
         if force or ours or theirs:
-            strategy |= GIT_CHECKOUT_FORCE
+            strategy |= CheckoutStrategy.FORCE
         else:
-            strategy |= GIT_CHECKOUT_SAFE
+            strategy |= CheckoutStrategy.SAFE
 
         if ours or theirs:
-            strategy |= GIT_CHECKOUT_ALLOW_CONFLICTS
+            strategy |= CheckoutStrategy.ALLOW_CONFLICTS
         strategy = self._get_checkout_strategy(strategy)
 
         index = self.repo.index
@@ -910,18 +909,7 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
     def status(
         self, ignored: bool = False, untracked_files: str = "all"
     ) -> tuple[Mapping[str, Iterable[str]], Iterable[str], Iterable[str]]:
-        from pygit2 import (
-            GIT_STATUS_IGNORED,
-            GIT_STATUS_INDEX_DELETED,
-            GIT_STATUS_INDEX_MODIFIED,
-            GIT_STATUS_INDEX_NEW,
-            GIT_STATUS_WT_DELETED,
-            GIT_STATUS_WT_MODIFIED,
-            GIT_STATUS_WT_NEW,
-            GIT_STATUS_WT_RENAMED,
-            GIT_STATUS_WT_TYPECHANGE,
-            GIT_STATUS_WT_UNREADABLE,
-        )
+        from pygit2.enums import FileStatus
 
         staged: Mapping[str, list[str]] = {
             "add": [],
@@ -932,19 +920,19 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         untracked: list[str] = []
 
         states = {
-            GIT_STATUS_WT_NEW: untracked,
-            GIT_STATUS_WT_MODIFIED: unstaged,
-            GIT_STATUS_WT_TYPECHANGE: staged["modify"],
-            GIT_STATUS_WT_DELETED: staged["modify"],
-            GIT_STATUS_WT_RENAMED: staged["modify"],
-            GIT_STATUS_INDEX_NEW: staged["add"],
-            GIT_STATUS_INDEX_MODIFIED: staged["modify"],
-            GIT_STATUS_INDEX_DELETED: staged["delete"],
-            GIT_STATUS_WT_UNREADABLE: untracked,
+            FileStatus.WT_NEW: untracked,
+            FileStatus.WT_MODIFIED: unstaged,
+            FileStatus.WT_TYPECHANGE: staged["modify"],
+            FileStatus.WT_DELETED: staged["modify"],
+            FileStatus.WT_RENAMED: staged["modify"],
+            FileStatus.INDEX_NEW: staged["add"],
+            FileStatus.INDEX_MODIFIED: staged["modify"],
+            FileStatus.INDEX_DELETED: staged["delete"],
+            FileStatus.WT_UNREADABLE: untracked,
         }
 
         if untracked_files != "no" and ignored:
-            states[GIT_STATUS_IGNORED] = untracked
+            states[FileStatus.IGNORED] = untracked
 
         for file, state in self.repo.status(
             untracked_files=untracked_files, ignored=ignored
@@ -963,22 +951,15 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
     def iter_remote_refs(self, url: str, base: Optional[str] = None, **kwargs):
         raise NotImplementedError
 
-    def merge(  # noqa: C901
+    def merge(  # noqa: C901, PLR0912
         self,
         rev: str,
         commit: bool = True,
         msg: Optional[str] = None,
         squash: bool = False,
     ) -> Optional[str]:
-        from pygit2 import (
-            GIT_MERGE_ANALYSIS_FASTFORWARD,
-            GIT_MERGE_ANALYSIS_NONE,
-            GIT_MERGE_ANALYSIS_UNBORN,
-            GIT_MERGE_ANALYSIS_UP_TO_DATE,
-            GIT_MERGE_PREFERENCE_FASTFORWARD_ONLY,
-            GIT_MERGE_PREFERENCE_NO_FASTFORWARD,
-            GitError,
-        )
+        from pygit2 import GitError
+        from pygit2.enums import MergeAnalysis, MergePreference
 
         if commit and squash:
             raise SCMError("Cannot merge with 'squash' and 'commit'")
@@ -991,9 +972,9 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
             except GitError as exc:
                 raise SCMError("Merge analysis failed") from exc
 
-            if analysis == GIT_MERGE_ANALYSIS_NONE:
+            if analysis == MergeAnalysis.NONE:
                 raise SCMError(f"'{rev}' cannot be merged into HEAD")
-            if analysis & GIT_MERGE_ANALYSIS_UP_TO_DATE:
+            if analysis & MergeAnalysis.UP_TO_DATE:
                 return None
 
             try:
@@ -1006,15 +987,15 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
                 raise MergeConflictError("Merge contained conflicts")
 
             try:
-                if not (squash or ff_pref & GIT_MERGE_PREFERENCE_NO_FASTFORWARD):
-                    if analysis & GIT_MERGE_ANALYSIS_FASTFORWARD:
+                if not (squash or ff_pref & MergePreference.NO_FASTFORWARD):
+                    if analysis & MergeAnalysis.FASTFORWARD:
                         return self._merge_ff(rev, obj)
 
-                    if analysis & GIT_MERGE_ANALYSIS_UNBORN:
+                    if analysis & MergeAnalysis.UNBORN:
                         self.repo.set_head(obj.id)
                         return str(obj.id)
 
-                if ff_pref & GIT_MERGE_PREFERENCE_FASTFORWARD_ONLY:
+                if ff_pref & MergePreference.FASTFORWARD_ONLY:
                     raise SCMError(f"Cannot fast-forward HEAD to '{rev}'")
 
                 if commit:
@@ -1105,19 +1086,15 @@ class Pygit2Backend(BaseGitBackend):  # pylint:disable=abstract-method
         attr: str,
         source: Optional[str] = None,
     ) -> Optional[Union[bool, str]]:
-        from pygit2 import (
-            GIT_ATTR_CHECK_FILE_THEN_INDEX,
-            GIT_ATTR_CHECK_INCLUDE_COMMIT,
-            GIT_ATTR_CHECK_INDEX_ONLY,
-            GitError,
-        )
+        from pygit2 import GitError
+        from pygit2.enums import AttrCheck
 
         commit: Optional["Commit"] = None
-        flags = GIT_ATTR_CHECK_FILE_THEN_INDEX
+        flags = AttrCheck.FILE_THEN_INDEX
         if source:
             try:
                 commit, _ref = self._resolve_refish(source)
-                flags = GIT_ATTR_CHECK_INDEX_ONLY | GIT_ATTR_CHECK_INCLUDE_COMMIT
+                flags = AttrCheck.INDEX_ONLY | AttrCheck.INCLUDE_COMMIT
             except (KeyError, GitError) as exc:
                 raise SCMError(f"Invalid commit '{source}'") from exc
         try:

{scmrepo-3.3.4.dist-info → scmrepo-3.3.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: scmrepo
-Version: 3.3.4
+Version: 3.3.5
 Summary: scmrepo
 Author-email: Iterative <support@dvc.org>
 License: Apache-2.0

{scmrepo-3.3.4.dist-info → scmrepo-3.3.5.dist-info}/RECORD

@@ -19,7 +19,7 @@ scmrepo/git/backend/gitpython.py,sha256=RTDNUVLiJmhcRgdaJpApTQi6sMN-an9P9xKZBKwk
 scmrepo/git/backend/dulwich/__init__.py,sha256=WT6B8LKzQN4pgNrA82CFqwKkR1wWVBRHwwA9LwuSHV4,34740
 scmrepo/git/backend/dulwich/asyncssh_vendor.py,sha256=y0mvx3tRR99Nv85e2n39nPYlhme8dplmQQSbSpx7FDM,11562
 scmrepo/git/backend/dulwich/client.py,sha256=bcDroljSvNz6s5WWv9UVvZHKkOJOVTK_zU7YCq62TN4,2360
-scmrepo/git/backend/pygit2/__init__.py,sha256=7Jvhp9mxOhTit9SPbaw6mc4VXFD3n8J_ZzUGypKMBG0,37132
+scmrepo/git/backend/pygit2/__init__.py,sha256=qquYBHB-gpW-AeDjO5OqSUpRKBhfLvGHFy-Od35g0DA,36356
 scmrepo/git/backend/pygit2/callbacks.py,sha256=Ky4YmUPhv9xjU_44ypBYIcaVHJixzaGb6t9HIeUmBP4,2751
 scmrepo/git/backend/pygit2/filter.py,sha256=2NlWfQ7soXN1H7Es6-LctE74hpj3QKQTlYqXRH83VpM,2128
 scmrepo/git/lfs/__init__.py,sha256=at5blRIKnKpg_g5dLRDsGWBFi6SbucRlF_DX6aAkGtE,257
@@ -31,8 +31,8 @@ scmrepo/git/lfs/pointer.py,sha256=BcVbtjoOUG9cEzyJSJDeweqehGZvq43P6NNLDYUGYEI,31
 scmrepo/git/lfs/progress.py,sha256=ELlBs2SeXhAcnPDN23w3FTeBRgB9RGqBD2CFMS6n9Xs,4750
 scmrepo/git/lfs/smudge.py,sha256=1O_fznptWo4CKXqcJgUoWP6cgWWhvGAZ3d87kasG3cQ,1610
 scmrepo/git/lfs/storage.py,sha256=LbHGhSRdORu4Iqzb7Ef-2tA_iJbZykXZFotWRJvyr2Y,2380
-scmrepo-3.3.4.dist-info/LICENSE,sha256=-1jhbPjoIVHR0cEgahL4Zhct75Ff4MzYCR_jOaJDPq8,11340
-scmrepo-3.3.4.dist-info/METADATA,sha256=OLF86Fy_0tg1dtbPC-VAIpdXoVRmuRA89sOb_EwoVOE,4731
-scmrepo-3.3.4.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-scmrepo-3.3.4.dist-info/top_level.txt,sha256=iunjod6w3GogERsAYfLRupnANXnqzX3jbIfbeIQG5cc,8
-scmrepo-3.3.4.dist-info/RECORD,,
+scmrepo-3.3.5.dist-info/LICENSE,sha256=-1jhbPjoIVHR0cEgahL4Zhct75Ff4MzYCR_jOaJDPq8,11340
+scmrepo-3.3.5.dist-info/METADATA,sha256=fS-xs4RY1ybtzHNycu6upzY6p4GJD2SN9-ExeS-sFaw,4731
+scmrepo-3.3.5.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
+scmrepo-3.3.5.dist-info/top_level.txt,sha256=iunjod6w3GogERsAYfLRupnANXnqzX3jbIfbeIQG5cc,8
+scmrepo-3.3.5.dist-info/RECORD,,