Merge pull request #156 from commitizen-tools/fix-mypy-check

Fix mypy check

Author: Lee-W

.flake8

@@ -66,10 +66,12 @@ def prerelease_generator(current_version: str, prerelease: Optional[str] = None)
         return ""
 
     version = Version(current_version)
-    new_prerelease_number: int = 0
-    if version.is_prerelease and prerelease.startswith(version.pre[0]):
-        prev_prerelease: int = list(version.pre)[1]
+    # version.pre is needed for mypy check
+    if version.is_prerelease and version.pre and prerelease.startswith(version.pre[0]):
+        prev_prerelease: int = version.pre[1]
         new_prerelease_number = prev_prerelease + 1
+    else:
+        new_prerelease_number = 0
     pre_version = f"{prerelease}{new_prerelease_number}"
     return pre_version
 
@@ -176,7 +178,8 @@ def create_tag(version: Union[Version, str], tag_format: Optional[str] = None):
 
     major, minor, patch = version.release
     prerelease = ""
-    if version.is_prerelease:
+    # version.pre is needed for mypy check
+    if version.is_prerelease and version.pre:
         prerelease = f"{version.pre[0]}{version.pre[1]}"
 
     t = Template(tag_format)

commitizen/bump.py

@@ -1,6 +1,6 @@
 import os
 import re
-from typing import Dict
+from typing import Dict, Optional
 
 from commitizen import factory, git, out
 from commitizen.config import BaseConfig
