fixed mypy issues

Author: felixscode

README.md

@@ -7,6 +7,7 @@ A simple and minimalistic Config Manager using YAML.
 [![Tests](https://github.com/felixscode/heracless/actions/workflows/test.yml/badge.svg)](https://github.com/felixscode/heracless/actions/workflows/test.yml)
 [![Publish](https://github.com/felixscode/heracless/actions/workflows/publish.yml/badge.svg)](https://github.com/felixscode/heracless/actions/workflows/publish.yml)
 [![codecov](https://codecov.io/gh/felixscode/heracless/branch/main/graph/badge.svg)](https://codecov.io/gh/felixscode/heracless)
+[![mypy](https://img.shields.io/badge/mypy-checked-blue)](http://mypy-lang.org/)
 [![Python Version](https://img.shields.io/pypi/pyversions/heracless)](https://pypi.org/project/heracless/)
 [![PyPI version](https://badge.fury.io/py/heracless.svg)](https://badge.fury.io/py/heracless)
 [![GitHub](https://img.shields.io/badge/GitHub-Repository-blue?logo=github)](https://github.com/felixscode/heracless)

heracless/utils/cfg_tree.py

@@ -152,6 +152,7 @@ def tree_builder(obj_type: Type[Node], name: str, value: Union[Value, Iterable])
 
     iterables = iterable_generator(value, name)
     # Handle empty iterables (e.g., empty dict)
+    children: tuple[Union[Node, Tree], ...]
     if not iterables:
         children = tuple()
     else:

heracless/wapp/pages/ConfigGenerator.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 import streamlit as st
 from streamlit_ace import st_ace
 from streamlit_ace import THEMES
@@ -28,7 +30,7 @@
 )
 if st.button("Generate ⚡"):
     if is_valid_yaml(content):
-        stub_string, template_str = get_file_content(content, config_dir, frozen)
+        stub_string, template_str = get_file_content(content, Path(config_dir), frozen)
         st.write("### Python Stub File")
         st.code(stub_string, language="python", line_numbers=True)
         download_button(label="Download 📥", data=stub_string, file_name=load_file_name.split(".")[0] + ".pyi")

tests/config/types.pyi

@@ -5,11 +5,10 @@ from pathlib import Path
 
 
 @dataclass(frozen=True)
-class Address:
-    lines: str
-    city: str
-    state: str
-    postal: int
+class ShipTo:
+    given: str
+    family: str
+    address: "Address"
 
 
 @dataclass(frozen=True)
@@ -33,14 +32,15 @@ class ProductItem:
 
 
 @dataclass(frozen=True)
-class BillTo:
-    given: str
-    family: str
-    address: "Address"
+class Address:
+    lines: str
+    city: str
+    state: str
+    postal: int
 
 
 @dataclass(frozen=True)
-class ShipTo:
+class BillTo:
     given: str
     family: str
     address: "Address"

tests/conftest.pyi

@@ -5,11 +5,10 @@ from pathlib import Path
 
 
 @dataclass(frozen=True)
-class Address:
-    lines: str
-    city: str
-    state: str
-    postal: int
+class ShipTo:
+    given: str
+    family: str
+    address: "Address"
 
 
 @dataclass(frozen=True)
@@ -33,14 +32,15 @@ class ProductItem:
 
 
 @dataclass(frozen=True)
-class BillTo:
-    given: str
-    family: str
-    address: "Address"
+class Address:
+    lines: str
+    city: str
+    state: str
+    postal: int
 
 
 @dataclass(frozen=True)
-class ShipTo:
+class BillTo:
     given: str
     family: str
     address: "Address"