1 """Python GIT interface
5 Copyright (C) 2005, Catalin Marinas <catalin.marinas@gmail.com>
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.
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.
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
21 import sys
, os
, glob
, popen2
23 from stgit
.utils
import *
26 class GitException(Exception):
30 # Different start-up variables read from the environment
31 if 'GIT_DIR' in os
.environ
:
32 base_dir
= os
.environ
['GIT_DIR']
36 head_link
= os
.path
.join(base_dir
, 'HEAD')
42 """Handle the commit objects
44 def __init__(self
, id_hash
):
45 self
.__id_hash
= id_hash
47 lines
= _output_lines('git-cat-file commit %s' % id_hash
)
49 for i
in range(len(lines
)):
53 field
= line
.strip().split(' ', 1)
54 if field
[0] == 'tree':
55 self
.__tree
= field
[1]
56 elif field
[0] == 'parent':
57 self
.__parents
.append(field
[1])
58 if field
[0] == 'author':
59 self
.__author
= field
[1]
60 if field
[0] == 'committer':
61 self
.__committer
= field
[1]
62 self
.__log
= ''.join(lines
[i
+1:])
64 def get_id_hash(self
):
71 return self
.__parents
[0]
73 def get_parents(self
):
79 def get_committer(self
):
80 return self
.__committer
85 # dictionary of Commit objects, used to avoid multiple calls to git
91 def get_commit(id_hash
):
92 """Commit objects factory. Save/look-up them in the __commits
97 if id_hash
in __commits
:
98 return __commits
[id_hash
]
100 commit
= Commit(id_hash
)
101 __commits
[id_hash
] = commit
105 """Return the list of file conflicts
107 conflicts_file
= os
.path
.join(base_dir
, 'conflicts')
108 if os
.path
.isfile(conflicts_file
):
109 f
= file(conflicts_file
)
110 names
= [line
.strip() for line
in f
.readlines()]
116 def _input(cmd
, file_desc
):
117 p
= popen2
.Popen3(cmd
)
119 line
= file_desc
.readline()
122 p
.tochild
.write(line
)
125 raise GitException
, '%s failed' % str
(cmd
)
129 string
= p
.fromchild
.read()
131 raise GitException
, '%s failed' % str
(cmd
)
134 def _output_one_line(cmd
, file_desc
= None):
136 if file_desc
!= None:
137 for line
in file_desc
:
138 p
.tochild
.write(line
)
140 string
= p
.fromchild
.readline().strip()
142 raise GitException
, '%s failed' % str
(cmd
)
145 def _output_lines(cmd
):
147 lines
= p
.fromchild
.readlines()
149 raise GitException
, '%s failed' % str
(cmd
)
152 def __run(cmd
, args
=None):
153 """__run: runs cmd using spawnvp.
155 Runs cmd using spawnvp. The shell is avoided so it won't mess up
156 our arguments. If args is very large, the command is run multiple
157 times; args is split xargs style: cmd is passed on each
158 invocation. Unlike xargs, returns immediately if any non-zero
159 return code is received.
165 for i
in range(0, len(args
)+1, 100):
166 r
=os
.spawnvp(os
.P_WAIT
, args_l
[0], args_l
+ args
[i
:min(i
+100, len(args
))])
171 def __check_base_dir():
172 return os
.path
.isdir(base_dir
)
174 def __tree_status(files
= [], tree_id
= 'HEAD', unknown
= False,
176 """Returns a list of pairs - [status, filename]
178 os
.system('git-update-index --refresh > /dev/null')
184 exclude_file
= os
.path
.join(base_dir
, 'info', 'exclude')
185 base_exclude
= ['--exclude=%s' % s for s in
186 ['*.[ao]', '*.pyc', '.*', '*~', '#*', 'TAGS', 'tags']]
187 base_exclude
.append('--exclude-per-directory=.gitignore')
189 if os
.path
.exists(exclude_file
):
190 extra_exclude
= ['--exclude-from=%s' % exclude_file
]
194 extra_exclude
= base_exclude
= []
196 lines
= _output_lines(['git-ls-files', '--others'] + base_exclude
198 cache_files
+= [('?', line
.strip()) for line
in lines
]
201 conflicts
= get_conflicts()
204 cache_files
+= [('C', filename
) for filename
in conflicts
]
207 for line
in _output_lines(['git-diff-index', '-r', tree_id
] + files
):
208 fs
= tuple(line
.rstrip().split(' ',4)[-1].split('\t',1))
209 if fs
[1] not in conflicts
:
210 cache_files
.append(fs
)
215 """Return true if there are local changes in the tree
217 return len(__tree_status()) != 0
223 """Verifies the HEAD and returns the SHA1 id that represents it
228 __head
= rev_parse('HEAD')
232 """Returns the name of the file pointed to by the HEAD link
234 return os
.path
.basename(_output_one_line('git-symbolic-ref HEAD'))
237 """Sets the HEAD value
241 if not __head
or __head
!= val
:
242 if __run('git-update-ref HEAD', [val
]) != 0:
243 raise GitException
, 'Could not update HEAD to "%s".' % val
246 def __clear_head_cache():
247 """Sets the __head to None so that a re-read is forced
253 def rev_parse(git_id
):
254 """Parse the string and return a verified SHA1 id
257 return _output_one_line(['git-rev-parse', '--verify', git_id
])
259 raise GitException
, 'Unknown revision: %s' % git_id
262 """Add the files or recursively add the directory contents
264 # generate the file list
267 if not os
.path
.exists(i
):
268 raise GitException
, 'Unknown file or directory: %s' % i
271 # recursive search. We only add files
272 for root
, dirs
, local_files
in os
.walk(i
):
273 for name
in [os
.path
.join(root
, f
) for f
in local_files
]:
274 if os
.path
.isfile(name
):
275 files
.append(os
.path
.normpath(name
))
276 elif os
.path
.isfile(i
):
277 files
.append(os
.path
.normpath(i
))
279 raise GitException
, '%s is not a file or directory' % i
282 if __run('git-update-index --add --', files
):
283 raise GitException
, 'Unable to add file'
285 def rm(files
, force
= False):
286 """Remove a file from the repository
289 git_opt
= '--force-remove'
295 if os
.path
.exists(f
):
296 raise GitException
, '%s exists. Remove it first' %f
298 __run('git-update-index --remove --', files
)
301 __run('git-update-index --force-remove --', files
)
303 def update_cache(files
= [], force
= False):
304 """Update the cache information for the given files
306 cache_files
= __tree_status(files
)
308 # everything is up-to-date
309 if len(cache_files
) == 0:
312 # check for unresolved conflicts
313 if not force
and [x
for x
in cache_files
314 if x
[0] not in ['M', 'N', 'A', 'D']]:
315 raise GitException
, 'Updating cache failed: unresolved conflicts'
318 add_files
= [x
[1] for x
in cache_files
if x
[0] in ['N', 'A']]
319 rm_files
= [x
[1] for x
in cache_files
if x
[0] in ['D']]
320 m_files
= [x
[1] for x
in cache_files
if x
[0] in ['M']]
322 if add_files
and __run('git-update-index --add --', add_files
) != 0:
323 raise GitException
, 'Failed git-update-index --add'
324 if rm_files
and __run('git-update-index --force-remove --', rm_files
) != 0:
325 raise GitException
, 'Failed git-update-index --rm'
326 if m_files
and __run('git-update-index --', m_files
) != 0:
327 raise GitException
, 'Failed git-update-index'
331 def commit(message
, files
= [], parents
= [], allowempty
= False,
332 cache_update
= True, tree_id
= None,
333 author_name
= None, author_email
= None, author_date
= None,
334 committer_name
= None, committer_email
= None):
335 """Commit the current tree to repository
337 # Get the tree status
338 if cache_update
and parents
!= []:
339 changes
= update_cache(files
)
340 if not changes
and not allowempty
:
341 raise GitException
, 'No changes to commit'
343 # get the commit message
344 if message
[-1:] != '\n':
348 # write the index to repository
350 tree_id
= _output_one_line('git-write-tree')
357 cmd
+= 'GIT_AUTHOR_NAME="%s" ' % author_name
359 cmd
+= 'GIT_AUTHOR_EMAIL="%s" ' % author_email
361 cmd
+= 'GIT_AUTHOR_DATE="%s" ' % author_date
363 cmd
+= 'GIT_COMMITTER_NAME="%s" ' % committer_name
365 cmd
+= 'GIT_COMMITTER_EMAIL="%s" ' % committer_email
366 cmd
+= 'git-commit-tree %s' % tree_id
372 commit_id
= _output_one_line(cmd
, message
)
374 __set_head(commit_id
)
378 def apply_diff(rev1
, rev2
):
379 """Apply the diff between rev1 and rev2 onto the current
380 index. This function doesn't need to raise an exception since it
381 is only used for fast-pushing a patch. If this operation fails,
382 the pushing would fall back to the three-way merge.
384 return os
.system('git-diff-tree -p %s %s | git-apply --index 2> /dev/null'
387 def merge(base
, head1
, head2
):
388 """Perform a 3-way merge between base, head1 and head2 into the
391 if __run('git-read-tree -u -m', [base
, head1
, head2
]) != 0:
392 raise GitException
, 'git-read-tree failed (local changes maybe?)'
394 # this can fail if there are conflicts
395 if os
.system('git-merge-index -o -q gitmergeonefile.py -a') != 0:
396 raise GitException
, 'git-merge-cache failed (possible conflicts)'
398 def status(files
= [], modified
= False, new
= False, deleted
= False,
399 conflict
= False, unknown
= False, noexclude
= False):
400 """Show the tree status
402 cache_files
= __tree_status(files
, unknown
= True, noexclude
= noexclude
)
403 all
= not (modified
or new
or deleted
or conflict
or unknown
)
418 cache_files
= [x
for x
in cache_files
if x
[0] in filestat
]
420 for fs
in cache_files
:
422 print '%s %s' %
(fs
[0], fs
[1])
426 def diff(files
= [], rev1
= 'HEAD', rev2
= None, out_fd
= None):
427 """Show the diff between rev1 and rev2
431 diff_str
= _output(['git-diff-tree', '-p', rev1
, rev2
] + files
)
433 os
.system('git-update-index --refresh > /dev/null')
434 diff_str
= _output(['git-diff-index', '-p', rev1
] + files
)
437 out_fd
.write(diff_str
)
441 def diffstat(files
= [], rev1
= 'HEAD', rev2
= None):
442 """Return the diffstat between rev1 and rev2
445 p
=popen2
.Popen3('git-apply --stat')
446 diff(files
, rev1
, rev2
, p
.tochild
)
448 str = p
.fromchild
.read().rstrip()
450 raise GitException
, 'git.diffstat failed'
453 def files(rev1
, rev2
):
454 """Return the files modified between rev1 and rev2
458 for line
in _output_lines('git-diff-tree -r %s %s' %
(rev1
, rev2
)):
459 str += '%s %s\n' %
tuple(line
.rstrip().split(' ',4)[-1].split('\t',1))
463 def barefiles(rev1
, rev2
):
464 """Return the files modified between rev1 and rev2, without status info
468 for line
in _output_lines('git-diff-tree -r %s %s' %
(rev1
, rev2
)):
469 str += '%s\n' % line
.rstrip().split(' ',4)[-1].split('\t',1)[-1]
473 def checkout(files
= [], tree_id
= None, force
= False):
474 """Check out the given or all files
476 if tree_id
and __run('git-read-tree -m', [tree_id
]) != 0:
477 raise GitException
, 'Failed git-read-tree -m %s' % tree_id
479 checkout_cmd
= 'git-checkout-index -q -u'
481 checkout_cmd
+= ' -f'
483 checkout_cmd
+= ' -a'
485 checkout_cmd
+= ' --'
487 if __run(checkout_cmd
, files
) != 0:
488 raise GitException
, 'Failed git-checkout-index'
491 """Switch the tree to the given id
493 if __run('git-read-tree -u -m', [get_head(), tree_id
]) != 0:
494 raise GitException
, 'git-read-tree failed (local changes maybe?)'
498 def reset(tree_id
= None):
499 """Revert the tree changes relative to the given tree_id. It removes
505 cache_files
= __tree_status(tree_id
= tree_id
)
506 rm_files
= [x
[1] for x
in cache_files
if x
[0] in ['D']]
508 checkout(tree_id
= tree_id
, force
= True)
511 # checkout doesn't remove files
512 map(os
.remove
, rm_files
)
514 def pull(repository
= 'origin', refspec
= None):
515 """Pull changes from the remote repository. At the moment, just
516 use the 'git pull' command
518 # 'git pull' updates the HEAD
525 if __run('git pull', args
) != 0:
526 raise GitException
, 'Failed "git pull %s"' % repository
528 def apply_patch(filename
= None, base
= None):
529 """Apply a patch onto the current or given index. There must not
530 be any local changes in the tree, otherwise the command fails
534 return __run('git-apply --index', [filename
]) == 0
537 _input('git-apply --index', sys
.stdin
)
542 os
.system('git-update-index --refresh > /dev/null')
545 orig_head
= get_head()
548 if not __apply_patch():
551 raise GitException
, 'Patch does not apply cleanly'
553 top
= commit(message
= 'temporary commit used for applying a patch',
556 merge(base
, orig_head
, top
)
558 def clone(repository
, local_dir
):
559 """Clone a remote repository. At the moment, just use the
562 if __run('git clone', [repository
, local_dir
]) != 0:
563 raise GitException
, 'Failed "git clone %s %s"' \
564 %
(repository
, local_dir
)