X-Git-Url: https://git.distorted.org.uk/~mdw/stgit/blobdiff_plain/84fcbc3b78f3a0156025d4eb6f6b76df7a5df233..bae29dddde41058567a5ef9b44c160dd76b1c025:/stgit/stack.py diff --git a/stgit/stack.py b/stgit/stack.py index 76da21a..dc7c19f 100644 --- a/stgit/stack.py +++ b/stgit/stack.py @@ -64,13 +64,13 @@ def __clean_comments(f): f.seek(0); f.truncate() f.writelines(lines) -def edit_file(series, string, comment, show_patch = True): +def edit_file(series, line, comment, show_patch = True): fname = '.stgit.msg' - tmpl = os.path.join(git.base_dir, 'patchdescr.tmpl') + tmpl = os.path.join(git.get_base_dir(), 'patchdescr.tmpl') f = file(fname, 'w+') - if string: - print >> f, string + if line: + print >> f, line elif os.path.isfile(tmpl): print >> f, file(tmpl).read().rstrip() else: @@ -88,7 +88,7 @@ def edit_file(series, string, comment, show_patch = True): git.diff([], series.get_patch(series.get_current()).get_bottom(), None, f) #Vim modeline must be near the end. - print >> f, __comment_prefix, 'vi: set textwidth=75 filetype=diff:' + print >> f, __comment_prefix, 'vi: set textwidth=75 filetype=diff nobackup:' f.close() # the editor @@ -108,12 +108,12 @@ def edit_file(series, string, comment, show_patch = True): __clean_comments(f) f.seek(0) - string = f.read() + result = f.read() f.close() os.remove(fname) - return string + return result # # Classes @@ -150,44 +150,50 @@ class Patch: def __get_field(self, name, multiline = False): id_file = os.path.join(self.__dir, name) if os.path.isfile(id_file): - string = read_string(id_file, multiline) - if string == '': + line = read_string(id_file, multiline) + if line == '': return None else: - return string + return line else: return None - def __set_field(self, name, string, multiline = False): + def __set_field(self, name, value, multiline = False): fname = os.path.join(self.__dir, name) - if string and string != '': - write_string(fname, string, multiline) + if value and value != '': + write_string(fname, value, multiline) elif os.path.isfile(fname): os.remove(fname) + def get_old_bottom(self): + return self.__get_field('bottom.old') + def get_bottom(self): return self.__get_field('bottom') - def set_bottom(self, string, backup = False): + def set_bottom(self, value, backup = False): if backup: curr = self.__get_field('bottom') - if curr != string: + if curr != value: self.__set_field('bottom.old', curr) else: self.__set_field('bottom.old', None) - self.__set_field('bottom', string) + self.__set_field('bottom', value) + + def get_old_top(self): + return self.__get_field('top.old') def get_top(self): return self.__get_field('top') - def set_top(self, string, backup = False): + def set_top(self, value, backup = False): if backup: curr = self.__get_field('top') - if curr != string: + if curr != value: self.__set_field('top.old', curr) else: self.__set_field('top.old', None) - self.__set_field('top', string) + self.__set_field('top', value) def restore_old_boundaries(self): bottom = self.__get_field('bottom.old') @@ -203,46 +209,46 @@ class Patch: def get_description(self): return self.__get_field('description', True) - def set_description(self, string): - self.__set_field('description', string, True) + def set_description(self, line): + self.__set_field('description', line, True) def get_authname(self): return self.__get_field('authname') - def set_authname(self, string): - if not string and config.has_option('stgit', 'authname'): - string = config.get('stgit', 'authname') - self.__set_field('authname', string) + def set_authname(self, name): + if not name and config.has_option('stgit', 'authname'): + name = config.get('stgit', 'authname') + self.__set_field('authname', name) def get_authemail(self): return self.__get_field('authemail') - def set_authemail(self, string): - if not string and config.has_option('stgit', 'authemail'): - string = config.get('stgit', 'authemail') - self.__set_field('authemail', string) + def set_authemail(self, address): + if not address and config.has_option('stgit', 'authemail'): + address = config.get('stgit', 'authemail') + self.__set_field('authemail', address) def get_authdate(self): return self.__get_field('authdate') - def set_authdate(self, string): - self.__set_field('authdate', string) + def set_authdate(self, authdate): + self.__set_field('authdate', authdate) def get_commname(self): return self.__get_field('commname') - def set_commname(self, string): - if not string and config.has_option('stgit', 'commname'): - string = config.get('stgit', 'commname') - self.__set_field('commname', string) + def set_commname(self, name): + if not name and config.has_option('stgit', 'commname'): + name = config.get('stgit', 'commname') + self.__set_field('commname', name) def get_commemail(self): return self.__get_field('commemail') - def set_commemail(self, string): - if not string and config.has_option('stgit', 'commemail'): - string = config.get('stgit', 'commemail') - self.__set_field('commemail', string) + def set_commemail(self, address): + if not address and config.has_option('stgit', 'commemail'): + address = config.get('stgit', 'commemail') + self.__set_field('commemail', address) class Series: @@ -257,13 +263,15 @@ class Series: self.__name = git.get_head_file() if self.__name: - self.__patch_dir = os.path.join(git.base_dir, 'patches', + base_dir = git.get_base_dir() + self.__patch_dir = os.path.join(base_dir, 'patches', self.__name) - self.__base_file = os.path.join(git.base_dir, 'refs', 'bases', + self.__base_file = os.path.join(base_dir, 'refs', 'bases', self.__name) self.__applied_file = os.path.join(self.__patch_dir, 'applied') self.__unapplied_file = os.path.join(self.__patch_dir, 'unapplied') self.__current_file = os.path.join(self.__patch_dir, 'current') + self.__descr_file = os.path.join(self.__patch_dir, 'description') def get_branch(self): """Return the branch name for the Series object @@ -314,6 +322,25 @@ class Series: def get_base_file(self): return self.__base_file + def get_protected(self): + return os.path.isfile(os.path.join(self.__patch_dir, 'protected')) + + def protect(self): + protect_file = os.path.join(self.__patch_dir, 'protected') + if not os.path.isfile(protect_file): + create_empty_file(protect_file) + + def unprotect(self): + protect_file = os.path.join(self.__patch_dir, 'protected') + if os.path.isfile(protect_file): + os.remove(protect_file) + + def get_description(self): + if os.path.isfile(self.__descr_file): + return read_string(self.__descr_file) + else: + return '' + def __patch_is_current(self, patch): return patch.get_name() == read_string(self.__current_file) @@ -352,12 +379,17 @@ class Series: return True return git.get_head() == Patch(crt, self.__patch_dir).get_top() + def is_initialised(self): + """Checks if series is already initialised + """ + return os.path.isdir(self.__patch_dir) + def init(self): """Initialises the stgit series """ - bases_dir = os.path.join(git.base_dir, 'refs', 'bases') + bases_dir = os.path.join(git.get_base_dir(), 'refs', 'bases') - if os.path.isdir(self.__patch_dir): + if self.is_initialised(): raise StackException, self.__patch_dir + ' already exists' os.makedirs(self.__patch_dir) @@ -366,8 +398,55 @@ class Series: create_empty_file(self.__applied_file) create_empty_file(self.__unapplied_file) + create_empty_file(self.__descr_file) self.__begin_stack_check() + def rename(self, to_name): + """Renames a series + """ + to_stack = Series(to_name) + + if to_stack.is_initialised(): + raise StackException, '"%s" already exists' % to_stack.get_branch() + if os.path.exists(to_stack.__base_file): + os.remove(to_stack.__base_file) + + git.rename_branch(self.__name, to_name) + + if os.path.isdir(self.__patch_dir): + os.rename(self.__patch_dir, to_stack.__patch_dir) + if os.path.exists(self.__base_file): + os.rename(self.__base_file, to_stack.__base_file) + + self.__init__(to_name) + + def delete(self, force = False): + """Deletes an stgit series + """ + if self.is_initialised(): + patches = self.get_unapplied() + self.get_applied() + if not force and patches: + raise StackException, \ + 'Cannot delete: the series still contains patches' + for p in patches: + Patch(p, self.__patch_dir).delete() + + if os.path.exists(self.__applied_file): + os.remove(self.__applied_file) + if os.path.exists(self.__unapplied_file): + os.remove(self.__unapplied_file) + if os.path.exists(self.__current_file): + os.remove(self.__current_file) + if os.path.exists(self.__descr_file): + os.remove(self.__descr_file) + if not os.listdir(self.__patch_dir): + os.rmdir(self.__patch_dir) + else: + print 'Series directory %s is not empty.' % self.__name + + if os.path.exists(self.__base_file): + os.remove(self.__base_file) + def refresh_patch(self, message = None, edit = False, show_patch = False, cache_update = True, author_name = None, author_email = None, @@ -583,6 +662,7 @@ class Series: top = patch.get_top() ex = None + modified = False # top != bottom always since we have a commit for each patch if head == bottom: @@ -600,6 +680,10 @@ class Series: # Try the fast applying first. If this fails, fall back to the # three-way merge if not git.apply_diff(bottom, top): + # if git.apply_diff() fails, the patch requires a diff3 + # merge and can be reported as modified + modified = True + # merge can fail but the patch needs to be pushed try: git.merge(bottom, head, top) @@ -626,6 +710,8 @@ class Series: else: raise StackException, str(ex) + return modified + def undo_push(self): name = self.get_current() assert(name) @@ -690,6 +776,9 @@ class Series: applied = self.get_applied() unapplied = self.get_unapplied() + if oldname == newname: + raise StackException, '"To" name and "from" name are the same' + if newname in applied or newname in unapplied: raise StackException, 'Patch "%s" already exists' % newname