How do I refactor different classes with the same methods in Python?

Question

In this minimum working example, Classes A and B do the same thing but their methods are slightly different. They save an item to their respective lists and display them. However, I would like to define an abstract Save class which refactors a lot of the repeated code here. Is there a way to do this?

class A():
  def __init__(self):
    self.items = []

  def save(self, item):
    item = "This is {}".format(item)
    self.items.append(item)
      
  def display(self):
    return self.items


class B():
  def __init__(self):
    self.items = []

  def save(self, item):
    item = "This is {}...!".format(item)
    self.items.append(item)
      
  def display(self):
    return self.items

a = A()
a.save('A')
a.save('B')
print(a.display())

b = B()
b.save('C')
b.save('D')
print(b.display())

Output

['This is A', 'This is B']
['This is C...!', 'This is D...!']

what are you trying to accomplish this this?

Ironkey
– Ironkey

2020-10-28 01:00:49 +00:00
Commented Oct 28, 2020 at 1:00 — Ironkey
– Ironkey, Commented Oct 28, 2020 at 1:00

Daniel Walker · Accepted Answer · 2020-10-28 01:03:41Z

2

class Save:
    def __init__(self):
        self.items = []

    def save(self, item):
        item = self.format_item(item)
        self.items.append(item)

    def display(self):
        return self.items

class A(Save):
    def format_item(self, item):
        return 'This is {}'.format(item)

class B(Save):
    def format_item(self, item):
        return 'This is {}...!'.format(item)

answered Oct 28, 2020 at 1:03

Daniel Walker

6,9067 gold badges25 silver badges65 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

How do I refactor different classes with the same methods in Python?

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related