presupuesto.py

import sys

mul_por_tipo = {
    'local': 1,
    'domiciliario': 1.5,
}

costos_insumos = {
    'fusible': 100,
    'cable_15_m': 200,
    'cable_25_m': 400,
}

precio_hora = 1000


def presupuesto(horas: float, tipo: str, insumo_a:float, insumo_b: float, insumo_c: float):
    mod_tipo = mul_por_tipo[tipo]
    return horas * mod_tipo * precio_hora + insumo_a * costos_insumos['fusible'] \
        + insumo_b * costos_insumos['cable_15_m'] + insumo_c * costos_insumos['cable_25_m']


if __name__ == "__main__":
    print(presupuesto(float(sys.argv[1]), sys.argv[2], float(sys.argv[3]),
                      float(sys.argv[4]), float(sys.argv[5])))