solution_014a.py

#!/usr/bin/python3
# ===================================================================
# Create arithmetic worksheets - addition
# ===================================================================

import random

def addition_integer(id, ilist):

    lst = []

    ans     = 0                # problem answer
    cnt     = 0                # line count
    idx     = 0                # loop counter
    lstlen  = len(ilist)       # length of list
    lastidx = lstlen-1         # last list index

    while(idx < lstlen):

        val = ilist[idx]

        if idx == 0:
            lst.append(f'  {id:2}.   {val:>4}')
        elif idx == lastidx:
            lst.append(f'      + {val:>4}')
        else:
            lst.append(f'        {val:>4}')
        cnt += 1

        ans += val
        idx += 1
    lst.append('      ------')
    cnt += 1

    return(lst,ans,12,cnt)

# -------------------------------------------------------------------
# ---- main
# -------------------------------------------------------------------

if __name__ == '__main__':

    ilist = []

    for _ in range(2):
        ilist.append(random.randrange(-99,+99))

    (lst,ans,w,h) = addition_integer(1,ilist)

    print()
    for s in lst:
        print(s)
    print()
    print(f'answer = {ans}')
    print(f'box    = {w} x {h}')
    print('---------------------------------------')

    ilist = []

    for _ in range(4):
        ilist.append(random.randrange(-99,+99))

    (lst,ans,w,h) = addition_integer(2,ilist)

    print()
    for s in lst:
        print(s)
    print()
    print(f'answer = {ans}')
    print(f'box    = {w} x {h}')
    print('---------------------------------------')