Refactor access to the generate-setup-file configuration

Author: sdispater

poetry/core/masonry/builders/sdist.py

@@ -89,7 +89,7 @@ def build(self, target_dir=None):  # type: (Path) -> Path
                 else:
                     tar.addfile(tar_info)  # Symlinks & ?
 
-            if self._poetry.package.build_config.get("generate-setup-file", True):
+            if self._poetry.package.build_should_generate_setup():
                 setup = self.build_setup()
                 tar_info = tarfile.TarInfo(pjoin(tar_dir, "setup.py"))
                 tar_info.size = len(setup)

poetry/core/masonry/builders/wheel.py

@@ -77,7 +77,7 @@ def build(self):
         with zipfile.ZipFile(
             os.fdopen(fd, "w+b"), mode="w", compression=zipfile.ZIP_DEFLATED
         ) as zip_file:
-            if not self._poetry.package.build_config.get("generate-setup-file", True):
+            if not self._poetry.package.build_should_generate_setup():
                 self._build(zip_file)
                 self._copy_module(zip_file)
             else:
@@ -96,7 +96,7 @@ def build(self):
 
     def _build(self, wheel):
         if self._package.build_script:
-            if not self._poetry.package.build_config.get("generate-setup-file", True):
+            if not self._poetry.package.build_should_generate_setup():
                 # Since we have a build script but no setup.py generation is required,
                 # we assume that the build script will build and copy the files
                 # directly.

poetry/core/packages/project_package.py

@@ -58,3 +58,6 @@ def urls(self):
         urls.update(self.custom_urls)
 
         return urls
+
+    def build_should_generate_setup(self):  # type: () -> bool
+        return self.build_config.get("generate-setup-file", True)