Giải phương trình bậc nhất và bậc hai (Có xử lý ngoại lệ đầu vào)

try:
    with open("Bai2.10.inp",mode='r',encoding='utf-8') as file_inp:
        option = int(file_inp.readline())
        lst_num = file_inp.read().split()

    if option == 1:
        a,b = map(float,lst_num)

        if a == 0:
            if b == 0:
                noti = "Phương trình có vô số nghiệm"
            else:
                noti = "Phương trình vô nghiệm"
        else:
            x1 = -b / a
            noti = f"Phương trình có 1 nghiệm duy nhất x = {x1}"
    elif option == 2:
        a,b,c = map(float,lst_num)
        delta = b**2 - 4*a*c

        if a == 0:
            if b == 0 and c == 0:
                noti = "Phương trình có vô số nghiệm"
            elif b == 0 and c != 0:
                noti = "Phương trình vô nghiệm"
            elif b != 0:
                x1 = x2 = -c / b
                noti = f"Phương trình có 1 nghiệm duy nhất x = {x1}"
        else:
            if delta < 0:
                noti = "Phương trình vô nghiệm"      
            elif delta == 0:
                x1 = -b / (2 * a)
                noti = f"Phương trình có nghiệm kép x1 = x2 = {x1}"
            else:
                from math import sqrt
                x1 = (-b + sqrt(delta)) / (2 * a)
                x2 = (-b - sqrt(delta)) / (2 * a)
                noti = f"Phương trình có hai nghiệm phân biệt: x1 = {x1}, x2 = {x2}"
    else:
        noti = "Vui lòng chọn một trong hai chức năng:\n1. Giải phương trình bậc nhất\n2. Giải phương trình bậc hai"

except FileNotFoundError:
    noti = "Không tìm thấy file input"
except:
    noti = "Định dạng đầu vào không hợp lệ!"

with open("Bai2.10.out",mode='w+',encoding='utf-8') as file_out:
    file_out.write(noti)
    file_out.seek(0)
    reading = file_out.read()
    print(reading)

import math

try:

    with open("Bai24.inp.txt", 'r', encoding='utf-8') as file:

        line1=file.readline().rstrip()

        if line1 == "1":

            bacpt = "1"

            line2=file.readline()

            lst = line2.split()

            a = float(lst[0])

            b = float(lst[1])

            if a == 0:

                if b == 0:

                    with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} có vô số nghiệm")

                if b != 0:

                    with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} vô nghiệm")

            else:

                nghiem = -b/a

                with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} có nghiệm duy nhất là {nghiem}")

        if line1 == "2":

            bacpt ="2"

            line2=file.readline()

            lst = line2.split()

            a = float(lst[0])

            b = float(lst[1])

            c = float(lst[2])

            if a == 0:

                if b == 0:

                    if c == 0:

                        with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                            file.write(f"Phương trình bậc {bacpt} có vô số nghiệm")

                    elif c != 0:

                        with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                            file.write(f"Phương trình bậc {bacpt} vô nghiệm")

                elif b != 0:

                    nghiem = -b/a

                    with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} có nghiệm duy nhất là {nghiem}")

            elif a != 0:

                delta = b * b - 4 * a * c

                if delta > 0:

                    nghiem1 = (-b -math.sqrt(delta))/(2*a)

                    nghiem2 = (-b +math.sqrt(delta))/(2*a)

                    with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} có 2 nghiệm riêng biệt là x1 = {nghiem1} và x2 = {nghiem2}")

                elif delta == 0:

                    nghiemkep = -b/(2*a)

                    with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} có nghiệm kép là: x = {nghiemkep}")

                else:

                   with open ("Bai24.out.txt", "w", encoding='utf-8') as file:

                        file.write(f"Phương trình bậc {bacpt} vô nghiệm")

except FileNotFoundError:

    with open('Bai24.out.txt', 'w', encoding='utf-8') as file:

        file.write("Không tìm thấy file Bai24.inp.txt.")

except ValueError as e:

    with open('Bai24.out.txt', 'w', encoding='utf-8') as file:

        file.write(str(e))

except Exception as e:

    with open('Bai24.out.txt', 'w', encoding='utf-8') as file:    

        file.write("Đã xảy ra lỗi: " + str(e))

import math
loaiphuongtrinh = ''
txt = ''
ptbacnhat = False
ptbachai = False

# Bước 1: Mở file và check loại chương trình nào
try:
    with open('Bai2.10.inp', mode = 'r') as InFile:
        TheFile = InFile.readlines()
        TheFile[0] = int(TheFile[0].rstrip('\n'))
        InFile.close()
        if TheFile[0] == 1:
            loaiphuongtrinh = 'bacnhat'
        elif TheFile[0] == 2:
            loaiphuongtrinh = 'bachai'
        else:
            txt = 'Vui long chon mot trong hai chuc nang:\n1. Giai phuong trinh bac nhat\n2. Giai phuong trinh bac hai'
except:
    txt = 'Khong tim thay file input!'

# Bước 2A: Giải phương trình bậc hai
#   2A.1 Check đầu vào hợp lệ
if loaiphuongtrinh == 'bachai':
    try:
        TheFile[1] = TheFile[1].split()
        a,b,c = map(float,TheFile[1])
        ptbachai = True
    except:
        txt = 'Dinh dang dau vao khong hop le!'
