equation.py

from math import sqrt
from typing import Optional


class Equation:
    def __init__(self, a: int, b: int, c: int, v: str = "x") -> None:
        self.a = a
        self.b = b
        self.c = c
        self.v = v

    def __str__(self) -> str:
        contSign = "+"
        coefSign = "+"

        equation = f"{self.a}{self.v}² <sign_b> {abs(self.b)}{self.v} <sign_c> {abs(self.c)}"

        if b < 0:
            coefSign = "-"
        if c < 0:
            contSign = "-"

        equation = equation.replace("<sign_b>", coefSign)
        equation = equation.replace("<sign_c>", contSign)

        return equation

    @property
    def canSolve(self) -> bool:
        discriminant: int = self.b**2 - 4 * self.a * self.c

        if discriminant < 0:
            return False
        else:
            return True

    def root(self) -> Optional[tuple[float, float]]:
        if not self.canSolve:
            return None

        a = self.a
        b = self.b
        c = self.c

        d = b**2 - 4 * a * c

        root1 = (-b + sqrt(d)) / 2 * a
        root2 = (-b - sqrt(d)) / 2 * a

        return root1, root2


if __name__ == "__main__":
    a, b, c = 3, 6, 67

    equation = Equation(a, b, c)
    if roots := equation.root():
        print("Roots are", roots[0], roots[1])

    else:
        print("Eqn has no solution")