feat: add support for single-quoted doc

fix #84

Author: zephvr

tap/utils.py

@@ -235,12 +235,21 @@ def get_class_variables(cls: type) -> OrderedDict:
             if token['token'].strip() == '':
                 continue
 
-            # Extract multiline comments
+            # Extract multiline comments in triple quote
             if (class_variable is not None
                     and token['token_type'] == tokenize.STRING
                     and token['token'][:3] in {'"""', "'''"}):
                 sep = ' ' if variable_to_comment[class_variable]['comment'] else ''
                 variable_to_comment[class_variable]['comment'] += sep + token['token'][3:-3].strip()
+                continue
+
+            # Extract multiline comments in single quote
+            if (class_variable is not None
+                    and token['token_type'] == tokenize.STRING
+                    and token['token'][:1] in {'"', "'"}):
+                sep = ' ' if variable_to_comment[class_variable]['comment'] else ''
+                variable_to_comment[class_variable]['comment'] += sep + token['token'][1:-1].strip()
+                continue
 
             # Match class variable
             class_variable = None

tests/test_utils.py

@@ -281,13 +281,30 @@ class TrickyMultiline:
         class_variables['foo'] = {'comment': comment}
         self.assertEqual(get_class_variables(TrickyMultiline), class_variables)
 
+    def test_triple_quote_multiline(self):
+        class TripleQuoteMultiline:
+            bar: int = 0
+            '''biz baz'''
+
+            hi: str
+            """Hello there"""
+
+        class_variables = OrderedDict()
+        class_variables['bar'] = {'comment': 'biz baz'}
+        class_variables['hi'] = {'comment': 'Hello there'}
+        self.assertEqual(get_class_variables(TripleQuoteMultiline), class_variables)
+
     def test_single_quote_multiline(self):
         class SingleQuoteMultiline:
             bar: int = 0
-            '''biz baz'''
+            'biz baz'
+
+            hi: str
+            "Hello there"
 
         class_variables = OrderedDict()
         class_variables['bar'] = {'comment': 'biz baz'}
+        class_variables['hi'] = {'comment': 'Hello there'}
         self.assertEqual(get_class_variables(SingleQuoteMultiline), class_variables)
 
     def test_functions_with_docs_multiline(self):