cmc/cleberg.net

My personal web garden & blog.

clone: git clone https://gitbay.org/cmc/cleberg.net.git

9007704c4b298a00b971e2ab80fdc404d8570edc

unsigned

author: Christian Cleberg <hello@cleberg.net> · 2024-10-28T01:46:22Z

comply with pylint

 .gitignore                    |  1 +
 utils/salary_visualization.py | 37 ++++++++++++++++++++-----------------
 2 files changed, 21 insertions(+), 17 deletions(-)

diff --git a/.gitignore b/.gitignore
index b9090b3..e2ad1cc 100644
--- a/.gitignore
+++ b/.gitignore
@@ -3,3 +3,4 @@
 .vscode
 public
 readme.html
+venv
diff --git a/utils/salary_visualization.py b/utils/salary_visualization.py
index 99b29fc..311e3c2 100644
--- a/utils/salary_visualization.py
+++ b/utils/salary_visualization.py
@@ -1,8 +1,10 @@
-%pip install plotly pandas
+"""
+A file that will visualize salary data from an incoming CSV.
+"""
 
+import locale
 import pandas as pd
 import plotly.graph_objs as go
-import locale
 
 locale.setlocale(locale.LC_ALL, 'en_US.UTF-8')
 
@@ -10,6 +12,9 @@ df = pd.read_csv('~/git/cleberg.net/static/salary.csv')
 
 # Function to format salary as US currency
 def format_currency(value):
+    """
+    Format values in USD currency format.
+    """
     return f"${value:,.2f}"
 
 # Reverse the order of the DataFrame
@@ -23,7 +28,7 @@ fig = go.Figure()
 
 # Adding each data point as a separate trace to display the text
 for index, row in df.iterrows():
-    title_company = f"{row['Title']} ({row['Company']})"
+    TITLE_COMPANY = f"{row['Title']} ({row['Company']})"
     salary_formatted = format_currency(row['Salary'])
     if pd.notna(row['Percentage Increase']):
         text = f"{salary_formatted} ({row['Percentage Increase']:.2f}%)"
@@ -34,27 +39,25 @@ for index, row in df.iterrows():
         y=[row['Salary'], row['Salary']],
         text=[text],
         mode='lines+text',
-        name=title_company,  # Combine title and company for legend
+        name=TITLE_COMPANY,
         textposition='top center'
     ))
 
+# Update visual styles of the figure
 fig.update_layout(
     title="Salary Data Over Time (annualized)",
     xaxis_title="Time",
     yaxis_title="Salary",
-    font=dict(
-        family="monospace",
-        size=16
-    ),
-    margin=dict(l=50, r=50, t=50, b=100),  # Adjust bottom margin for legend space
-    legend=dict(
-        orientation="h",  # Horizontal orientation
-        yanchor="top",    # Anchor to the top of the legend box
-        y=-0.3,           # Position below the graph
-        xanchor="center", # Center the legend horizontally
-        x=0.5             # Center the legend
-    ),
-    height=800  # Increase vertical size
+    font={"family": 'monospace', "size": 16},
+    margin={"l": 50, "r": 50, "t": 50, "b": 100},
+    legend={
+	"orientation": 'h',
+	"yanchor": 'top',
+	"y": -0.3,
+	"xanchor": 'center',
+	"x": 0.5
+    },
+    height=800
 )
 
 fig.show()