~mdw / stgit / blame
| Commit | Line | Data |
|---|---|---|
| 170f576b CM |
1 | """Access to the GIT base directory |
| 2 | """ | |
| 3 | ||
| 4 | __copyright__ = """ | |
| 5 | Copyright (C) 2006, Catalin Marinas <catalin.marinas@gmail.com> | |
| 6 | ||
| 7 | This program is free software; you can redistribute it and/or modify | |
| 8 | it under the terms of the GNU General Public License version 2 as | |
| 9 | published by the Free Software Foundation. | |
| 10 | ||
| 11 | This program is distributed in the hope that it will be useful, | |
| 12 | but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 13 | MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 14 | GNU General Public License for more details. | |
| 15 | ||
| 16 | You should have received a copy of the GNU General Public License | |
| 17 | along with this program; if not, write to the Free Software | |
| 18 | Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
| 19 | """ | |
| 20 | ||
| 21 | import os | |
| 22 | ||
| 23 | class BaseDirException(Exception): | |
| 24 | pass | |
| 25 | ||
| 26 | def __output(cmd): | |
| 27 | f = os.popen(cmd, 'r') | |
| 28 | string = f.readline().rstrip() | |
| 29 | if f.close(): | |
| 30 | raise BaseDirException, 'Error: failed to execute "%s"' % cmd | |
| 31 | return string | |
| 32 | ||
| 33 | # GIT_DIR value cached | |
| 34 | __base_dir = None | |
| 35 | ||
| 36 | def get(): | |
| 37 | """Return the .git directory location | |
| 38 | """ | |
| 39 | global __base_dir | |
| 40 | ||
| 41 | if not __base_dir: | |
| 42 | if 'GIT_DIR' in os.environ: | |
| 43 | __base_dir = os.environ['GIT_DIR'] | |
| 44 | else: | |
| 45 | __base_dir = __output('git-rev-parse --git-dir') | |
| 46 | ||
| 47 | return __base_dir |