@@ -22,8 +22,8 @@ def __init__(self, config: BaseConfig, arguments: Dict[str, str], cwd=os.getcwd(
             the flags provided by the user
 
         """
-        self.commit_msg_file: str = arguments.get("commit_msg_file")
-        self.rev_range: str = arguments.get("rev_range")
+        self.commit_msg_file: Optional[str] = arguments.get("commit_msg_file")
+        self.rev_range: Optional[str] = arguments.get("rev_range")
 
         self._valid_command_argument()
 
@@ -76,4 +76,4 @@ def _get_commit_messages(self):
     def validate_commit_message(commit_msg: str, pattern: str) -> bool:
         if commit_msg.startswith("Merge") or commit_msg.startswith("Revert"):
             return True
-        return re.match(pattern, commit_msg)
+        return bool(re.match(pattern, commit_msg))

commitizen/commands/check.py

@@ -1,6 +1,6 @@
 import warnings
 from pathlib import Path
-from typing import Optional
+from typing import Optional, Union
 
 from commitizen import defaults, git, out
 from commitizen.error_codes import NOT_A_GIT_PROJECT
@@ -29,7 +29,7 @@ def load_global_conf() -> Optional[IniConfig]:
     with open(global_cfg, "r") as f:
         data = f.read()
 
-    conf = IniConfig(data)
+    conf = IniConfig(data=data, path=global_cfg)
     return conf
 
 
@@ -56,6 +56,7 @@ def read_cfg() -> BaseConfig:
         with open(filename, "r") as f:
             data: str = f.read()
 
+        _conf: Union[TomlConfig, IniConfig]
         if "toml" in filename.suffix:
             _conf = TomlConfig(data=data, path=filename)
         else:

commitizen/config/__init__.py

@@ -1,20 +1,21 @@
 import warnings
-from typing import Optional
+from pathlib import Path
+from typing import Any, Dict, Optional, Union
 
 from commitizen.defaults import DEFAULT_SETTINGS
 
 
 class BaseConfig:
     def __init__(self):
-        self._settings: dict = DEFAULT_SETTINGS.copy()
-        self._path: Optional[str] = None
+        self._settings: Dict[str, Any] = DEFAULT_SETTINGS.copy()
+        self._path: Optional[Path] = None
 
     @property
-    def settings(self) -> dict:
+    def settings(self) -> Dict[str, Any]:
         return self._settings
 
     @property
-    def path(self) -> str:
+    def path(self) -> Optional[Path]:
         return self._path
 
     def set_key(self, key, value):
@@ -28,8 +29,8 @@ def set_key(self, key, value):
     def update(self, data: dict):
         self._settings.update(data)
 
-    def add_path(self, path: str):
-        self._path = path
+    def add_path(self, path: Union[str, Path]):
+        self._path = Path(path)
 
     def _parse_setting(self, data: str) -> dict:
         raise NotImplementedError()

commitizen/config/base_config.py

@@ -1,12 +1,14 @@
 import configparser
 import json
 import warnings
+from pathlib import Path
+from typing import Union
 
 from .base_config import BaseConfig
 
 
 class IniConfig(BaseConfig):
-    def __init__(self, *, data: str, path: str):
+    def __init__(self, *, data: str, path: Union[Path, str]):
         warnings.simplefilter("always", DeprecationWarning)
         warnings.warn(
             (

commitizen/config/ini_config.py

@@ -1,10 +1,13 @@
+from pathlib import Path
+from typing import Union
+
 from tomlkit import exceptions, parse
 
 from .base_config import BaseConfig
 
 
 class TomlConfig(BaseConfig):
-    def __init__(self, *, data: str, path: str):
+    def __init__(self, *, data: str, path: Union[Path, str]):
         super(TomlConfig, self).__init__()
         self.is_empty_config = False
         self._parse_setting(data)

commitizen/config/toml_config.py

@@ -1,17 +1,19 @@
 import importlib
 import pkgutil
+from typing import Dict, Type
 
+from commitizen.cz.base import BaseCommitizen
 from commitizen.cz.conventional_commits import ConventionalCommitsCz
 from commitizen.cz.customize import CustomizeCommitsCz
 from commitizen.cz.jira import JiraSmartCz
 
-registry = {
+registry: Dict[str, Type[BaseCommitizen]] = {
     "cz_conventional_commits": ConventionalCommitsCz,
     "cz_jira": JiraSmartCz,
     "cz_customize": CustomizeCommitsCz,
 }
 plugins = {
-    name: importlib.import_module(name).discover_this
+    name: importlib.import_module(name).discover_this  # type: ignore
     for finder, name, ispkg in pkgutil.iter_modules()
     if name.startswith("cz_")
 }

commitizen/cz/__init__.py

@@ -3,6 +3,8 @@
 
 from prompt_toolkit.styles import Style, merge_styles
 
+from commitizen.config.base_config import BaseConfig
+
 
 class BaseCommitizen(metaclass=ABCMeta):
     bump_pattern: Optional[str] = None
@@ -20,7 +22,7 @@ class BaseCommitizen(metaclass=ABCMeta):
         ("disabled", "fg:#858585 italic"),
     ]
 
-    def __init__(self, config: dict):
+    def __init__(self, config: BaseConfig):
         self.config = config
         if not self.config.settings.get("style"):
             self.config.settings.update({"style": BaseCommitizen.default_style_config})
@@ -42,18 +44,18 @@ def style(self):
             ]
         )
 
-    def example(self) -> str:
+    def example(self) -> Optional[str]:
         """Example of the commit message."""
         raise NotImplementedError("Not Implemented yet")
 
-    def schema(self) -> str:
+    def schema(self) -> Optional[str]:
         """Schema definition of the commit message."""
         raise NotImplementedError("Not Implemented yet")
 
-    def schema_pattern(self) -> str:
+    def schema_pattern(self) -> Optional[str]:
         """Regex matching the schema used for message validation"""
         raise NotImplementedError("Not Implemented yet")
 
-    def info(self) -> str:
+    def info(self) -> Optional[str]:
         """Information about the standardized commit message."""
         raise NotImplementedError("Not Implemented yet")

commitizen/cz/base.py

@@ -1,4 +1,5 @@
 import os
+from typing import Any, Dict, List
 
 from commitizen import defaults
 from commitizen.cz.base import BaseCommitizen
@@ -29,8 +30,8 @@ class ConventionalCommitsCz(BaseCommitizen):
     bump_pattern = defaults.bump_pattern
     bump_map = defaults.bump_map
 
-    def questions(self) -> list:
-        questions = [
+    def questions(self) -> List[Dict[str, Any]]:
+        questions: List[Dict[str, Any]] = [
             {
                 "type": "list",
                 "name": "prefix",