feat: add file reader

main.py

@@ -2,6 +2,8 @@
 
 from pathlib import Path
 
+from data.visualize import visualize
+
 from loguru import logger
 
 logger.add(
@@ -15,7 +17,7 @@ logger.add(
 
 @logger.catch
 def main() -> None:
-    pass
+    visualize()
 
 
 if __name__ == "__main__":

pyproject.toml

@@ -6,15 +6,21 @@ authors = [{ name = "Kristofers Solo", email = "dev@kristofers.xyz" }]
 readme = "README.md"
 requires-python = ">=3.11"
 license = { text = "MIT" }
-dependencies = ["attrs==23.1.0", "loguru==0.7.2"]
+dependencies = [
+  "openpyxl==3.1.2",
+  "attrs==23.1.0",
+  "loguru==0.7.2",
+  "matplotlib==3.8.2",
+  "pandas==2.1.4",
+]
 
 [tool.mypy]
 check_untyped_defs = true
 disallow_any_generics = true
 ignore_missing_imports = true
-no_implicit_optional = true
-no_implicit_reexport = true
-show_error_codes = true
+# no_implicit_optional = true
+# no_implicit_reexport = true
+# show_error_codes = true
 strict_equality = true
 warn_redundant_casts = true
 warn_return_any = true

requirements.txt

@@ -1,2 +1,5 @@
 attrs>=23.1.0
 loguru>=0.7.2
+matplotlib>=3.8.2
+openpyxl>=3.1.2
+pandas>=2.1.4

src/data/read.py

@@ -0,0 +1,13 @@
+from pathlib import Path
+
+import pandas as pd
+
+
+def read_file(file_name: Path) -> pd.DataFrame:
+    """Reads a file and returns a pandas DataFrame."""
+    return pd.read_excel(file_name)
+
+
+def base_path() -> Path:
+    """Returns the base path of the project."""
+    return Path(__file__).parent.parent.parent

src/data/visualize.py

@@ -0,0 +1,24 @@
+import matplotlib.pyplot as plt
+
+from data.read import base_path, read_file
+
+
+# function that visualizes the data in the form of a bar chart
+def visualize():
+    # read the data from the file
+    path = base_path().joinpath("data", "vejaAtrumsFaktiskais.xlsx")
+    data = read_file(path)
+    # create a list of the years
+    years = [year for year in data.keys()]
+    # create a list of the number of movies per year
+    num_movies = [len(data[year]) for year in data.keys()]
+    # create a bar chart
+    plt.bar(years, num_movies)
+    # set the title of the bar chart
+    plt.title("Number of Movies per Year")
+    # set the x-axis label
+    plt.xlabel("Year")
+    # set the y-axis label
+    plt.ylabel("Number of Movies")
+    # show the bar chart
+    plt.show()