arithmetic_formatter/maths.py at main · danielle3625/arithmetic_formatter · 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
def arithmetic_arranger(problems, solution=False):

    while True:
      # Check to see if problems exceed maximum amount
      if (len(problems) > 5):
        return "Error: Too many problems."
        break

      uppernum = ''
      lowernum = ''
      dividerline = ''
      answernum = ''

      # Split the list so that each index can be called and assigned
      for prob in problems:
        probs = prob.split()


        if probs[1] not in ['+', '-',]:
          return "Error: Operator must be '+' or '-'."
          break

        if not probs[0].isdigit() or not probs[2].isdigit():
          return 'Error: Numbers must only contain digits.'
          break

        for item in probs:
          if len(item) > 4:
            return 'Error: Numbers cannot be more than four digits.'
            break

        if probs[1] == '+':
          result = int(probs[0]) + int(probs[2])
        else:
          result = int(probs[0]) - int(probs[2])


        max_length = max(len(probs[0]), len(probs[2])) + 2
        upperline = str(probs[0]).rjust(max_length) + '    '
        lowerline = str(probs[1]) + str(probs[2].rjust(max_length - 1)) + "    "
        divider = '-' * max_length + "    "
        answer = str(result).rjust(max_length) + "    "

        uppernum += upperline
        lowernum += lowerline
        dividerline += divider
        answernum += answer


      if solution == True:
        return (f'{uppernum.rstrip()}\n{lowernum.rstrip()}\n{dividerline.rstrip()}\n{answernum.rstrip()}')
      elif solution == False:
        return (f'{uppernum.rstrip()}\n{lowernum.rstrip()}\n{dividerline.rstrip()}')