HDPG1D/hdpg1d/coefficients.py

class coefficients:
    def __init__(self, diff, conv, flux, porder, nele):
        self.diffusion = diff
        self.covection = conv
        self.flux = flux
        self.porder = porder
        self.nele = nele

    @classmethod
    def from_input(cls):
        while True:
            try:
                print("Please provide the following coefficients.")
                diff = float(input("Diffusion coefficient: "))
                conv = float(input("Covection coefficient: "))
                flux = float(input("Flux: "))
                porder = int(input("Order of polynomials: "))
                nele = int(input("Number of elements: "))
            except ValueError:
                print("Sorry, wrong data type.")
                continue
            else:
                break
        return cls(diff, conv, flux, porder, nele)