Есть код на функциях, как его переписать на ооп. При запуске скрипта нужно ввести число (это будет ip адрес), до которого будет выполняться пинг

Код который нужно исправить:

import subprocess

class Dignostic():

    def only_ping(self, address):
        self.make_ping = subprocess.Popen(['ping', '-t',  address], shell=True, stdout=subprocess.PIPE)
        for line in self.make_ping.stdout:
            yield(line.decode('cp866'))

    def make_ping(self):
        self.ping_data = input()
        print('До какого IP адреса выполнить ping ? Пример: 8.8.8.8')
        data2 = self.only_ping(int(self.ping_data))
        for i in data2:
            print(i)

    new_variable = 1000

obj_Diagnostic = Dignostic()

obj_Diagnostic.make_ping

Пример кода на функциях:

def fun1(data1):
    for data in range(data1+1):
        yield(data)

def fun2():
    data_input = input()
    data2 = fun1(int(data_input))
    for i in data2:
        print(i)

fun2()