Scripts/Arithmetic_Formatter.py at main · Le0nard/Scripts · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
def arithmetic_arranger(problems, show_answers=False):

    if len(problems) > 5:
        return "Error: Too many problems."

    dictionary = "0123456789"
    unacceptable_operators = ["*", "/"]

    # Per ogni elemento contenuto nella lista problems, spacchetto il primo e il secondo numero e l'operatore
    first_numbers = []
    second_numbers = []
    dashes = []
    third_numbers = []
    for problem in problems:
        parts = problem.split()

        # Check if the first operand is too long
        if len(parts[0]) > 4:
            return 'Error: Numbers cannot be more than four digits.'
        if not all(char in dictionary for char in parts[0]):
            return 'Error: Numbers must only contain digits.'
        #else:
        #    first_numbers.append(parts[0])

        # Check if the operator is valid
        if parts[1] in unacceptable_operators:
            return "Error: Operator must be '+' or '-'."
        #else:
        #    operators.append(parts[1])

        # Check if the second operand is too long
        if len(parts[2]) > 4:
            return 'Error: Numbers cannot be more than four digits.'
        if not all(char in dictionary for char in parts[2]):
            return 'Error: Numbers must only contain digits.'
        #else:
        #    second_numbers.append(parts[2])

        max_length = max(len(parts[0]), len(parts[2])) + 2 # +2 for the operator and the space
        first_numbers.append(parts[0].rjust(max_length))
        second_numbers.append(parts[1] + parts[2].rjust(max_length-1))
        dashes.append('-' * max_length)
        if parts[1] == '+':
            third_numbers.append(str(int(parts[0]) + int(parts[2])).rjust(max_length))
        elif parts[1] == '-':
            if int(parts[0]) - int(parts[2]) > 0:
                third_numbers.append(str(int(parts[0]) - int(parts[2])).rjust(max_length))
            else:
                third_numbers.append((parts[1] + str(int(parts[2]) - int(parts[0]))).rjust(max_length))


        first_numbers.append('    ')
        second_numbers.append('    ')
        dashes.append('    ')
        third_numbers.append('    ')

    first_string = ''.join(first_numbers)
    second_string = ''.join(second_numbers)
    dashes_string = ''.join(dashes)
    third_string = ''.join(third_numbers)

    # remove the last 4 spaces from each string
    first_string = first_string.rstrip()
    second_string = second_string.rstrip()
    dashes_string = dashes_string.rstrip()
    third_string = third_string.rstrip()


    string_with_answer = first_string + '\n' + second_string + '\n' + dashes_string + '\n' + third_string

    string_without_answer = first_string + '\n' + second_string + '\n' + dashes_string

    if show_answers:
        return string_with_answer
    else:
        return string_without_answer


#print(f'\n{arithmetic_arranger(["32 + 698", "3801 - 2", "45 + 43", "123 + 49"])}')

#operation = ["32 + 698", "3801 - 2", "45 + 43", "123 + 49"]
#operation_1 = ["3801 - 2", "123 + 49"]
#operation_2 = ["3 + 855", "3801 - 2", "45 + 43", "123 + 49"]
#operation_3 = ["3 / 855", "3801 - 2", "45 + 43", "123 + 49"]
#operation_6 = ["3 / 855", "3801 - 2", "45 + 43", "123 + 49"]
#operation_8 = ["98 + 3g5", "3801 - 2", "45 + 43", "123 + 49"]
operation_10 = ["32 - 698", "1 - 3801", "45 + 43", "123 + 49", "988 + 40"]
print()
print(arithmetic_arranger(operation_10, True))