academon/core/models/vessel.py

from django.db import models

from typing import List, Dict

from pycruisemapper.classes import CruiseMapper, Ship


class Vessel(models.Model):
    name = models.CharField(max_length=64)
    imo = models.IntegerField("IMO", null=True, blank=True)
    mmsi = models.IntegerField("MMSI", null=True, blank=True)

    def get_names(self) -> List[str]:
        names = [self.name]

        for alias in self.vesselalias_set:
            names.append(alias.alias)

        return names

    def name_dict(cls) -> Dict:
        names = {}

        for vessel in cls.objects.all():
            for name in vessel.get_names():
                names[name] = vessel

        return names

    def query_cruisemapper(self) -> Ship:
        cm = CruiseMapper()
        all_ships = cm.get_ships()

        try:
            if self.imo:
                filtered = filter(lambda x: x.imo == self.imo, all_ships)
            elif self.mmsi:
                filtered = filter(lambda x: x.mmsi == self.mmsi, all_ships)
            else:
                return False

            ship = list(filtered)[0]

        except:
            return False

        try:
            ship = cm.fill_ship(ship)
        except:
            pass

        return ship