Create rule to not try Receipt when is not processed

luizalabs · Mar 6, 2020 · 62f8876 · 62f8876
1 parent f9e1131
commit 62f8876
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/correios/client.py b/correios/client.py
@@ -141,7 +141,10 @@ def build_post_info(self, data, user: User, validate_package: bool = True) -> Po
         )
         return post_info
 
-    def build_receipt(self,  data) -> Receipt:
+    def build_receipt(self,  data) -> Optional[Receipt]:
+        if data.status_processamento != Receipt.STATUS_PROCESSED:
+            return None
+
         receipt = Receipt(
             number=data.numero_comprovante_postagem,
             post_date=data.data_postagem_sara.text,

diff --git a/correios/models/posting.py b/correios/models/posting.py
@@ -452,6 +452,9 @@ def _validate_weight(cls, weight, service: Optional[Union[Service, str, int]] =
 
 class Receipt:
 
+    STATUS_UNPROCESSED = 0
+    STATUS_PROCESSED = 1
+
     def __init__(
         self,
         number: Union[int, str],