improve imports

Author: Kaevan89

tests/test_errors.py

@@ -1,8 +1,8 @@
 import responses
 import requests
 
-from veryfi.errors import *
-from veryfi import *
+from veryfi.errors import BadRequest, VeryfiClientError
+from veryfi import Client
 import pytest
 
 

veryfi/_documents/line_items.py

@@ -1,4 +1,4 @@
-from typing import *
+from typing import Dict
 
 from veryfi.client_base import Client
 
@@ -7,7 +7,7 @@ class LineItems:
     def __init__(self, client: Client):
         self.client = client
 
-    def get_line_items(self, document_id):
+    def get_line_items(self, document_id: int):
         """
         Retrieve all line items for a document.
         https://docs.veryfi.com/api/receipts-invoices/get-document-line-items/
@@ -17,7 +17,7 @@ def get_line_items(self, document_id):
         """
         return self.client._request("GET", f"/documents/{document_id}/line-items/")
 
-    def get_line_item(self, document_id, line_item_id):
+    def get_line_item(self, document_id: int, line_item_id: int):
         """
         Retrieve a line item for existing document by ID.
         https://docs.veryfi.com/api/receipts-invoices/get-a-line-item/
@@ -53,7 +53,7 @@ def update_line_item(self, document_id: int, line_item_id: int, payload: Dict) -
             "PUT", f"/documents/{document_id}/line-items/{line_item_id}", payload
         )
 
-    def delete_line_items(self, document_id):
+    def delete_line_items(self, document_id: int):
         """
         Delete all line items on an existing document.
         https://docs.veryfi.com/api/receipts-invoices/delete-all-document-line-items/
@@ -62,7 +62,7 @@ def delete_line_items(self, document_id):
         """
         self.client._request("DELETE", f"/documents/{document_id}/line-items/")
 
-    def delete_line_item(self, document_id, line_item_id):
+    def delete_line_item(self, document_id: int, line_item_id: int):
         """
         Delete an existing line item on an existing document.
         https://docs.veryfi.com/api/receipts-invoices/delete-a-line-item/

veryfi/_documents/pdf_split.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, List, Optional
 
 from veryfi.client_base import Client
 

veryfi/_documents/tags.py

@@ -1,4 +1,3 @@
-from typing import *
 from veryfi.client_base import Client
 
 

veryfi/_w2s/w2_split.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, List, Optional
 
 from veryfi.client_base import Client
 

veryfi/a_docs.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, List, Optional
 
 from veryfi.client_base import Client
 

veryfi/bank_statements.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, Optional
 
 from veryfi.client_base import Client
 
@@ -33,7 +33,7 @@ def process_bank_statement_document_url(
 
     def process_bank_statement_document(
         self, file_path: str, file_name: Optional[str] = None, **kwargs
-    ):
+    ) -> Dict:
         """
         Process bank statement document from url and extract all the fields from it.
         https://docs.veryfi.com/api/bank-statements/process-a-bank-statement/

veryfi/bussines_cards.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, Optional
 
 from veryfi.client_base import Client
 
@@ -33,7 +33,7 @@ def process_bussines_card_document_url(
 
     def process_bussines_card_document(
         self, file_path: str, file_name: Optional[str] = None, **kwargs
-    ):
+    ) -> Dict:
         """
         Process bussiness card from url and extract all the fields from it.
         https://docs.veryfi.com/api/business-cards/process-a-business-card/
@@ -55,7 +55,7 @@ def process_bussines_card_document(
         request_arguments.update(kwargs)
         return self.client._request("POST", endpoint_name, request_arguments)
 
-    def get_business_cards(self, **kwargs: Dict):
+    def get_business_cards(self, **kwargs):
         """
         Get list of business card documents.
         https://docs.veryfi.com/api/business-cards/get-business-cards/
@@ -66,7 +66,7 @@ def get_business_cards(self, **kwargs: Dict):
         endpoint_name = "/business-cards/"
         return self.client._request("GET", endpoint_name, {}, kwargs)
 
-    def get_business_card(self, document_id: int, **kwargs: Dict):
+    def get_business_card(self, document_id: int, **kwargs) -> Dict:
         """
         Get a business card document.
         https://docs.veryfi.com/api/business-cards/get-a-business-card/

veryfi/checks.py

@@ -1,6 +1,6 @@
 import os
 import base64
-from typing import *
+from typing import Dict, List, Optional
 
 from veryfi.client_base import Client
 
@@ -16,7 +16,7 @@ def get_checks(
         created_date__lt: Optional[str] = None,
         created_date__lte: Optional[str] = None,
         **kwargs,
-    ) -> List[Dict]:
+    ):
         """
         Get list of checks
         https://docs.veryfi.com/api/checks/get-checks/
@@ -59,7 +59,7 @@ def process_check(
         self,
         file_path: str,
         **kwargs,
-    ):
+    ) -> Dict:
         """
         Process a check document and extract all the fields from it
         https://docs.veryfi.com/api/checks/process-a-check/

veryfi/client.py

@@ -13,13 +13,13 @@
 class Client(ClientBase, ADocs, BankStatements, BussinesCards, Checks, Documents, W2s, W8s, W9s):
     def __init__(
         self,
-        client_id,
-        client_secret,
-        username,
-        api_key,
-        base_url=ClientBase.BASE_URL,
-        api_version=ClientBase.API_VERSION,
-        timeout=ClientBase.API_TIMEOUT,
+        client_id: str,
+        client_secret: str,
+        username: str,
+        api_key: str,
+        base_url: str = ClientBase.BASE_URL,
+        api_version: str = ClientBase.API_VERSION,
+        timeout: int = ClientBase.API_TIMEOUT,
     ):
         super().__init__(
             client_id=client_id,