Python

Overloading operators in Python

# Support the operators >, < and == to compare file size for File

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

  # Overload operator ==

  # Overload operator <

  # Overload operator >
  

file_one = File("file1.txt", 243432)
file_two = File("file2.txt", 6755352)

if file_two < file_one:
  print(f"{file_one.filename} is larger in size")
elif file_one < file_two:
  print(f"{file_two.filename} is larger in size")
elif  file_one == file_two:
  print(f"{file_two.filename} and {file_two.filename} are the same size")


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

  def __eq__(self, file):
    return self.filesize == file.filesize

  def __gt__(self, file):
    return self.filesize > file.filesize

  def __lt__(self, file):
    return self.filesize < file.filesize

file_one = File("file1.txt", 243432)
file_two = File("file2.txt", 6755352)

if file_two < file_one:
  print(f"{file_one.filename} is larger in size")
elif file_one < file_two:
  print(f"{file_two.filename} is larger in size")
elif  file_one == file_two:
  print(f"{file_two.filename} and {file_two.filename} are the same size")
