How to run code again in module ?

[rajamani]

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()

Yoriz write Nov-09-2022, 06:22 AM:
Please post all code, output and errors (in their entirety) between their respective tags. Refer to BBCode help topic on how to post. Use the "Preview Post" button to make sure the code is presented as you expect before hitting the "Post Reply/Thread" button.

[deanhystad]

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()

[snippsat]

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🧵.