Add attrs export to python client

stone/backends/python_client.py

@@ -105,6 +105,17 @@
     type=str,
     help='The auth type of the client to generate.',
 )
+_cmdline_parser.add_argument(
+    '-a',
+    '--attribute',
+    action='append',
+    type=str,
+    default=[],
+    help=('Route attributes that the backend will have access to and '
+          'presumably expose in generated code. Use ":all" to select all '
+          'attributes defined in stone_cfg.Route. Attributes will be '
+          "exposed in the documentation, as the client doesn't use them."),
+)
 
 
 class PythonClientBackend(CodeBackend):
@@ -259,6 +270,7 @@ def _generate_route_helper(self, namespace, route, download_to_file=False):
                 extra_request_args=extra_request_args,
                 extra_return_arg=extra_return_arg,
                 footer=footer,
+                attrs=route.attrs,
             )
 
             self._maybe_generate_deprecation_warning(route)
@@ -356,7 +368,7 @@ def _maybe_generate_deprecation_warning(self, route):
     def _generate_docstring_for_func(self, namespace, arg_data_type,
                                      result_data_type=None, error_data_type=None,
                                      overview=None, extra_request_args=None,
-                                     extra_return_arg=None, footer=None):
+                                     extra_return_arg=None, footer=None, attrs=None):
         """
         Generates a docstring for a function or method.
 
@@ -380,7 +392,14 @@ def _generate_docstring_for_func(self, namespace, arg_data_type,
         :param str footer: Additional notes at the end of the docstring.
         """
         fields = [] if is_void_type(arg_data_type) else arg_data_type.fields
-        if not fields and not overview:
+
+        attrs_list = []
+        if self.args.attribute and attrs:
+            for attribute in self.args.attribute:
+                if attribute in attrs:
+                    attrs_list.append('{}: {}'.format(attribute, attrs[attribute]))
+
+        if not fields and not overview and not attrs_list:
             # If we don't have an overview or any input parameters, we skip the
             # docstring altogether.
             return
@@ -389,10 +408,16 @@ def _generate_docstring_for_func(self, namespace, arg_data_type,
         if overview:
             self.emit_wrapped_text(overview)
 
+        if attrs_list:
+            if overview:
+                self.emit()
+            self.emit('Route attributes:')
+            [self.emit_wrapped_text(a, '    ') for a in attrs_list]
+
         # Description of all input parameters
         if extra_request_args or fields:
-            if overview:
-                # Add a blank line if we had an overview
+            if overview or attrs_list:
+                # Add a blank line if we had an overview or attrs
                 self.emit()
 
             if extra_request_args:
@@ -446,7 +471,7 @@ def _generate_docstring_for_func(self, namespace, arg_data_type,
                 self.emit(':type arg: {}'.format(
                     self._format_type_in_doc(namespace, arg_data_type)))
 
-        if overview and not (extra_request_args or fields):
+        if (overview or attrs_list) and not (extra_request_args or fields):
             # Only output an empty line if we had an overview and haven't
             # started a section on declaring types.
             self.emit()

test/test_python_client.py

@@ -27,7 +27,7 @@ def _evaluate_namespace(self, ns):
 
         backend = PythonClientBackend(
             target_folder_path='output',
-            args=['-m', 'files', '-c', 'DropboxBase', '-t', 'dropbox'])
+            args=['-a', 'scope', '-a', 'another_attribute', '-m', 'files', '-c', 'DropboxBase', '-t', 'dropbox'])
         backend._generate_routes(ns)
         return backend.output_buffer_to_string()
 
@@ -218,4 +218,33 @@ def test_route_argument_doc_string(self):
         self.assertEqual(backend._format_type_in_doc(ns, Map(String(), Int32())),
                          'Map[str, int]')
 
+    def test_route_arbitrary_attrs(self):
+        # type: () -> None
+
+        route1 = ApiRoute('get_metadata', 1, None)
+        route1.set_attributes(None, None, Void(), Void(), Void(), {'scope':'events.read'})
+        ns = ApiNamespace('files')
+        ns.add_route(route1)
+
+        result = self._evaluate_namespace(ns)
+        expected = textwrap.dedent('''\
+            def files_get_metadata(self):
+                """
+                Route attributes:
+                    scope: events.read
+
+                :rtype: None
+                """
+                arg = None
+                r = self.request(
+                    files.get_metadata,
+                    'files',
+                    arg,
+                    None,
+                )
+                return None
+
+        ''')
+        self.assertEqual(result, expected)
+
     # TODO: add more unit tests for client code generation