[Spambayes-checkins] spambayes/spambayes message.py,1.2,1.3

[Tim Stone]

Update of /cvsroot/spambayes/spambayes/spambayes
In directory sc8-pr-cvs1:/tmp/cvs-serv24156

Modified Files:
	message.py 
Log Message:
Added a couple more methods to support copying one message to another

Index: message.py
===================================================================
RCS file: /cvsroot/spambayes/spambayes/spambayes/message.py,v
retrieving revision 1.2
retrieving revision 1.3
diff -C2 -d -r1.2 -r1.3
*** message.py	8 Apr 2003 07:35:42 -0000	1.2
--- message.py	8 Apr 2003 15:24:43 -0000	1.3
***************
*** 187,190 ****
--- 187,195 ----
      def getId(self):
          return self.id
+ 
+     def copy(self, old_msg):
+         self.setPayload(old_msg.payload())  # this is expensive...
+         self.setClassification(old_msg.getClassification())
+         self.setTraining(old_msg.getTraining())
          
      def addSBHeaders(self, prob, clues):
***************
*** 275,278 ****
--- 280,290 ----
      def getClassification(self):
          return self.c
+ 
+     def setClassification(self, cls):
+         if cls == 's' or cls == 'h' or cls == 'u' or cls is None:
+             self.c = cls
+             self.modified()
+         else:
+             raise ValueError
          
      def isTrndSpam(self):
***************
*** 312,316 ****
      def getTraining(self):
          return self.t
!         
      def __repr__(self):
          return "core.Message%r" % repr(self.__getstate__())
--- 324,335 ----
      def getTraining(self):
          return self.t
! 
!     def setTraining(self, trn):
!         if trn == 's' or trn == 'h' or trn is None:
!             self.t = trn
!             self.modified()
!         else:
!             raise ValueError
!          
      def __repr__(self):
          return "core.Message%r" % repr(self.__getstate__())