Как можно оптимизировать код?

import random
import string
 
try:
    # Создание пустых списков
    total = []
    result = []
 
    # Вход в цикл проверки длины
    T = True
    while T:
        # Ввод длины
        length = int(input('Длина (от 8 до 16)\n>>> '))
        
        # Проверка длины
        if length < 8:
            print('Длина должна быть больше или равна 8\n')
        elif length > 16:
            print('Длина должна быть меньше или равна 16\n')
        else:
            # Выход из цикла проверки длины
            T = False
 
    # Выбор и добавление нужных элементов в список total
    o = input('Добавить заглавные буквы? (Y/N)\n>>> ')
    if o == 'Y':
        total += list(string.ascii_uppercase)
        o = input('Добавить цифры? (Y/N)\n>>> ')
        if o == 'Y':
            total += list(string.digits)
            o = input('Добавить строчные буквы? (Y/N)\n>>> ')
            if o != 'N' or 'Y':
                print('Принимается только Y - Да и N - Нет')
            elif o == 'Y':
                total += list(string.ascii_lowercase)
        elif o == 'N':
            o = input('Добавить строчные буквы? (Y/N)\n>>> ')
            if o != 'N' or 'Y':
                print('Принимается только Y - Да и N - Нет')
            elif o == 'Y':
                total += list(string.ascii_lowercase)
        elif o != 'N' or 'Y':
            print('Принимается только Y - Да и N - Нет')
    elif o == 'N':
        o = input('Добавить цифры? (Y/N)\n>>> ')
        if o == 'Y':
            total += list(string.digits)
            o = input('Добавить строчные буквы? (Y/N)\n>>> ')
            if o != 'N' or 'Y':
                print('Принимается только Y - Да и N - Нет')
            elif o == 'Y':
                total += list(string.ascii_lowercase)
        elif o == 'N':
            o = input('Добавить строчные буквы? (Y/N)\n>>> ')
            if o != 'N' or 'Y':
                print('Принимается только Y - Да и N - Нет')
            elif o == 'Y':
                total += list(string.ascii_lowercase)
        elif o != 'N' or 'Y':
            print('Принимается только Y - Да и N - Нет')
    elif o != 'N' or 'Y':
        print('Принимается только Y - Да и N - Нет')
 
    if len(total) == 0:
        # Вывод результата при отстутствии элементов в списке total
        print('\nРезультат: Ничего')
    else:
        for i in range(length):
            # Добавление рандомных элементов из списка total в список result в диапазоне length
            result.append(random.choice(total))
        # Вывод результата при нахождении элементов в списке result
        print('\nРезультат:', ''.join(result))
 
# Исключение ошибки числа
except ValueError:
    print('Длина вводится в виде араб. цифр и не может состоять из вещественного числа!')

import random
import string
 
 
def pass_gradient(star_p: int, end_p: int, grade: int, n: int) -> str:
    """
    Программа которая генерирует пароли первая степень генерации-цифры,
    вторая-цифры и буквы третья-цифры,букву спец символы.
    Ещё она должна проверять пароль который
    ты вводишь на сложность.Код должен быть один.Не отдельно.
    :param star_p:начала диапазона случайных чисел
    :param end_p: конец диапазона случайных чисел
    :param grade: степень сложности пароля
    :param n: колмчество символов в нем
    :return:пароль
    образец вызова
    print(pass_gradient(star_p=1, end_p=10, grade=1, n=3))
    print(pass_gradient(star_p=1, end_p=10, grade=2, n=3))
    print(pass_gradient(star_p=1, end_p=10, grade=3, n=3))
    print(pass_gradient(star_p=1, end_p=10, grade=4, n=3))
 
 
    """
    if isinstance(star_p, int) and isinstance(end_p, int) and isinstance(grade, int) and isinstance(n, int):
        grades = [1, 2, 3]
        if grade in grades and grade == 1:
            print('Первая степень')
            return f'{str([random.randint(star_p, end_p) for _ in range(n)])}'.removeprefix('[').removesuffix(
                ']').strip()
        if grade in grades and grade == 2:
            print('вторая степень')
            sep = ''
            res = sep.join([
                f'{str([random.randint(star_p, end_p) for _ in range(n)])}'.removeprefix('[').removesuffix(']').strip(),
                f'{sep.join(random.choice(string.ascii_letters) for _ in range(n))}'])
            return sep.join(random.sample(res, len(res)))
        if grade in grades and grade == 3:
            print('третья степень')
            sep = ''
            res = sep.join([
                f'{str([random.randint(star_p, end_p) for _ in range(n)])}'.removeprefix('[').removesuffix(']').strip(),
                f'{sep.join(random.choice(string.ascii_letters) for _ in range(n))}'
                f'{sep.join(random.choice(string.punctuation) for _ in range(n))}'
            ])
            return sep.join(random.sample(res, len(res)))
        if grade not in grades:
            return 'wrong grade'
    else:
        raise TypeError('Введите целые числа')

from unittest import TestCase, main
from tested.pass_grade_package import pass_grade
from tested.pass_grade_package.pass_grade import pass_gradient
import doctest
import random
import re
import string
 
 
def load_tests(loader, tests, ignore):
    tests.addTests(doctest.DocTestSuite(pass_grade))
    return tests
 
 
class PassGradeTest(TestCase):
    def test_is_random_int(self):
        self.assertTrue((1 <= random.randint(1, 10) <= 10) is True, 'не рандом')
 
    def test_num_test(self):
        self.assertTrue((all(isinstance(i, int) for i in list(
            map(int, re.findall(r'\d+',
                                f'{str([random.randint(1, 10) for _ in range(5)])}'.removeprefix(
                                    '[').removesuffix(']').strip()))))) is True, 'не первая степень  ')
 
    def test_alpha_test(self):
        sep = ''
        fnd = sep.join([
            f'{str([random.randint(1, 10) for _ in range(5)])}'.removeprefix('[').removesuffix(']').strip(),
            f'{sep.join(random.choice(string.ascii_letters) for _ in range(5))}'])
        self.assertTrue((any(isinstance(i, int) for i in list(map(int, re.findall(r'\d+', fnd)))) and (
            any(i.isalpha() for i in re.findall(r'[^0-9,\s]', fnd)))
                         ) is True, 'не вторая степень')
 
    def test_last(self):
        sp = ''
        f = sp.join([
            f'{str([random.randint(1, 10) for _ in range(5)])}'.removeprefix('[').removesuffix(']').strip(),
            f'{sp.join(random.choice(string.ascii_letters) for _ in range(5))}'
            f'{sp.join(random.choice(string.punctuation) for _ in range(5))}'
        ])
 
        self.assertTrue((any(isinstance(i, int) for i in list(map(int, re.findall(r'\d+', f)))) and any(
            i not in string.punctuation and not i.isspace() and not i.isdigit() for i in f) and any(
            i for i in re.findall(r'[^0-9,a-zA-Z\s]', f))) is True, 'не третья степень')
 
    def test_wrong_input(self):
        with self.assertRaises(TypeError) as e:
            pass_gradient(star_p='1', end_p='10', grade='4', n='3')
        self.assertEqual('Введите целые числа', e.exception.args[0])
 
 
if __name__ == '__main__':
    main()