Linting and add yfinance as optional dependency

Author: JerBouma

app/initalization_model.py

@@ -172,12 +172,16 @@ def create_configurations_section(session_state: st.session_state):
     with col1:
         session_state["start_date"] = st.date_input(
             "Start Date",
-            value=datetime.now() - timedelta(days=365 * 10)
-            if session_state["premium_plan"]
-            else datetime.now() - timedelta(days=365 * 5),
-            min_value=datetime(1980, 1, 1)
-            if session_state["premium_plan"]
-            else datetime.now() - timedelta(days=365 * 5),
+            value=(
+                datetime.now() - timedelta(days=365 * 10)
+                if session_state["premium_plan"]
+                else datetime.now() - timedelta(days=365 * 5)
+            ),
+            min_value=(
+                datetime(1980, 1, 1)
+                if session_state["premium_plan"]
+                else datetime.now() - timedelta(days=365 * 5)
+            ),
             max_value=datetime.now(),
         )
     with col2:
@@ -274,12 +278,16 @@ def initalize_financetoolkit(session_state: st.session_state):
             tickers=session_state["tickers"],
             api_key=session_state["api_key_value"],
             quarterly=session_state["quarterly_data"],
-            start_date=session_state["start_date"].strftime("%Y-%m-%d")
-            if session_state["start_date"]
-            else None,
-            end_date=session_state["end_date"].strftime("%Y-%m-%d")
-            if session_state["end_date"]
-            else None,
+            start_date=(
+                session_state["start_date"].strftime("%Y-%m-%d")
+                if session_state["start_date"]
+                else None
+            ),
+            end_date=(
+                session_state["end_date"].strftime("%Y-%m-%d")
+                if session_state["end_date"]
+                else None
+            ),
             progress_bar=False,
         )
 

app/metrics_view.py

@@ -8,6 +8,8 @@
 
 from financetoolkit import Toolkit
 
