UPDATE

Author: hsz1273327

schema_entry/entrypoint.py

@@ -23,8 +23,8 @@
 from yaml import load as yaml_load
 
 from .protocol import SUPPORT_SCHEMA
-from .utils import SchemaType, get_parent_tree, parse_value_string_by_schema, parse_schema_as_cmd
-from .entrypoint_base import EntryPointABC
+from .utils import get_parent_tree, parse_value_string_by_schema, parse_schema_as_cmd
+from .entrypoint_base import SchemaType, EntryPointABC
 
 
 class EntryPoint(EntryPointABC):
@@ -125,10 +125,12 @@ def __init__(self, *,
             self._config_file_parser_map = config_file_parser_map
         if main is not None:
             self._main = main
+        else:
+            self._main = None
 
         self._check_schema()
         self._subcmds = {}
-        self._main = None
+
         self._config = {}
 
     @ property

schema_entry/entrypoint_base.py

@@ -2,8 +2,14 @@
 import abc
 import argparse
 from pathlib import Path
-from typing import Callable, Sequence, Dict, Any, Optional, Tuple
-from .utils import SchemaType
+from typing import Callable, Sequence, Dict, Any, Optional, Tuple, List
+from mypy_extensions import TypedDict
+
+
+class SchemaType(TypedDict):
+    required: List[str]
+    type: str
+    properties: Dict[str, Dict[str, Any]]
 
 
 class EntryPointABC(abc.ABC):

schema_entry/utils.py

@@ -5,16 +5,9 @@
 import warnings
 import argparse
 from typing import List, Dict, Any, Optional
-from mypy_extensions import TypedDict
 from .entrypoint_base import EntryPointABC
 
 
-class SchemaType(TypedDict):
-    required: List[str]
-    type: str
-    properties: Dict[str, Dict[str, Any]]
-
-
 def _get_parent_tree(c: EntryPointABC, result: List[str]) -> None:
     if c.parent:
         result.append(c.parent.name)

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = schema_entry
-version = 0.1.1
+version = 0.1.2
 url = https://github.com/Python-Tools/schema_entry
 author = hsz
 author_email = hsz1273327@gmail.com

tests/test_entrypoint_without_subclass.py

@@ -130,7 +130,7 @@ def test_load_json_configfile(self) -> None:
                 str(Path.home().joinpath(".test_config.json")),
                 "./test_config.json"
             ],
-            main=lambda a_a: None)
+            main=lambda a: None)
         root([])
         self.assertDictEqual(root.config, {
             "a": 1
@@ -155,7 +155,7 @@ def test_load_json_configfile_onlyneed(self) -> None:
                 },
                 "required": ["a"]
             },
-            main=lambda a_a: None
+            main=lambda a: None
         )
         root([])
         self.assertDictEqual(root.config, {
@@ -170,7 +170,7 @@ def test_load_yaml_configfile(self) -> None:
                 str(Path.home().joinpath(".test_config.yml")),
                 "./test_config.yml"
             ],
-            main=lambda a_a: None
+            main=lambda a: None
         )
 
         root([])