如何使用dulwich库获取Git当前分支名称

我之前也碰到过这个问题——Dulwich的文档确实有些地方不够直白，找这个功能花了好一会儿功夫。不过其实有几个简单可靠的方法可以实现，下面给你详细说：

方法一：直接解析HEAD引用 #

这是最直接的方式，利用Repo对象的head()方法获取当前HEAD指向的引用，再提取分支名：

from dulwich.repo import Repo

# 初始化仓库对象，这里用当前目录，你也可以指定仓库路径
repo = Repo(".")

# 获取HEAD指向的引用（字节串格式）
head_ref = repo.head()

# 判断是否处于分支状态（而非detached HEAD）
if head_ref.startswith(b'refs/heads/'):
    # 解码为字符串后，分割出最后一部分就是分支名
    current_branch = head_ref.decode("utf-8").split("/")[-1]
    print(f"当前分支：{current_branch}")
else:
    # 如果是detached HEAD，会直接返回commit的哈希值
    print(f"处于detached HEAD状态，当前Commit哈希：{head_ref.decode('utf-8')}")

方法二：通过分支引用匹配HEAD的Commit哈希 #

如果需要更严谨地确认分支与HEAD的对应关系，可以遍历本地分支，找到与HEAD指向相同Commit的分支：

from dulwich.repo import Repo

repo = Repo(".")
head_sha = repo.head()

# 获取所有本地分支的引用（以b'refs/heads/xxx'形式存在）
local_branches = [ref for ref in repo.refs.keys() if ref.startswith(b'refs/heads/')]

current_branch = None
for ref in local_branches:
    # 对比分支指向的Commit哈希与HEAD的哈希
    if repo.refs[ref] == repo[head_sha]:
        current_branch = ref.decode("utf-8").split("/")[-1]
        break

if current_branch:
    print(f"当前分支：{current_branch}")
else:
    print(f"处于detached HEAD状态，当前Commit哈希：{head_sha.decode('utf-8')}")

这两种方法都能解决问题，第一种更简洁高效，适合大多数场景；第二种在一些特殊情况（比如HEAD指向的Commit同时被多个分支引用）下能帮你明确对应关系。

内容的提问来源于stack exchange，提问作者Dull Bananas