todo_list_db_ex2a.py

'''
Created on Apr 04, 2016
Copyright (c) 2015-2016 Teodoro Montanaro

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License
@author: tmontanaro
'''


# this script is an extension of the script todo_list_tts_ex1.py

import db_interaction

def new_task():
    '''
    Add a new element to the list of tasks in db
    '''

    # ask the user to insert the task she wants to add
    string = raw_input("Type the new task:\n>")

    urgent = -1
    # ask user to specify if the just inserted task is urgent or not
    # we continue to ask it until she inserts Y or N
    while (urgent == -1):
        urgent_string = raw_input("Is this task urgent (Y/N)?\n>")
        if len(urgent_string) == 1:
            if urgent_string.upper() == "Y" :
                urgent = 1
            elif urgent_string.upper() == "N":
                urgent = 0

    #insert the task in the db
    db_interaction.db_insert_task(string, urgent)

    print "The new task was successfully added to the list"


if __name__ == '__main__':
    # main program

    # set a variable to False: it will be used to re-execute the program multiple times
    ended = False

    # keep asking strings until the user types 2 (to exit)
    while not ended:

        # print the menu every time we finish to perform an operation
        print "Insert the number corresponding to the action you want to perform"
        print "1: insert a new task"
        print "2: close the program"

        # get the action as input
        string = raw_input("Your choice:\n>")

        # check if the inserted string is actually a number
        # we will ask the user a new input until it will insert a number
        while string.isdigit() != True:
            # if the string is not a number we will ask a new input
            string = raw_input("Wrong input! Your choice:\n>")

        # convert the string to int (integer number)
        choice = int(string)

        # depending on the chosen input we perform the right action
        if (choice == 1):  # insert a new task
            new_task()
        elif (choice == 2): # exit
            ended = True