#   2B.2 Giải phương trình bậc 2
if ptbachai:
    a,b,c = map(float,TheFile[1])
    delta = b*b - 4*a*c
    if a == 0 and b == 0 and c == 0:
        txt = 'Phuong trinh co vo so nghiem'
    elif a == 0 and b == 0 and c != 0:
        txt = 'Phương trình vô nghiệm'
    elif a == 0 and b != 0:
        x = -c/b
        txt = 'Phương trình có một nghiệm duy nhất:\nx = {}'.format(x)
    elif a != 0 and b != 0 and c != 0:
        if delta < 0:
            txt = 'Phương trình vô nghiệm'
        elif delta == 0:
            x = -b/(2*a)
            txt = 'Phương trình có nghiệm kép:\nx = {}'.format(x)
        elif delta > 0:
            x1 = (-b + math.sqrt(delta))/a
            x2 = (-b - math.sqrt(delta))/a
            txt = 'Phương trình có hai nghiệm phân biệt:\nx1 = {}\nx2 = {}'.format(x1,x2)

# Bước 2B: Giải phương trình bậc nhất
#   2A.1 Check đầu vào hợp lệ
if loaiphuongtrinh == 'bacnhat':
    try:
        TheFile[1] = TheFile[1].split()
        a,b = map(float,TheFile[1])
        ptbacnhat = True
    except:
        txt = 'Dinh dang dau vao khong hop le!'
#   2B.2 Giải phương trình bậc 2
if ptbacnhat:
    if a==0 and b==0:
        txt = 'Phuong trinh co vo so nghiem'
    elif a==0 and b!=0:
        txt = 'Phương trình vô nghiệm'
    elif a!=0 and b==0:
        txt = 'Phương trình có một nghiệm duy nhất:\nx = 0'
    elif a!=0 and b!=0:
        x = -b/a
        txt = 'Phương trình có một nghiệm duy nhất:\nx = {}'.format(x)

# Bước 3: Xuất ra output
print(txt)
with open('Bai2.10.out', mode='w') as OutFile:
    OutFile.write(txt)
    OutFile.close()

Mình cho là không cần code phần xử lý phương trình có vô số nghiệm, vì trong định nghĩa pt bậc nhất và bậc hai một ẩn thì a phải khác 0 gòi. Thay vào đó, mình  thêm phần xử lý ZeroDivisionError trong trường hợp a = 0. Mọi người tham khảo nha!

import math
try:
	with open("Input.txt") as in_file:
		data = in_file.readlines()
		mode = data[0].rstrip("\n")
	if mode == "1":
		a, b = map(float, data[1].split())
		x = -b/a
		report = "Phuong trinh co nghiem x = {}".format(x)
	elif mode == "2":
		a, b, c = map(float, data[1].split())
		delta = b*b - 4*a*c
		if delta > 0:
			x1 = (-b - math.sqrt(delta))/(2*a)
			x2 = (-b + math.sqrt(delta))/(2*a)
			report = "Phuong trinh co 2 nghiem phan biet:\nx1 = {}\nx2 = {}".format(x1, x2)
		elif delta == 0:
			x = -b/(2*a)
			report = "Phuong trinh co nghiem kep: x1 = x2 = {}".format(x)
		else:
			report = "Phuong trinh vo nghiem"	
	else:
		report = "Chon 1 trong 2 chuc nang:\n1. Giai phuong trinh bac 1\n2. Giai phuong trinh bac 2"		
except ZeroDivisionError:
	report = "a phai khac 0!"
except:
	report = "Dinh dang dau vao khong hop le!"
with open("Output.txt", mode = "w") as out_file:
	out_file.write(report)

Ngắn nhất có thể:
import math
import fractions

try:
    with open('input.inp',mode='r') as Phuong_trinh_dau_vao:
        readline_1 = Phuong_trinh_dau_vao.readline().rstrip('\n\n')
        
        #NẾU CHỌN GIẢI PHƯƠNG TRÌNH BẬC 1:
        if readline_1 == '1':
            readline_2 = Phuong_trinh_dau_vao.readline().rstrip('\n')
            a,b = map(float,readline_2.split())
            if a == 0 and b == 0:
                notice = "Phương trình có vô số nghiệm"
            elif a == 0 and b != 0:
                notice = "Phương trình vô nghiệm"
            else:
                PT_1nghiem = -b/a
                notice = ("Phương trình có 1 nghiệm duy nhất là\n X = {} ".format(PT_1nghiem))
        
        #NẾU CHỌN GIẢI PHƯƠNG TRÌNH BẬC 2
        elif readline_1 == '2':
            readline_2 = Phuong_trinh_dau_vao.readline().rstrip('\n')
            a,b,c = map(float,readline_2.split())
            
            #TÍNH DELTA
            nghiem_kep =fractions.Fraction(-b / (2*a))
            delta = b*b - 4*a*c
            
            if delta > 0:
                X_1 = (-b + math.sqrt(delta))/(2*a)
                X_2 = (-b - math.sqrt(delta))/(2*a)
                notice = ("Phương trình có 2 nghiệm phân biệt:\n X1 = {}\n X2 = {}".format(X_1,X_2))
            
            elif delta == 0:
                notice = ("Phương trình có nghiệm kép là X1 = X2 = {} ".format(nghiem_kep))

            elif delta < 0:
                notice =("Phương trình đã cho vô nghiệm")

        #NẾU CHỌN SAI Ở DÒNG 1
        elif readline_1 != '1' or '2':
            notice =("Vui lòng chọn lại định dạng 1 hoặc 2 ở dòng thứ nhất")

except FileNotFoundError:
    notice =("File not found")


except:
    notice = ("Bạn đã nhập sai định dạng")

with open("output.txt",mode='wb')  as file_output:
    file_output.write(notice.encode('utf8'))