Mega Code Archive

 

Categories / Python / Class

 

Implement class add (+) operator

import string class spam:     def __init__(self):         self.eggs = 1     def __del__(self):         pass     def __add__(self,other):          rt = spam()          rt.eggs = self.eggs + other.eggs          return rt     def __coerce__(self,other):          rt = spam()          if type(other) == type(rt):              return (self,other)          elif type(other) == type(""):              e = string.atoi(other)              rt.eggs = e              return(self,rt)          else:              return None     def __radd__(self,other):          return self + other if __name__ == "__main__":      a = spam()      b = spam()      a = a + b      print "a now has", a.eggs, "eggs"      a = a + "24"      print "a now has", a.eggs, "eggs"      a = "24" + a      print "a now has", a.eggs, "eggs"