field_class.py

from potato_class import *
from wheat_class import *
from cow_class import *
from sheep_class import *

import random

class Field:
    """Simulate a field that can contain animals and crops"""

    #constructor
    def __init__(self,max_crops,max_animals):
        self._crops = []
        self._animals = []
        self._max_animals = max_animals
        self._max_crops = max_crops

    def plant_crop(self,crop):
        if len(self._crops) < self._max_crops:
            self._crops.append(crop)
            return True
        else:
            return False

    def add_animal(self,animal):
        if len(self._animals) < self._max_animals:
            self._animals.append(animal)
            return True
        else:
            return False

    def harvest_crop(self,position):
        return self._crops.pop(position)

    def remove_animal(self,position):
        return self._animals.pop(position)

    def report_contents(self):
        crop_report = []
        animal_report = []
        for crop in self._crops:
            crop_report.append(crop.report())
        for animal in self._animals:
            animal_report.append(animal.report())
        return {"crops": crop_report,"animals":animal_report}

    def report_needs(self):
        food = 0
        light = 0
        water = 0
        for crop in self._crops:
            needs = crop.needs()
            if needs["light need"] > light:
                light = needs["light need"]
            if needs["water need"] > water:
                water = needs["water need"]
        for animal in self._animals:
            needs = animal.needs()
            food += needs["food need"]
            if needs["water need"] > water:
                water = needs["water need"]
        return {"food":food,"light":light,"water":water}

    def grow(self,light,food,water):
        #grow the crops (light and water are available to all crops in same amounts)
        if len(self._crops) > 0:
            for crop in self._crops:
                crop.grow(light,water)
        if len(self._animals) > 0:
            #grow the animals (water available to all animals in same amounts but food is a total that must be shared)
            food_required = 0
            #get a total of the food needed by the animals in the field
            for animal in self._animals:
                needs = animal.needs()
                food_required += needs["food need"]
            #if we have more food available than is required work out the additional available food
            if food > food_required:
                additional_food = food - food_required
                food = food_required
            else:
                additional_food = 0
            #grow each animal
            for animal in self._animals:
                #get the animals food needs
                needs = animal.needs()
                if food >= needs["food need"]:
                    #remove food for this animal from total
                    food -= needs["food need"]
                    feed = needs["food need"]
                    #see if there is any additional food to give
                    if additional_food > 0:
                        #remove some additional food for this animal
                        additional_food -= 1
                        #add this to the foot to be given to the animal
                        feed += 1
                    #grow the animal
                    animal.grow(feed,water)

def manual_grow(self):
    #get the light, water and food values from the user
    valid = False
    while not valid:
        try:
            light = int(input("Please enter a light value (1-10): "))
            if 1 <= light <= 10:
                valid = True
            else:
                print("Value entered not valid - please enter a value between 1 and 10")
        except ValueError:
            print("Value entered not valid - please enter a value between 1 and 10")
    valid = False
    while not valid:
        try:
            water = int(input("Please enter a water value (1-10): "))
            if 1 <= water <= 10:
                valid = True
            else:
                print("Value entered not valid - please enter a value between 1 and 10")
        except ValueError:
            print("Value entered not valid - please enter a value between 1 and 10")
    valid = False
    while not valid:
        try:
            food = int(input("Please enter a food value (1-100): "))
            if 1 <= food <= 100:
                valid = True
            else:
                print("Value entered not valid - please enter a value between 1 and 100")
        except ValueError:
            print("Value entered not valid - please enter a value between 1 and 100")
    #grow the field
    self.grow(light,food,water)

def auto_grow(self,days):
    #grow the field automatically over x days
    for day in range(days):
        light = random.randint(1,10)
        water = random.randint(1,10)
        food = random.randint(1,100)
        self.grow(light,food,water)

def display_crops(crop_list):
    print()
    print("The following crops are in this field:")
    for pos in range(len(crop_list)):
        print("{0:>4}. {1}.".format(pos+1,crop_list[pos].report()))

def display_animals(animal_list):
    print("The following animals are in this field:")
    for pos in range(len(animal_list)):
        print("{0:>4}. {1}.".format(pos+1,animal_list[pos].report()))

