#Author - Andrei Pascu
#Date - 04/02/16

#The cost variable is globalised and defined so that it can be used in other functions.
global cost
cost = 0

def welcomemessage():
    #This function prints a welcome message whenever called
    print('Welcome! \n Please all the details below to calculate your total costs.\n')

def exitmessage():
    #This program prints a welcome message and closes the program whenever called
    print('Your estimate has been calculated finished, goodbye! ')
    #Here the SystemExit function is raised so that the whole system module does not have to be imported. Once the function is imported and executed the program is stopped to stop the loop of asking if the user wants to start again,
    #but only if he answers no to the question
    raise SystemExit

def startmain():
    #This function starts the main sequence of the program, allowing the user to see the welcome message, and then be asked all the details to calculate the total
    welcomemessage()
    takedetails()

def takedetails():
    #This function is one of the main parts of the program, it asks the user for all the details in order to calculate the total in the second function of the program

    #Here all of the variables necessary to make to function work and take in the user's details are made global
    global area, height, width, wallpremove
    #Here the variables required to make the code function correctly are defined
    roomnum = 0
    wallnum = 0
    width = 0
    height = 0
    area = 0
    cost = 0
    wallpremove = 0
    #Below the user is asked for the following details: costumer number, date of estimate, Number of rooms that require painting, how many rooms need wallpaper removal.
    costumerNo=input('Please enter your customer number ')
    estdate=input('What is the date of your estimate? (DD/MM/YY) ')
    roomno = int(input('How many rooms require painting? '))
    wallpremove=int(input('For how many of your rooms would you like to have the wallpaper removed? '))
    #Here the user is asked if he wants to add another room
    response=input('Would you like to add another room? (Y or N) ')
    #If the response is 'N' , the user is asked to input the information for the rooms entered above.
    if response == 'N':
        #The user is asked the name, number of walls, measurements (height, width) here for the amount of rooms he entered.
        for i in range(roomno):

            #The roomnum variable is used to display the room number so that the user knows how many rooms he's entered so far, this variable increments in +1 so that for the amount of rooms entered,
            #each time the loop starts again 1 is added.
            roomnum = roomnum + 1
            #The name of the room is asked here, and then the message telling the user to enter the measurements for the room.
            print('Please enter the name of Room',roomnum,': ')
            roomname=str(input('Name:'))
            wallno=int(input('Please enter the number of walls in this room '))
            print('Please enter measurements for room ',roomnum)
            #Here the variable wallnum is defined so that when the loop moves onto the next room, the wall number indicator is reset.
            wallnum = 0

            for i in range(wallno):
                #The the amount of walls, firstly, the height and width variables are reset so that when the loop moves onto the next room, the height and width does not get overwritten by the previous room dimensions, and so the area
                #can be calculated correctly.
                height = 0
                width = 0
                #Much like in the first loop, another incrementing variable is defined here to indicate which wall number the user is on.
                wallnum = wallnum + 1
                #Here the wall height and width are asked to be entered.
                print('Please enter the measurements for wall ',wallnum,)
                height= float(input('Enter height of the wall ')) + height
                width=float(input('Enter the width of the wall ')) + width
                #The area is worked out by multiplying the height and the width, and adding it to the area of previous rooms.
                area= area + (height*width)
        #When the loop is finished here, the function costcalucation is initiated and the user is asked which type of worker is wanted (Apprentice or Fully Qualified), the total cost is displayed, and then the user is asked if he wants
        #to start over again.
        costcalculation()

    while response == 'Y':

        roomadd=int(input('How many rooms would you like to add? '))
        #The extra number of rooms are added to the initial number of rooms here.
        roomno=roomno+roomadd
        for i in range(roomno):
            #The roomnum variable is used to display the room number so that the user knows how many rooms he's entered so far, this variable increments in +1 so that for the amount of rooms entered,
            #each time the loop starts again 1 is added.
            roomnum = roomnum + 1
            #The name of the room is asked here, and then the message telling the user to enter the measurements for the room.
            print('Please enter the name of Room',roomnum,': ')
            roomname=str(input('Name: '))
            wallno=int(input('Please enter the number of walls in this room '))
            print('Please enter measurements for room ',roomnum)
            #Here the variable wallnum is defined so that when the loop moves onto the next room, the wall number indicator is reset.
            wallnum = 0

            for i in range(wallno):
                #The the amount of walls, firstly, the height and width variables are reset so that when the loop moves onto the next room, the height and width does not get overwritten by the previous room dimensions, and so the area
                #can be calculated correctly.
                height = 0
                width = 0
                #Much like in the first loop, another incrementing variable is defined here to indicate which wall number the user is on.
                wallnum = wallnum + 1
                #Here the wall height and width are asked to be entered.
                print('Please enter the measurements for wall ',wallnum,)
                height= float(input('Enter height of the wall ')) + height
                width=float(input('Enter the width of the wall ')) + width
                area= area +(height*width)
        #When the loop ends and the user has entered all the details, he is asked if he wants to enter another room for the final time before going onto the cost calculations.
        response2=input('Would you like to add another room? (Y or N) ')
        #If the user enters yes to the questions below, the loop starts again
        if response2=='Y':
            #When the user enters yes, the roomno variable is reset so that the user can enter how many rooms he wants to add, and not have to re-enter the details of the previous rooms again.
            roomno= 0
        #When the user enters no, the costcalculation variable is initiated and the user is asked which type of worker is wanted (Apprentice or Fully Qualified), the total cost is displayed, and then the user is asked if he wants
        #to start over again.
        if response2=='N':
            costcalculation()

