From abcc326dc7b3562ce531febcba0ed3c19be6415b Mon Sep 17 00:00:00 2001
From: Mikhail Golubev <mikhail.golubev@jetbrains.com>
Date: Fri, 15 Nov 2019 22:53:51 +0300
Subject: [PATCH] Allow to change multiprocessing start method via environment
 for testing purposes

GitOrigin-RevId: 2e0cb4c240f02a600b37e7511bdb6903a6708d1d
---
 python/helpers/generator3/__main__.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/python/helpers/generator3/__main__.py b/python/helpers/generator3/__main__.py
index d047732f9706..14f95d53a056 100644
--- a/python/helpers/generator3/__main__.py
+++ b/python/helpers/generator3/__main__.py
@@ -31,6 +31,14 @@ def _enable_segfault_tracebacks():
         pass
 
 
+def _configure_multiprocessing():
+    required_start_method = os.environ.get('GENERATOR3_MULTIPROCESSING_START_METHOD')
+    if required_start_method:
+        import multiprocessing
+        # Available only since Python 3.4
+        multiprocessing.set_start_method(required_start_method)
+
+
 def parse_args(gen_version):
     parser = argparse.ArgumentParser(prog='generator3',
                                      description='Generates interface skeletons (binary stubs) for binary and '
@@ -155,4 +163,5 @@ if __name__ == "__main__":
     _bootstrap_sys_path()
     _setup_logging()
     _enable_segfault_tracebacks()
+    _configure_multiprocessing()
     main()