mitoge/parent: Refactor _get_source_first_stage

Add comments so it's clearer what is meant with.

-mitogen.parent        99212  96.9KiB  51385 50.2KiB 51.8%  12999 12.7KiB 13.1%
+mitogen.parent        99532  97.2KiB  51483 50.3KiB 51.7%  13033 12.7KiB 13.1%

Signed-off-by: Marc Hartmayer <[email protected]>

Author: mhartmay

mitogen/parent.py

@@ -1478,10 +1478,16 @@ def get_python_argv(self):
             return self.options.python_path
         return [self.options.python_path]
 
+    def _get_source_first_stage(self):
+        # Remove the staticmethod decorator and the function name. If we ever
+        # add a docstring to `_first_stage` we need to change this.
+        source = inspect.getsourcelines(self._first_stage)[0][2:]
+        # Remove line comments, leading indentation, trailing newline
+        source = textwrap.dedent(''.join(s for s in source if '#' not in s))[:-1]
+        return source.replace('    ', ' ')
+
     def get_boot_command(self):
-        source = inspect.getsource(self._first_stage)
-        source = textwrap.dedent('\n'.join(source.strip().split('\n')[2:]))
-        source = source.replace('    ', ' ')
+        source = self._get_source_first_stage()
         compressor = zlib.compressobj(
             zlib.Z_BEST_COMPRESSION, zlib.DEFLATED, -zlib.MAX_WBITS,
         )