refactor: add type hints to functions

Author: jakr-s

implement-shell-tools/ls/ls.py

@@ -28,15 +28,15 @@ def parse_args():
     return parser.parse_args()
 
 
-def filter_hidden(files):
+def filter_hidden(files: list[str]) -> list[str]:
     return [name for name in files if not name.startswith(".")]
 
 
-def get_visible_entries(files, include_hidden):
+def get_visible_entries(files: list[str], include_hidden: bool):
     return files if include_hidden else filter_hidden(files)
 
 
-def format_entries(files, one_per_line):
+def format_entries(files: list[str], one_per_line: bool):
     if len(files) == 0:
         return
     print(("\n" if one_per_line else "\t").join(files))
@@ -50,8 +50,8 @@ def main():
         include_hidden = bool(args.include_hidden)
         one_per_line = bool(args.one_per_line)
 
-        result_files = []
-        result_dirs = {}
+        result_files: list[str] = []
+        result_dirs: dict[str, list[str]] = {}
 
         for file_path in file_paths:
             st = os.stat(file_path)

implement-shell-tools/wc/wc.py

@@ -37,8 +37,8 @@ def main():
     args = parse_args()
 
     try:
-        file_paths = args.paths
-        results = {}
+        file_paths: list[str] = args.paths
+        results: dict[str, dict[str, int]] = {}
 
         for file_path in file_paths:
             stats = os.stat(file_path)
@@ -62,7 +62,7 @@ def main():
             results["total"] = total
 
         no_options_provided = not (args.lines or args.words or args.bytes)
-        selected_option_keys = []
+        selected_option_keys: list[str] = []
 
         if args.lines:
             selected_option_keys.append("lines")
@@ -73,7 +73,7 @@ def main():
 
         output_columns = [
             "lines", "words", "bytes"] if no_options_provided else selected_option_keys
-        rows = []
+        rows: list[list[str | int]] = []
         for name, values in results.items():
             rows.append([name] + [values[column] for column in output_columns])