用于确定目录是否为git存储库的Python脚本

Question

我刚开始使用Python,我的第一个任务是编写一个脚本来确定运行它的目录是否是一个git存储库.一位同学建议使用以下代码:

#! /usr/bin/env python

from subprocess import Popen, PIPE, STDOUT
if Popen(("git", "branch"), stderr=STDOUT, stdout=PIPE).returncode != 0:
    print("Nope!")
else:
    print("Yup!")

它应该打印输出,具体取决于控制台命令"git branch"的返回码.但是,该脚本在存储库中不起作用.

无论如何,对于这方面的任何建议,我将不胜感激.

作业还包括:

• 能够在Windows上使用相同的脚本
• 最终传递路径以确定脚本而不必将其复制到目标目录

非常感谢!

Answer 1

安装gitpython,例如pip install gitpython.

然后创建一个这样的函数:

import git

...

def is_git_repo(path):
    try:
        _ = git.Repo(path).git_dir
        return True
    except git.exc.InvalidGitRepositoryError:
        return False

Answer 2

关！Popen是一个更复杂的对象，它启动一个进程，但需要其他交互才能获取信息。在您的情况下，您需要调用wait（），以便Popen对象等待程序完成以获取返回代码。如果命令返回的信息太多而无法放入管道中，则还可能使程序挂起。尝试“调用”（调用将等待您）并将命令输出发送到位存储桶。

#! /usr/bin/env python

from subprocess import call, STDOUT
import os
if call(["git", "branch"], stderr=STDOUT, stdout=open(os.devnull, 'w')) != 0:
    print("Nope!")
else:
    print("Yup!")

Answer 3

尽管tdelaney的答案是正确的，但我想发布一个更通用的函数，该函数可以快速复制粘贴到某人的脚本中：

该功能有两个要求：

import os
import subprocess

而且功能很简单：

def is_git_directory(path = '.'):
    return subprocess.call(['git', '-C', path, 'status'], stderr=subprocess.STDOUT, stdout = open(os.devnull, 'w')) == 0