Define configuration options

Author: peterjc

flake8_rst_docstrings.py

@@ -976,6 +976,32 @@ def __init__(self, tree, filename="(none)"):
             self.source = None
             self.err = err
 
+    @classmethod
+    def add_options(cls, parser):
+        """Add RST directives and roles options."""
+        parser.add_option(
+            "--rst-directives",
+            metavar="LIST",
+            default="",
+            parse_from_config=True,
+            comma_separated_list=True,
+            help="Comma-separated list of additional RST directives.",
+        )
+        parser.add_option(
+            "--rst-roles",
+            metavar="LIST",
+            default="",
+            parse_from_config=True,
+            comma_separated_list=True,
+            help="Comma-separated list of additional RST roles.",
+        )
+
+    @classmethod
+    def parse_options(cls, options):
+        """Adding black-config option."""
+        cls.extra_directives = options.rst_directives
+        cls.extra_roles = options.rst_roles
+
     def run(self):
         """Use docutils to check docstrings are valid RST."""
         # Is there any reason not to call load_source here?

tests/run_tests.sh

@@ -26,6 +26,12 @@ echo "flake8 --select RST test_cases/"
 flake8 --select RST test_cases/
 echo "Good, no RST style violations reported, as expected."
 
+echo "========="
+echo "Help text"
+echo "========="
+flake8 -h | grep -i RST
+echo "Good, RST options appear in the help text"
+
 echo "============"
 echo "Tests passed"
 echo "============"