package Aerei;

import java.util.HashMap;
import java.util.Map;


public class AeroportiMondo {
    Map<String, Aeroporto> aeroporti = new HashMap<String, Aeroporto>();

    public void addAeroporto(Aeroporto arpt) throws InvalidCode{

String cod = arpt.getCode();
if (aeroporti.containsKey(cod)){throw new InvalidCode();}
aeroporti.put(cod, arpt);

    }

    public  void removeAeroporto(String cod) {
    aeroporti.remove(cod);

    }

    public  Aeroporto getAeroporto(String cod) {

    if ( !  aeroporti.containsKey(cod)) return null;
        return aeroporti.get(aeroporti);
    }