Python Forum

Hi all,
I'm a new to python and coding.

How to run back the code when a user chooses an option ?

- the code i have currently-

print('Now you try')
print('This is how to get the volumetric weight for cargo space calculation')


num1 = input('Enter length number: ')
num2 = input('Enter width number: ')
num3 = input('Enter height number: ')
sum = float(num1) * float(num2) * float(num3) / 5000

print('This is for your shipment information')

char1 = input ('Please enter your name: ')
char2 = input ('Please enter comp name: ')
char3 = input ('Please enter reference: ')

char4 = input ('Please enter you consignee name: ')
char5 = input ('Please enter consignee comp name: ')
char6 = input ('Please enter shipment value: ')

print ('Conclusion')

print('The sum of {0} and {1} and {2} divided by 5000 is = {3} cm3'.format(num1, num2, num3, sum))
print('Your name is {0}'.format(char1, ))
print('Your company is {0}'.format(char2, ))
print('Your reference is {0}'.format(char3, ))
print('Your consignee name is {0}'.format(char4, ))
print('Your consignee company name is {0}'.format(char5, ))
print('Your shipment value is ${0}'.format(char6, ))


print('What do you want to do next ?')

choice = input("""
                      A: Continue
                      B: Logout

                      Please enter your choice: """)

if choice == "A" or choice =="a":restart("test1.py",) stuck here, I don't know which command to use to loop back the code on top
elif choice == "B" or choice =="b":quit() - this works.
else:
        print("You must only select either A or B")
        print("Please try again")
        menu()

Put all the code you want to re-run inside a function. Call the function from inside a loop.

def my_func():
    # All the code I want to run over an over

while True:
    choice = get_input() # Replace with you code to read user choice
    if choice == Logout:
        break
    my_func()

To show code a example that works fine for task like this
So it's like deanhystad code taken a little futher,
so menu function is where start and always fall back into after run a function.
Then can restart an other task or Quit out.

from rich import print

def cargo_calc():
    '''This is how to get the volumetric weight for cargo space calculation'''
    num1 = input('Enter length number: ')
    num2 = input('Enter width number: ')
    num3 = input('Enter height number: ')
    return float(num1) * float(num2) * float(num3) / 5000

def shipment_calc():
    '''Not finish yet'''
    pass

def menu_choies():
    return '''\
    (1) Cargo space calculation)
    (2) shipment space calculation)
    ([bold red]Q[/bold red]) Quit\n'''

def menu():
    while True:
        print(menu_choies())
        choice = input('Enter your choice: ').lower()
        if choice == '1':
            print(f'Volumetric weight for cargo is {cargo_calc()}\n')
        elif choice == '2':
            shipment_calc()
        elif choice == 'q':
            return
        else:
            print(f'Not a correct choice: {choice},try again\n')

if __name__ == '__main__':
    menu()

Also trown in Rich for some colors which work for on all OS/shell even cmd🧵.