From 43418c5314016274f53a166c1c283b0b710a283b Mon Sep 17 00:00:00 2001
From: Osvaldo Santana Neto <github@osantana.me>
Date: Wed, 19 Apr 2017 10:32:44 -0300
Subject: [PATCH] Add type annotations for new properties

---
 correios/models/posting.py | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/correios/models/posting.py b/correios/models/posting.py
index 4000124..f345c29 100644
--- a/correios/models/posting.py
+++ b/correios/models/posting.py
@@ -257,11 +257,11 @@ class Package:
 
     def __init__(self,
                  package_type: int = TYPE_BOX,
-                 width: Union[float, int] = 0,
-                 height: Union[float, int] = 0,
-                 length: Union[float, int] = 0,
-                 diameter: Union[float, int] = 0,
-                 weight: Union[float, int] = 0,
+                 width: Union[float, int] = 0,  # cm
+                 height: Union[float, int] = 0,  # cm
+                 length: Union[float, int] = 0,  # cm
+                 diameter: Union[float, int] = 0,  # cm
+                 weight: Union[float, int] = 0,  # g
                  sequence=(1, 1),
                  service: Optional[Union[Service, int]] = None) -> None:
 
@@ -278,13 +278,13 @@ def __init__(self,
         self.real_width = width  # cm
         self.real_height = height  # cm
         self.real_length = length  # cm
-        self.real_diameter = diameter  # in cm
-        self.real_weight = diameter  # in grams
+        self.real_diameter = diameter  # cm
+        self.real_weight = diameter  # g
         self.sequence = sequence
         self.service = service
 
     @property
-    def width(self):
+    def width(self) -> int:
         return max(MIN_WIDTH, int(math.ceil(self.real_width)))
 
     @width.setter
@@ -293,7 +293,7 @@ def width(self, width):
         self.real_width = width
 
     @property
-    def height(self):
+    def height(self) -> int:
         return max(MIN_HEIGHT, int(math.ceil(self.real_height)))
 
     @height.setter
@@ -302,7 +302,7 @@ def height(self, height):
         self.real_height = height
 
     @property
-    def length(self):
+    def length(self) -> int:
         return max(MIN_LENGTH, int(math.ceil(self.real_length)))
 
     @length.setter
@@ -311,7 +311,7 @@ def length(self, length):
         self.real_length = length
 
     @property
-    def diameter(self):
+    def diameter(self) -> int:
         return max(MIN_DIAMETER, int(math.ceil(self.real_diameter)))
 
     @diameter.setter
@@ -320,7 +320,7 @@ def diameter(self, diameter):
         self.real_diameter = diameter
 
     @property
-    def weight(self):
+    def weight(self) -> int:
         return int(math.ceil(self.real_weight))
 
     @weight.setter