hgext/mq.py
changeset 8654 f6cc3638f468
parent 8653 aa011d123f71
child 8700 a96b049075a8
equal deleted inserted replaced
8653:aa011d123f71 8654:f6cc3638f468
   137         self.user = user
   137         self.user = user
   138         self.date = date
   138         self.date = date
   139         self.haspatch = diffstart > 1
   139         self.haspatch = diffstart > 1
   140 
   140 
   141     def setuser(self, user):
   141     def setuser(self, user):
   142         if not self.setheader(['From: ', '# User '], user):
   142         if not self.updateheader(['From: ', '# User '], user):
   143             try:
   143             try:
   144                 patchheaderat = self.comments.index('# HG changeset patch')
   144                 patchheaderat = self.comments.index('# HG changeset patch')
   145                 self.comments.insert(patchheaderat + 1,'# User ' + user)
   145                 self.comments.insert(patchheaderat + 1,'# User ' + user)
   146             except ValueError:
   146             except ValueError:
   147                 self.comments = ['From: ' + user, ''] + self.comments
   147                 self.comments = ['From: ' + user, ''] + self.comments
   148         self.user = user
   148         self.user = user
   149 
   149 
   150     def setdate(self, date):
   150     def setdate(self, date):
   151         if self.setheader(['# Date '], date):
   151         if self.updateheader(['# Date '], date):
   152             self.date = date
   152             self.date = date
   153 
   153 
   154     def setmessage(self, message):
   154     def setmessage(self, message):
   155         if self.comments:
   155         if self.comments:
   156             self._delmsg()
   156             self._delmsg()
   157         self.message = [message]
   157         self.message = [message]
   158         self.comments += self.message
   158         self.comments += self.message
   159 
   159 
   160     def setheader(self, prefixes, new):
   160     def updateheader(self, prefixes, new):
   161         '''Update all references to a field in the patch header.
   161         '''Update all references to a field in the patch header.
   162         If none found, add it email style.'''
   162         Return whether the field is present.'''
   163         res = False
   163         res = False
   164         for prefix in prefixes:
   164         for prefix in prefixes:
   165             for i in xrange(len(self.comments)):
   165             for i in xrange(len(self.comments)):
   166                 if self.comments[i].startswith(prefix):
   166                 if self.comments[i].startswith(prefix):
   167                     self.comments[i] = prefix + new
   167                     self.comments[i] = prefix + new