如何在Snowflake Workspace的Python中获取当前Git分支（无.git文件）

可以实现，Snowflake Workspace会通过内部环境变量或官方API暴露Git分支信息，无需直接访问.git文件夹（该文件夹可能被Snowflake隐藏或存放在内部路径中）。以下是具体方案：

方法1：读取Snowflake内置环境变量 #

Snowflake在Workspace运行代码时，会自动注入Git相关的环境变量，直接读取即可：

import os
# 获取当前分支名
branch_name = os.getenv("SNOWFLAKE_GIT_BRANCH")
print(branch_name)

如果不确定变量名，可先打印所有含Git关键字的环境变量排查：

import os
for key, value in os.environ.items():
    if "GIT" in key.upper():
        print(f"{key}: {value}")

方法2：通过Snowpark API获取仓库信息 #

使用Snowflake官方的Snowpark会话对象，直接查询当前关联的Git仓库详情：

from snowflake.snowpark import Session

# 初始化Snowpark会话（需提前配置好连接参数）
session = Session.builder.getOrCreate()
# 获取当前Workspace绑定的Git仓库信息
repo_info = session.get_current_git_repository()
# 提取分支名
branch_name = repo_info.get("BRANCH")
print(branch_name)

注意：此方法需要会话具备访问Git仓库元数据的权限。

可选：查找隐藏的.git文件夹（非推荐） #

若确需定位.git文件夹，可通过Python遍历目录查找（Snowflake可能将其挂载在非根路径下）：

import os

def search_git_directory(start_path):
    for root, dirs, files in os.walk(start_path):
        if '.git' in dirs:
            return os.path.join(root, '.git')
    return None

# 从当前目录开始递归查找
git_dir_path = search_git_directory('.')
print(git_dir_path)

但Snowflake通常会隐藏内部Git文件结构，优先使用前两种方法更稳定。

内容的提问来源于stack exchange，提问作者Evgeniya Garkova