From: Scott Gasch <scott@gasch.org>
Date: Tue, 13 Jun 2023 04:23:47 +0000 (-0700)
Subject: Adds type annotations.
X-Git-Url: https://wannabe.guru.org/gitweb/?a=commitdiff_plain;h=b1b832298313b7fe061f90722c7bca75a78f0e09;p=pyutils.git

Adds type annotations.
---

diff --git a/src/pyutils/argparse_utils.py b/src/pyutils/argparse_utils.py
index 97a5774..6920bfa 100644
--- a/src/pyutils/argparse_utils.py
+++ b/src/pyutils/argparse_utils.py
@@ -18,7 +18,7 @@ import argparse
 import datetime
 import logging
 import os
-from typing import Any
+from typing import Any, Optional
 
 from overrides import overrides
 
@@ -51,7 +51,14 @@ class ActionNoYes(argparse.Action):
 
     """
 
-    def __init__(self, option_strings, dest, default=None, required=False, help=None):
+    def __init__(
+        self,
+        option_strings: str,
+        dest: str,
+        default: Optional[str] = None,
+        required: bool = False,
+        help: Optional[str] = None,
+    ):
         if default is None:
             msg = "You must provide a default with Yes/No action"
             logger.critical(msg)
@@ -79,11 +86,12 @@ class ActionNoYes(argparse.Action):
         )
 
     @overrides
-    def __call__(self, parser, namespace, values, option_strings=None):
-        if option_strings.startswith("--no-") or option_strings.startswith("--no_"):
-            setattr(namespace, self.dest, False)
-        else:
-            setattr(namespace, self.dest, True)
+    def __call__(self, parser, namespace, values, option_strings: Optional[str] = None):
+        if option_strings is not None:
+            if option_strings.startswith("--no-") or option_strings.startswith("--no_"):
+                setattr(namespace, self.dest, False)
+            else:
+                setattr(namespace, self.dest, True)
 
 
 def valid_bool(v: Any) -> bool: