Python

Overloading operators in Python

# Overload the + operator to accept string or file as input arg
# If input is str, append to content
# if input is File, append content of file to content of current file

class File:
  def __init__(self, filename, content):
    self.filename = filename
    self.content = content

  def __repr__(self):
    return f"File(contents={self.content})"

  # Overload the + operator here

file_one = File("file1.txt", "Hello World. ")
file_two = File("file2.txt", "How are you doing? ")

file_one + "This is Jamie. "
file_one + file_two
print(file_one)


# Overload the + operator to accept string or file as data arg
# If data is str, append to content
# if data is File, append content of file to content of current file

class File:
  def __init__(self, filename, content):
    self.filename = filename
    self.content = content

  def __repr__(self):
    return f"File(contents={self.content})"

  def __add__(self, data):
    if isinstance(data, File):
      self.content += data.content
    elif isinstance(data, str):
      self.content += data

file_one = File("file1.txt", "Hello World. ")
file_two = File("file2.txt", "How are you doing? ")

file_one + "This is Jamie. "
file_one + file_two
print(file_one)
