SimulationFramework.py

import FMMFramework as FMM
import random
from Connection import client_hack, accessToken
from Projection import hackathon_projector, minlat_, maxlat_, minlon_, maxlon_
from Utilities import get_address
from config import vehicle_count, fmm_api_url
from typing import Union, Generator
from numbers import Number

app_opening_task_type= (
    client_hack.get_database('fmm').get_collection('taskType')
).find_one({'name': 'App Opening'})

charging_station_task_type = (
    client_hack.get_database('fmm').get_collection('taskType')
).find_one({'name': 'Charging Station'})

admin_user_id = '55b95aa8010149cd54d6e089'


def all_license_plates() -> Generator:
    return ('NH-{}'.format(i) for i in range(vehicle_count))


def random_coordinates_from_hackathon_location():
    lat = str(random.uniform(minlat_, maxlat_))
    lon = str(random.uniform(minlon_, maxlon_))
    
    address = get_address(lat, lon)
    
    return {
        'lat': lat,
        'latitude': lat,
        'lon': lon,
        'longitude': lon,
        'full_address': address,
        'address': address
}


def get_fuel_level(license_plate: str) -> float:
    return float(FMM.get_vehicle_by_license_plate(
        client_hack, license_plate
    )['fuel_level'])


def set_fuel_level(license_plate: str, fuel_level: Union[str, Number]):
    fuel_level = float(fuel_level)
    if fuel_level < .0:
        fuel_level = .0
    elif fuel_level > 100.0:
        fuel_level = 100.0

    vehicle_id = FMM.get_vehicle_by_license_plate(
        client_hack, license_plate
    )['_id']

    FMM.set_vehicle_fields(
        client_hack, vehicle_id, {
            'fuel_level': '{:.4}'.format(fuel_level),
            'fuel': '{:.4f}'.format(fuel_level)
        }
    )


def set_battery_voltage(license_plate: str,
                        battery_voltage: Union[str, Number]):
    battery_voltage = float(battery_voltage)
    assert .0 <= battery_voltage <= 14.0

    vehicle_id = FMM.get_vehicle_by_license_plate(
        client_hack, license_plate
    )['_id']

    FMM.set_vehicle_fields(
        client_hack, vehicle_id, {
            'battery_voltage': '{:.3f}'.format(battery_voltage),
            'batteryVoltage': '{:.3f}'.format(battery_voltage)
        }
    )


def create_app_opening(license_plate: str):
    vehicle_id = FMM.get_vehicle_by_license_plate(client_hack,
                                                  license_plate)['_id']

    while True:
        origin = random_coordinates_from_hackathon_location()
        destination = random_coordinates_from_hackathon_location()
        distance = hackathon_projector.distance(origin, destination)
        if distance >= 300.0:
            break

    location_id = FMM.get_location_id(client_hack, 'New Hackshire')

    additional_fields = {
        'origin': origin,
        'destination': destination,
        'revenue': distance * 1/100,
        'picked_up': False,
        'delivered': False
    }

    FMM.create_task(
        client_hack, app_opening_task_type,
        vehicle_id, location_id, admin_user_id,
        fmm_api_url, accessToken, additional_fields
    )

def create_charging_station_task(charging_station: str):
    vehicle_id = FMM.get_vehicle_by_license_plate(client_hack,
                                                  charging_station)['_id']

    location_id = FMM.get_location_id(client_hack, 'New Hackshire')

    FMM.create_task(client_hack, charging_station_task_type, vehicle_id,
                    location_id, admin_user_id, fmm_api_url, accessToken)

def revive_12v_battery(license_plate: str):
    set_battery_voltage(license_plate, random.uniform(12.0, 14.0))


def kill_12v_battery(license_plate: str):
    new_battery_voltage = random.uniform(0.0, 11.6)
    set_battery_voltage(license_plate, new_battery_voltage)


def drain_fuel(license_plate: str, drain: Union[str, Number]):
    old_fuel_level = get_fuel_level(license_plate)
    new_fuel_level = max(old_fuel_level-float(drain), .0)
    set_fuel_level(license_plate, new_fuel_level)


def get_lon_lat_of_vehicle(license_plate: str) -> dict:
    vehicle = FMM.get_vehicle_by_license_plate(client_hack, license_plate)
    return {'lon': vehicle['lon'], 'lat': vehicle['lat']}


def get_all_charging_stations():
    stations = client_hack.get_database('fmm').get_collection('vehicle').find(
        filter={'status': 'blue'}
    )

    return {station['plate']: {'lon': station['lon'], 'lat': station['lat']}
            for station in stations}


def get_tasks_for_vehicle(license_plate: str):
    return FMM.get_tasks(
        client_hack,
        {
            'vehicleId': FMM.get_vehicle_by_license_plate(client_hack,
                                                          license_plate)
        }
    )


def delete_tasks_for_vehicle(license_plate: str):
    vehicle_id = FMM.get_vehicle_by_license_plate(client_hack,
                                                  license_plate)['_id']

    FMM.delete_tasks(client_hack, {'vehicleId': vehicle_id})


if __name__ == '__main__':
    """
    vehicle = FMM.get_vehicle_by_license_plate(client_hack, 'NH-0')
    print(get_fuel_level(plate))
    print(set_fuel_level(plate, 120))
    print(get_fuel_level(plate))
    print(get_tasks_for_vehicle(plate))
    
    # delete_tasks_for_vehicle(plate)
    print([task for task in get_tasks_for_vehicle(plate)])
    # kill_12v_battery(plate)
    revive_12v_battery(plate)
    # delete_tasks_for_vehicle(plate)
    """
    create_app_opening('NH-0')
    """
    for plate in all_license_plates():
        vehicle = FMM.get_vehicle_by_license_plate(client_hack, plate)
        client_hack.get_database('fmm').get_collection('vehicle').update_many(
            {}, {'$set': {'locationAlias': 'newhack'}}
        )
    """