I have an object class:
class Color(object):
def __init__(self, color):
self.color = color
I want to run the following commands:
blue = Color("blue")
print blue
"The pen is {}".format(blue)
"The pen is blue".replace(blue, "red")
This returns:
# <__main__.Color object at 0x000001A527675908>
# The pen is <__main__.Color object at 0x000001A527675908>
# # Original exception was:
# Traceback (most recent call last):
# File "<maya console>", line 2, in <module>
# TypeError: expected a string or other character buffer object #
I can fix the print and format by including repr to the class.
class Color(object):
def __init__(self, color):
self.color = color
def __repr__(self):
return self.color
blue = Color("blue")
print blue
"The pen is {}".format(blue)
"The pen is blue".replace(blue, "red")
# blue
# The pen is blue
# # Original exception was:
# Traceback (most recent call last):
# File "<maya console>", line 2, in <module>
# TypeError: expected a string or other character buffer object #
How can I get the replace function to work with this object class?
It works if I put a str() wrapper around the object class:
"The pen is blue".replace(str(blue), "red")
However I don't want to do it this way, as it would require a lot of special code support to function.
A solution is to make your class a subclass of str. This is similar to How to make a class that acts like a string?
Modified Code
class Color(str):
def __init__(self, color):
self.color = color
def __repr__(self):
return self.color
Usage
blue = Color("blue")
print("The pen is {}".format(blue))
print("The pen is blue".replace(blue, "red"))
Output
The pen is blue
The pen is red