Improve mypy support

Author: twizmwazin

angrmanagement/plugins/plugin_description.py

@@ -1,14 +1,17 @@
 import pathlib
 from dataclasses import field
-from typing import Dict, List, Optional
+from typing import ClassVar, Dict, List, Optional, Type
 
 import marshmallow.validate
 import tomlkit
+from marshmallow import Schema
 from marshmallow_dataclass import dataclass
 
 
 @dataclass
 class MetadataDescription:
+    Schema: ClassVar[Type[Schema]] = Schema  # placate mypy
+
     version: int = field(metadata={"validate": marshmallow.validate.OneOf([0])})
 
 
@@ -18,6 +21,8 @@ class PackageDescription:
     Describes a plugin package.
     """
 
+    Schema: ClassVar[Type[Schema]] = Schema  # placate mypy
+
     name: str = field()
     version: str = field()
     platforms: List[str] = field(default_factory=lambda: ["any"])
@@ -33,6 +38,8 @@ class PluginDescription:
     Describes an angr management plugin. Can be generated from plugin.toml.
     """
 
+    Schema: ClassVar[Type[Schema]] = Schema  # placate mypy
+
     name: str = field()
     entrypoint: str = field()
     platforms: Optional[List[str]] = field(default=None)
@@ -46,6 +53,8 @@ class PluginConfigFileDescription:
     Describes a plugin config file.
     """
 
+    Schema: ClassVar[Type[Schema]] = Schema  # placate mypy
+
     metadata: MetadataDescription = field()
     package: PackageDescription = field()
     plugins: Dict[str, PluginDescription] = field(default_factory=dict)

angrmanagement/py.typed

@@ -0,0 +1 @@
+PARTIAL

setup.cfg

@@ -49,6 +49,7 @@ pyinstaller =
 
 [options.package_data]
 angrmanagement =
+    py.typed
     resources/fonts/*.ttf
     resources/images/*
     resources/themes/**/*