Răsfoiți Sursa

Add more typing.

Jelmer Vernooij 2 ani în urmă
părinte
comite
f901e842f3
1 a modificat fișierele cu 13 adăugiri și 7 ștergeri
  1. 13 7
      dulwich/refs.py

+ 13 - 7
dulwich/refs.py

@@ -511,12 +511,12 @@ class DictRefsContainer(RefsContainer):
 
 
     def add_if_new(
     def add_if_new(
         self,
         self,
-        name,
-        ref,
+        name: bytes,
+        ref: bytes,
         committer=None,
         committer=None,
         timestamp=None,
         timestamp=None,
         timezone=None,
         timezone=None,
-        message=None,
+        message: Optional[bytes] = None,
     ):
     ):
         if name in self._refs:
         if name in self._refs:
             return False
             return False
@@ -883,12 +883,12 @@ class DiskRefsContainer(RefsContainer):
 
 
     def add_if_new(
     def add_if_new(
         self,
         self,
-        name,
-        ref,
+        name: bytes,
+        ref: bytes,
         committer=None,
         committer=None,
         timestamp=None,
         timestamp=None,
         timezone=None,
         timezone=None,
-        message=None,
+        message: Optional[bytes] = None,
     ):
     ):
         """Add a new reference only if it does not already exist.
         """Add a new reference only if it does not already exist.
 
 
@@ -1137,7 +1137,11 @@ def _set_origin_head(refs, origin, origin_head):
             refs.set_symbolic_ref(origin_ref, target_ref)
             refs.set_symbolic_ref(origin_ref, target_ref)
 
 
 
 
-def _set_default_branch(refs, origin, origin_head, branch, ref_message):
+def _set_default_branch(
+        refs: RefsContainer, origin: bytes, origin_head: bytes, branch: bytes,
+        ref_message: Optional[bytes]) -> bytes:
+    """Set the default branch.
+    """
     origin_base = b"refs/remotes/" + origin + b"/"
     origin_base = b"refs/remotes/" + origin + b"/"
     if branch:
     if branch:
         origin_ref = origin_base + branch
         origin_ref = origin_base + branch
@@ -1165,6 +1169,8 @@ def _set_default_branch(refs, origin, origin_head, branch, ref_message):
             )
             )
         except KeyError:
         except KeyError:
             pass
             pass
+    else:
+        raise ValueError('neither origin_head nor branch are provided')
     return head_ref
     return head_ref