+# pylint: disable=R0914,W0212,R1735
+
 
 def plot_data(
     variables: list[str],
@@ -176,11 +178,11 @@ def plot_data(
             freq = (
                 "D"
                 if "MD" in variable
-                else "D"
-                if "TI" in variable
-                else "Q"
-                if finance_toolkit._quarterly
-                else "Y"
+                else (
+                    "D"
+                    if "TI" in variable
+                    else "Q" if finance_toolkit._quarterly else "Y"
+                )
             )
 
             if column_one:

dashboard.py

@@ -4,9 +4,7 @@
 
 from app import helpers, initalization_model, metrics_view, socials_model
 
-st.set_page_config(
-    page_title="Finance Toolkit Dashboard", page_icon="🛠️", layout="wide"
-)
+st.set_page_config(page_title="Finance Toolkit Dashboard", page_icon="🛠️", layout="wide")
 
 
 st.session_state = initalization_model.initalize_session_state(st.session_state)

examples/Finance Toolkit - 0. README Examples.ipynb

@@ -4092,7 +4092,6 @@
    ],
    "source": [
     "import matplotlib.pyplot as plt\n",
-    "import seaborn as sns\n",
     "\n",
     "# Display corporate bond yields\n",
     "display(corporate_bond_yields)\n",
@@ -4433,23 +4432,23 @@
    "source": [
     "display(unemployment_rates)\n",
     "\n",
-    "df = unemployment_rates[[\"Colombia\", \"United States\", \"Sweden\", \"Japan\", \"Germany\"]]\n",
+    "unemployment_rates_df = unemployment_rates[[\"Colombia\", \"United States\", \"Sweden\", \"Japan\", \"Germany\"]]\n",
     "\n",
     "# Copy to clipboard (this is just to paste the data in the README)\n",
-    "pd.io.clipboards.to_clipboard(df.to_markdown(), excel=False)\n",
+    "pd.io.clipboards.to_clipboard(unemployment_rates_df.to_markdown(), excel=False)\n",
     "\n",
     "index = pd.period_range(\"2017\", \"2022\", freq=\"Y\")\n",
     "\n",
     "# Convert PeriodIndex to string\n",
-    "df.index = df.index.astype(str)\n",
+    "unemployment_rates_df.index = unemployment_rates_df.index.astype(str)\n",
     "\n",
     "# Plotting with Seaborn for better aesthetics\n",
     "plt.figure(figsize=(12, 8))\n",
     "sns.set_style(\"whitegrid\")\n",
-    "palette = sns.color_palette(\"husl\", len(df.columns))\n",
+    "palette = sns.color_palette(\"husl\", len(unemployment_rates_df.columns))\n",
     "\n",
     "# Plotting a bar plot\n",
-    "(df * 100).plot(kind=\"bar\", width=0.8, color=palette, edgecolor=\"black\")\n",
+    "(unemployment_rates_df * 100).plot(kind=\"bar\", width=0.8, color=palette, edgecolor=\"black\")\n",
     "\n",
     "plt.title(\"Unemployment Rates Over Time\", fontsize=16)\n",
     "plt.xlabel(\"Year\", fontsize=12)\n",

examples/Finance Toolkit - 5. Options Module.ipynb

@@ -88,8 +88,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "import pandas as pd\n",
     "import matplotlib.pyplot as plt\n",
+    "import pandas as pd\n",
     "\n",
     "from financetoolkit import Toolkit\n",
     "\n",

financetoolkit/discovery/discovery_controller.py

@@ -1,4 +1,5 @@
 """Discovery Module"""
+
 __docformat__ = "google"
 
 import pandas as pd

financetoolkit/discovery/discovery_model.py

@@ -1,4 +1,5 @@
 """Discovery Model"""
+
 __docformat__ = "google"
 
 import pandas as pd

financetoolkit/economics/economics_controller.py

@@ -1,4 +1,5 @@
 """Economics Module"""
+
 __docformat__ = "google"
 
 
@@ -651,9 +652,7 @@ def get_share_prices(
         period = (
             period
             if period is not None
-            else "quarterly"
-            if self._quarterly
-            else "yearly"
+            else "quarterly" if self._quarterly else "yearly"
         )
 
         share_prices = oecd_model.get_share_prices(period=period)
@@ -728,9 +727,7 @@ def get_long_term_interest_rate(
         period = (
             period
             if period is not None
-            else "quarterly"
-            if self._quarterly
-            else "yearly"
+            else "quarterly" if self._quarterly else "yearly"
         )
 
         long_term_interest_rate = oecd_model.get_long_term_interest_rate(
@@ -809,9 +806,7 @@ def get_short_term_interest_rate(
         period = (
             period
             if period is not None
-            else "quarterly"
-            if self._quarterly
-            else "yearly"
+            else "quarterly" if self._quarterly else "yearly"
         )
 
         short_term_interest_rate = oecd_model.get_short_term_interest_rate(
@@ -884,9 +879,7 @@ def get_exchange_rates(
         period = (
             period
             if period is not None
-            else "quarterly"
-            if self._quarterly
-            else "yearly"
+            else "quarterly" if self._quarterly else "yearly"
         )
 
         exchange_rates = oecd_model.get_exchange_rates(period=period)
@@ -1184,9 +1177,7 @@ def get_unemployment_rate(
         period = (
             period
             if period is not None
-            else "quarterly"
-            if self._quarterly
-            else "yearly"
+            else "quarterly" if self._quarterly else "yearly"
         )
 
         unemployment_rate = oecd_model.get_unemployment_rate(period=period)

financetoolkit/economics/oecd_model.py

@@ -1,4 +1,5 @@
 """OECD Model"""
+
 __docformat__ = "google"
 
 import pandas as pd

financetoolkit/fixedincome/bond_model.py

@@ -1,4 +1,5 @@
 """Bond Model Module"""
+
 import numpy as np