From b181130ef3973c55661b78d158488ffbc20a0deb Mon Sep 17 00:00:00 2001
From: Igor Radovanovic <74266147+IgorWounds@users.noreply.github.com>
Date: Wed, 12 Jul 2023 16:55:33 +0200
Subject: [PATCH] Cash flow

---
 .../builtin_providers/fmp/balance_sheet.py    | 64 ++++++++++---------
 .../model/data/balance_sheet.py               |  2 -
 2 files changed, 33 insertions(+), 33 deletions(-)

diff --git a/openbb_providers/builtin_providers/fmp/balance_sheet.py b/openbb_providers/builtin_providers/fmp/balance_sheet.py
index af82f0c8c02..2bd639c3c54 100644
--- a/openbb_providers/builtin_providers/fmp/balance_sheet.py
+++ b/openbb_providers/builtin_providers/fmp/balance_sheet.py
@@ -14,6 +14,7 @@ from openbb_provider.model.data.balance_sheet import (
     BalanceSheetQueryParams,
 )
 from openbb_provider.provider.abstract.fetcher import Fetcher
+from openbb_provider.provider.provider_helpers import data_transformer
 
 # IMPORT THIRD-PARTY
 from pydantic import Field, NonNegativeInt, root_validator
@@ -135,35 +136,36 @@ class FMPBalanceSheetFetcher(
     def transform_data(
         data: List[FMPBalanceSheetData],
     ) -> List[BalanceSheetData]:
-        final_items = []
-        for item in data:
-            tot_se = item.totalStockholdersEquity
-            minority = item.minorityInterest
-            parent: Optional[int] = None
-            if tot_se:
-                if tot_se is None:
-                    pass
-                elif minority is None:
-                    parent = tot_se
-                else:
-                    parent = tot_se - minority
+        # final_items = []
+        # for item in data:
+        #     tot_se = item.totalStockholdersEquity
+        #     minority = item.minorityInterest
+        #     parent: Optional[int] = None
+        #     if tot_se:
+        #         if tot_se is None:
+        #             pass
+        #         elif minority is None:
+        #             parent = tot_se
+        #         else:
+        #             parent = tot_se - minority
 
-            final = BalanceSheetData(
-                date=item.date,
-                # symbol=item.symbol,
-                # currency=item.reportedCurrency,
-                cik=item.cik,
-                # period=item.period,
-                assets=item.totalAssets,
-                current_assets=item.totalCurrentAssets,
-                current_liabilities=item.totalCurrentLiabilities,
-                equity=item.totalStockholdersEquity,
-                equity_attributable_to_noncontrolling_interest=item.minorityInterest,
-                equity_attributable_to_parent=parent,
-                liabilities=item.totalLiabilities,
-                liabilities_and_equity=item.totalLiabilitiesAndTotalEquity,
-                noncurrent_assets=item.totalNonCurrentAssets,
-                noncurrent_liabilities=item.totalNonCurrentLiabilities,
-            )
-            final_items.append(final)
-        return final_items
+        # final = BalanceSheetData(
+        #     date=item.date,
+        #     # symbol=item.symbol,
+        #     # currency=item.reportedCurrency,
+        #     cik=item.cik,
+        #     # period=item.period,
+        #     assets=item.totalAssets,
+        #     current_assets=item.totalCurrentAssets,
+        #     current_liabilities=item.totalCurrentLiabilities,
+        #     equity=item.totalStockholdersEquity,
+        #     equity_attributable_to_noncontrolling_interest=item.minorityInterest,
+        #     equity_attributable_to_parent=parent,
+        #     liabilities=item.totalLiabilities,
+        #     liabilities_and_equity=item.totalLiabilitiesAndTotalEquity,
+        #     noncurrent_assets=item.totalNonCurrentAssets,
+        #     noncurrent_liabilities=item.totalNonCurrentLiabilities,
+        # )
+        # final_items.append(final)
+        return data_transformer(data, BalanceSheetData)
+        # return final_items
diff --git a/openbb_providers/openbb_provider/model/data/balance_sheet.py b/openbb_providers/openbb_provider/model/data/balance_sheet.py
index c680008c258..2c871f7deba 100644
--- a/openbb_providers/openbb_provider/model/data/balance_sheet.py
+++ b/openbb_providers/openbb_provider/model/data/balance_sheet.py
@@ -54,9 +54,7 @@ class BalanceSheetData(Data):
     """
 
     date: dateType
-    # currency: Optional[str]
     cik: Optional[int]
-    # period: Optional[str]
     assets: Optional[int]
     current_assets: Optional[int]
     current_liabilities: Optional[int]