Added agents_info supplier_info

basemodels.py

@@ -37,6 +37,8 @@ class PayingAgent(BaseModel):
 
 class SupplierInfo(BaseModel):
     phones: Optional[List[PhoneNumber]] = None
+    name: Optional[str] = None
+    inn: Optional[str] = None
 
 
 class ReceivePaymentsOperator(BaseModel):
@@ -183,21 +185,23 @@ class SellBaseModel(BaseModel):
     def parse(self, db_dict_list):
         sell_list = []
         for db_dict in db_dict_list:
-            receipt = db_dict['reciept']
-            sell = db_dict['sell']
-            vats = db_dict['vats']
-            payments = db_dict['payments']
-            items = db_dict['items']
-            company = db_dict['company']
-            client = db_dict['client']
-            items_model = [Item(**item) for item in items]
+            receipt = db_dict.get('reciept')
+            sell = db_dict.get('sell')
+            vats = db_dict.get('vats')
+            payments = db_dict.get('payments')
+            items = db_dict.get('items')
+            company = db_dict.get('company')
+            client = db_dict.get('client')
             payments_model = [Payment(**payment) for payment in payments]
-            vats_model = [Vat(**vat) for vat in vats]
+            items_model = [Item(**item) for item in items]
+            #vats_model = [Vat(**vat) for vat in vats]
 
             receipt['client'] = ClientItem(
                 **client if client.get('phone') != None else {'email': 'test@test.ru'})
-            receipt['vats'] = vats_model
+            #receipt['vats'] = vats_model if len(vats_model) > 0 else None
+
             receipt['items'] = items_model
+            receipt['vats'] = None
             receipt['payments'] = payments_model
             receipt['company'] = Company(**company)
             receipt_model = Receipt(**receipt)