def select_crop(length_list):
    selected = 0
    while selected not in range(1,length_list+1):
        selected = int(input("Please select a crop: "))
        if selected not in range(1,length_list+1):
            print("Please select a valid option")
    return selected - 1

def select_animal(length_list):
    selected = 0
    while selected not in range(1,length_list+1):
        selected = int(input("Please select an animal: "))
        if selected not in range(1,length_list+1):
            print("Please select a valid option")
    return selected - 1

def harvest_crop_from_field(new_field):
    display_crops(new_field._crops)
    selected_crop = select_crop(len(new_field._crops))
    return new_field.harvest_crop(selected_crop)

def remove_animal_from_field(new_field):
    display_animals(new_field._animals)
    selected_animal = select_animal(len(new_field._animals))
    return new_field.remove_animal(selected_animal)

def display_crop_menu():
    print()
    print("Which crop type would you like to add?")
    print()
    print("1. Potato")
    print("2. Wheat")
    print()
    print("0. I don't want to add a crop - return me to the main menu")
    print()
    print('Please select an option from the above menu')

def plant_crop_in_field(new_field):
    display_crop_menu()
    crop_choice = get_menu_choice(0,2)
    if crop_choice == 1:
        if new_field.plant_crop(Potato()):
            print()
            print("Crop planted")
            print()
        else:
            print()
            print("Wheat is full - wheat not planted")
            print()
    if crop_choice == 2:
        if new_field.plant_crop(Wheat()):
            print()
            print("Wheat planted")
            print()
        else:
            print()
            print("Field is full - wheat not planted")
            print()

def add_animal_to_field(new_field):
    display_animal_menu()
    animal_choice = get_menu_choice(0,2)
    name = input("Please give your new animal a name: ")
    if animal_choice == 1:
        if new_field.add_animal(Cow(name)):
            print()
            print("Cow added")
            print()
        else:
            print()
            print("Field is full - cow not added")
            print()
    if animal_choice == 2:
        if new_field.add_animal(Sheep(name)):
            print()
            print("Sheep added")
            print()
        else:
            print()
            print("Field is full - sheep not planted")
            print()


def display_main_menu():
    print('1. Plant a new crop')
    print('2. Harvest a crop')
    print()
    print('3. Add an animal')
    print('4. Remove an animal')
    print()
    print('5. Grow field manually over 1 day')
    print('6. Grow field automatically over 30 days')
    print()
    print('7. Report field status')
    print()
    print('0. Exit test program')
    print()
    print('Please select an option from the above menu')

def get_menu_choice(lower,upper):
    optionValid = False
    while optionValid == False:
        try:
            choice = int(input('Option Selected: '))
            if (choice >= lower) and (choice <= upper):
                optionValid = True
            else:
                print('Please enter a valid option')
        except ValueError:
            print('Please enter a valid option')
    return choice

def display_animal_menu():
    print()
    print("Which animal type would you like to add?")
    print()
    print("1. Cow")
    print("2. Sheep")
    print()
    print("0. I don't want to add an animal - return me to the main menu")
    print()
    print('Please select an option from the above menu')

def manage_field(self):
    print('This is field management program')
    print()
    exit = False
    while not exit:
        display_main_menu()
        option = get_menu_choice(0,7)
        print()
        if option == 1:
            plant_crop_in_field(self)
        if option == 2:
            removed_crop = harvest_crop_from_field(self)
            print("You removed the crop: {0}".format(removed_crop))
        if option == 3:
            add_animal_to_field(self)
        if option == 4:
            removed_animal = remove_animal_from_field(self)
            print("You removed the animal: {0}".format(removed_animal))
        if option == 5:
            manual_grow(self)
            print()
        elif option == 6:
            auto_grow(self,30)
            print()
        elif option == 7:
            print(self.report_contents())
            print()
        elif option == 0:
            exit = True
            print()
    print('Thank you for using the field management program')

def main():
    #instaniate the class
    new_field = Field(2,5)
    #test to see whether it works or not
    manage_field(new_field)

if __name__ == '__main__':
    main()