Fix get_plp_info: Box with diameter

luizalabs · Jan 9, 2018 · a30afa4 · a30afa4
1 parent 718c43b
commit a30afa4
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/correios/client.py b/correios/client.py
@@ -268,10 +268,18 @@ def _load_receiver_address(self, data) -> ReceiverAddress:
  def _load_package(self, data) -> Package:
  dimensions = data.dimensao_objeto
 
+ type_ = dimensions.tipo_objeto.text.strip()
+ type_ = int(type_) if type_.isdigit() else None
+
+ diameter = float(
+ dimensions.dimensao_diametro.text.replace(',', '.')
+ )
+
+ if type_ == Package.TYPE_BOX:
+ diameter = 0
+
  package = Package(
- diameter=float(
- dimensions.dimensao_diametro.text.replace(',', '.')
- ),
+ diameter=diameter,
  height=float(dimensions.dimensao_altura.text.replace(',', '.')),
  length=float(
  dimensions.dimensao_comprimento.text.replace(',', '.')