interactive_warhorn.py 文件源码

def add_drivers_to_tags(tag, drivers, driver_numbers):
    """ This function appends the driver tags sets the attributes and
    attribute names to the corresponding driver tag

    :Arguments:

    1. tag (xml.etree.ElementTree.Element) = Current tag to which the newly
    formed driver tags may be appended
    2. drivers (list[str]) = list of driver names available to the user
    3. driver_numbers (list[int]) = list of the numbers which correspond to
    the driver names that the user wants.

    """
    print "Selected drivers:"
    for driver_number in driver_numbers:
        driver_number = driver_number * 2 - 1
        if driver_number > len(drivers):
            print "Corresponding driver for " + str((driver_number+1)/2) +\
                  " not found."
            continue
        print str((driver_number+1)/2) + ". " + drivers[driver_number]
        driver_tag = Element("driver")
        driver_tag.set("name", drivers[driver_number])
        tag.append(driver_tag)