def costcalculation():
    #The costcalucation function asks the user for the type of worker they want, and then asks them if they want to start over again until they say no. If they say yes, all the variables that are needed to make the program function
    #correctly are reset so that nothing is overwritten or added onto, and so the correct total cost can be displayed everytime it resets. If they answer no, the exitmessage function is initiated where an exit message is printed out
    #and the program is stopped.
    #The below variables are made global so that they can be used again in other functions as well as this one.
    global wallpremove
    global area
    #Here the user is asked what type of worker is wanted for the job so that the cost can be added on.
    workertype=input('What type of worker would you like? AP for Apprentice, or FQ for fully qualified. ')
    #Here the total cost is calculated
    if workertype == 'AP':
        #If the user enters AP for Apprentice, the cost is increased by £100, and the total cost is worked out by adding the cost by the total cost of the wallpaper (number of rooms to have wallpaper removed multiplied by £70), and adds
        #the cost of of the paint by multiplying the area calculated by the width and height of each wall before in the first takedetails function. The total of this is then multiplied by 1.2 for the VAT value of 20%.
        cost = (cost + 100)
        cost = (cost +(wallpremove * 70)+(area*15)) * 1.2

    else:
        #If the user enters FQ for Fully Qualified, the cost is increased by £100, and the total cost is worked out by adding the cost by the total cost of the wallpaper (number of rooms to have wallpaper removed multiplied by £70), and adds
        #the cost of of the paint by multiplying the area calculated by the width and height of each wall before in the first takedetails function. The total of this is then multiplied by 1.2 for the VAT value of 20%.
        cost = (cost + 250)
        cost = (cost +(wallpremove * 70)+(area*15)) * 1.2
    #Once the cost has been calculated, it is then displayed.
    print('Your estimated cost is: £',cost,)
    #After the cost has been displayed, the user is asked if he/she would like to restart and re-enter all the details
    restart=input('Would you like to re-enter details and calculate another estimate? (Y or N)')
    #If the user enters no to the question above, the exitmessage function is initiated which displays and exit message and uses a specific function from the system module to stop the program.
    if restart == 'N':
            exitmessage()

    while restart == 'Y':
        #This loop makes the user start the program again until he/she says no to the question at the end, if he/she says no, the exitmessage function is initiated to display an exit message and stop the program.
        #Here all of the variables are reset so that they are not overwritten or added onto the previous inputs before the restart.
        roomnum = 0
        wallnum = 0
        width = 0
        height = 0
        area = 0
        cost = 0
        wallpremove = 0
        #The startmain function initiates the welcomemessage and takedetails function so that the user can start again and re-enter all of the details.
        startmain()

        if restart == 'N':
            #If the user enters no to the previous, the exitmessage function is initiated which displays and exit message and uses a specific function from the system module to stop the program.
            exitmessage()