from rest_framework import serializers
from gardens.models import *

class PlanSerializer(serializers.ModelSerializer):
    class Meta:
        model = FlightPlans
        fields = ["flight_plan"]

class FlightSerializer(serializers.ModelSerializer):
    class Meta:
        model = Flights
        fields = ["id","plan_id","start_date", "end_date"]

class MissionSerializer(serializers.ModelSerializer):
    class Meta:
        model = Missions
        fields = ["id", "name", "generated_date", "block_id" ]

class TreeSerializer(serializers.ModelSerializer):
    class Meta:
        model = Trees
        fields = ["id", "latitude", "longitude", "block_id", "width", "height", "cultivares_id", "row_id"]

class DroneStationSerializer(serializers.ModelSerializer):
    class Meta:
        model = DroneStations
        fields = ["latitude", "longitude"]

class FlightDiagnosticsSerializer(serializers.ModelSerializer):
    class Meta:
        model = Flights
        fields = ["diagnostics"]

class PhotoSerializer(serializers.ModelSerializer):
    class Meta:
        model = Photos
        fields = ["id","image_uri", "altitude", "photo_location_num","detected_scab", "tree_id", "mission_id"]

class EnterpriseTreeBreedSerializer(serializers.ModelSerializer):
    class Meta:
        model = EnterpriseSpecificBreeds
        fields = ["id","title", "width", "height","yields", "injection", "planting_distance", "cultivar"]

class MLModelsSerializer(serializers.ModelSerializer):
    class Meta:
        model = MLModels
        fields = ["id","title", "file_uri", "classes", "active_classes", "model_type"]