feat: Support individual packaging with poetry (#682)

Author: BrandonLWhite

.gitignore

@@ -59,7 +59,6 @@ dist/
 downloads/
 eggs/
 .eggs/
-lib/
 lib64/
 parts/
 sdist/

index.js

@@ -13,7 +13,6 @@ const { injectAllRequirements } = require('./lib/inject');
 const { layerRequirements } = require('./lib/layer');
 const { installAllRequirements } = require('./lib/pip');
 const { pipfileToRequirements } = require('./lib/pipenv');
-const { pyprojectTomlToRequirements } = require('./lib/poetry');
 const { cleanup, cleanupCache } = require('./lib/clean');
 BbPromise.promisifyAll(fse);
 
@@ -203,7 +202,6 @@ class ServerlessPythonRequirements {
       }
       return BbPromise.bind(this)
         .then(pipfileToRequirements)
-        .then(pyprojectTomlToRequirements)
         .then(addVendorHelper)
         .then(installAllRequirements)
         .then(packRequirements)

lib/pip.js

@@ -7,7 +7,7 @@ const spawn = require('child-process-ext/spawn');
 const { quote } = require('shell-quote');
 const { buildImage, getBindPath, getDockerUid } = require('./docker');
 const { getStripCommand, getStripMode, deleteFiles } = require('./slim');
-const { isPoetryProject } = require('./poetry');
+const { isPoetryProject, pyprojectTomlToRequirements } = require('./poetry');
 const {
   checkForAndDeleteMaxCacheVersions,
   sha256Path,
@@ -60,16 +60,9 @@ function generateRequirementsFile(
   pluginInstance
 ) {
   const { serverless, servicePath, options, log } = pluginInstance;
-  if (
-    options.usePoetry &&
-    fse.existsSync(path.join(servicePath, 'pyproject.toml')) &&
-    isPoetryProject(servicePath)
-  ) {
-    filterRequirementsFile(
-      path.join(servicePath, '.serverless/requirements.txt'),
-      targetFile,
-      pluginInstance
-    );
+  const modulePath = path.dirname(requirementsPath);
+  if (options.usePoetry && isPoetryProject(modulePath)) {
+    filterRequirementsFile(targetFile, targetFile, pluginInstance);
     if (log) {
       log.info(`Parsed requirements.txt from pyproject.toml in ${targetFile}`);
     } else {
@@ -570,11 +563,7 @@ function copyVendors(vendorFolder, targetFolder, { serverless, log }) {
  * @param {string} fileName
  */
 function requirementsFileExists(servicePath, options, fileName) {
-  if (
-    options.usePoetry &&
-    fse.existsSync(path.join(servicePath, 'pyproject.toml')) &&
-    isPoetryProject(servicePath)
-  ) {
+  if (options.usePoetry && isPoetryProject(path.dirname(fileName))) {
     return true;
   }
 
@@ -609,6 +598,8 @@ async function installRequirementsIfNeeded(
   // Our source requirements, under our service path, and our module path (if specified)
   const fileName = path.join(servicePath, modulePath, options.fileName);
 
+  await pyprojectTomlToRequirements(modulePath, pluginInstance);
+
   // Skip requirements generation, if requirements file doesn't exist
   if (!requirementsFileExists(servicePath, options, fileName)) {
     return false;

lib/poetry.js

@@ -8,24 +8,25 @@ const tomlParse = require('@iarna/toml/parse-string');
 /**
  * poetry install
  */
-async function pyprojectTomlToRequirements() {
-  if (!this.options.usePoetry || !isPoetryProject(this.servicePath)) {
+async function pyprojectTomlToRequirements(modulePath, pluginInstance) {
+  const { serverless, servicePath, options, log, progress } = pluginInstance;
+
+  const moduleProjectPath = path.join(servicePath, modulePath);
+  if (!options.usePoetry || !isPoetryProject(moduleProjectPath)) {
     return;
   }
 
   let generateRequirementsProgress;
-  if (this.progress && this.log) {
-    generateRequirementsProgress = this.progress.get(
+  if (progress && log) {
+    generateRequirementsProgress = progress.get(
       'python-generate-requirements-toml'
     );
     generateRequirementsProgress.update(
       'Generating requirements.txt from "pyproject.toml"'
     );
-    this.log.info('Generating requirements.txt from "pyproject.toml"');
+    log.info('Generating requirements.txt from "pyproject.toml"');
   } else {
-    this.serverless.cli.log(
-      'Generating requirements.txt from pyproject.toml...'
-    );
+    serverless.cli.log('Generating requirements.txt from pyproject.toml...');
   }
 
   try {
@@ -42,15 +43,15 @@ async function pyprojectTomlToRequirements() {
           '--with-credentials',
         ],
         {
-          cwd: this.servicePath,
+          cwd: moduleProjectPath,
         }
       );
     } catch (e) {
       if (
         e.stderrBuffer &&
         e.stderrBuffer.toString().includes('command not found')
       ) {
-        throw new this.serverless.classes.Error(
+        throw new serverless.classes.Error(
           `poetry not found! Install it according to the poetry docs.`,
           'PYTHON_REQUIREMENTS_POETRY_NOT_FOUND'
         );
@@ -59,16 +60,16 @@ async function pyprojectTomlToRequirements() {
     }
 
     const editableFlag = new RegExp(/^-e /gm);
-    const sourceRequirements = path.join(this.servicePath, 'requirements.txt');
+    const sourceRequirements = path.join(moduleProjectPath, 'requirements.txt');
     const requirementsContents = fse.readFileSync(sourceRequirements, {
       encoding: 'utf-8',
     });
 
     if (requirementsContents.match(editableFlag)) {
-      if (this.log) {
-        this.log.info('The generated file contains -e flags, removing them');
+      if (log) {
+        log.info('The generated file contains -e flags, removing them');
       } else {
-        this.serverless.cli.log(
+        serverless.cli.log(
           'The generated file contains -e flags, removing them...'
         );
       }
@@ -78,10 +79,10 @@ async function pyprojectTomlToRequirements() {
       );
     }
 
-    fse.ensureDirSync(path.join(this.servicePath, '.serverless'));
+    fse.ensureDirSync(path.join(servicePath, '.serverless'));
     fse.moveSync(
       sourceRequirements,
-      path.join(this.servicePath, '.serverless', 'requirements.txt'),
+      path.join(servicePath, '.serverless', modulePath, 'requirements.txt'),
       { overwrite: true }
     );
   } finally {

test.js

@@ -1479,6 +1479,25 @@ test(
   { skip: !hasPython(3.6) }
 );
 
+test(
+  'poetry py3.6 can package flask with package individually option',
+  async (t) => {
+    process.chdir('tests/poetry_individually');
+    const path = npm(['pack', '../..']);
+    npm(['i', path]);
+
+    sls(['package'], { env: {} });
+    const zipfiles = await listZipFiles(
+      '.serverless/module1-sls-py-req-test-dev-hello.zip'
+    );
+    t.true(zipfiles.includes(`flask${sep}__init__.py`), 'flask is packaged');
+    t.true(zipfiles.includes(`bottle.py`), 'bottle is packaged');
+    t.true(zipfiles.includes(`boto3${sep}__init__.py`), 'boto3 is packaged');
+    t.end();
+  },
+  { skip: !hasPython(3.6) }
+);
+
 test(
   'py3.6 can package flask with package individually option',
   async (t) => {

tests/base/package.json

@@ -9,6 +9,6 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "serverless-python-requirements": "file:serverless-python-requirements-5.1.1.tgz"
+    "serverless-python-requirements": "file:serverless-python-requirements-5.3.1.tgz"
   }
 }

tests/non_build_pyproject/package.json

@@ -9,6 +9,6 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "serverless-python-requirements": "file:serverless-python-requirements-5.1.1.tgz"
+    "serverless-python-requirements": "file:serverless-python-requirements-5.3.1.tgz"
   }
 }

tests/non_poetry_pyproject/package.json

@@ -9,6 +9,6 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "serverless-python-requirements": "file:serverless-python-requirements-5.1.1.tgz"
+    "serverless-python-requirements": "file:serverless-python-requirements-5.3.1.tgz"
   }
 }

tests/pipenv/package.json

@@ -9,6 +9,6 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "serverless-python-requirements": "file:serverless-python-requirements-5.1.1.tgz"
+    "serverless-python-requirements": "file:serverless-python-requirements-5.3.1.tgz"
   }
 }

tests/poetry/package.json

@@ -9,6 +9,6 @@
   "author": "",
   "license": "ISC",
   "dependencies": {
-    "serverless-python-requirements": "file:serverless-python-requirements-5.1.1.tgz"
+    "serverless-python-requirements": "file:serverless-python-requirements-5.3.1.tgz"
   }
 }