index.js

// https://github.com/semantic-release/semantic-release/blob/master/docs/usage/plugins.md
import { parse } from "@iarna/toml";
import { execa } from "execa";
import { readFileSync } from "fs";
import { join, resolve } from "path";

const channelMap = {
  alpha: "a",
  beta: "b",
  next: "rc",
  rc: "rc",
  // dev is not definded as it is the default fallback value
};

// nextRelease.type: patch
// nextRelease.channel: beta
// nextRelease.version: 1.2.3-beta.1
// nextRelease.gitTag: v1.2.3-beta.1
// nextRelease.name: v1.2.3-beta.1

export async function prepare(pluginConfig, context) {
  // https://github.com/semantic-release/npm/blob/master/lib/prepare.js
  const {
    cwd,
    env,
    stdout,
    stderr,
    nextRelease: { version, channel },
    logger,
  } = context;
  const { pkgRoot } = pluginConfig;

  const basePath = pkgRoot ? resolve(cwd, pkgRoot) : cwd;

  const [mainVersion, versionSuffix] = version.split("-");
  let pepVersion = mainVersion;

  if (versionSuffix !== undefined) {
    const [, buildVersion] = versionSuffix.split(".");

    if (buildVersion !== undefined) {
      const separator = channelMap[channel] ?? "dev";
      pepVersion = `${mainVersion}${separator}${buildVersion}`;
    }
  }

  const versionResult = execa("poetry", ["version", pepVersion], {
    cwd: basePath,
    env,
    preferLocal: true,
  });
  versionResult.stdout.pipe(stdout, { end: false });
  versionResult.stderr.pipe(stderr, { end: false });

  await versionResult;

  logger.log("Creating pypi package version %s", version);
  const buildresult = execa("poetry", ["build"], {
    cwd: basePath,
    env,
    preferLocal: true,
  });
  buildresult.stdout.pipe(stdout, { end: false });
  buildresult.stderr.pipe(stderr, { end: false });

  await buildresult;
}

export async function publish(pluginConfig, context) {
  // https://github.com/semantic-release/npm/blob/master/lib/publish.js
  const { cwd, env, stdout, stderr, logger } = context;
  const { pkgRoot } = pluginConfig;
  const { PYPI_TOKEN } = env;

  if (PYPI_TOKEN !== undefined && PYPI_TOKEN !== "") {
    const basePath = pkgRoot ? resolve(cwd, pkgRoot) : cwd;

    const pyprojectContent = readFileSync(
      join(basePath, "pyproject.toml")
    ).toString();
    const pyproject = parse(pyprojectContent);
    const pypiName = pyproject.tool.poetry.name;
    const pypiVersion = pyproject.tool.poetry.version;

    logger.log(`Publishing version ${pypiVersion} to pypi registry`);

    const result = execa(
      "poetry",
      [
        "publish",
        "--username",
        "__token__",
        "--password",
        PYPI_TOKEN,
        "--no-interaction",
        "-vvv",
      ],
      { cwd: basePath, env, preferLocal: true }
    );
    result.stdout.pipe(stdout, { end: false });
    result.stderr.pipe(stderr, { end: false });
    await result;

    logger.log(`Published ${pypiName}==${pypiVersion} to pypi`);

    return {
      name: `PyPI Package`,
      url: `https://pypi.org/project/${pypiName}/${pypiVersion}/`,
    };
  }

  logger.log(`Skip publishing to npm registry due to empty PYPI_TOKEN`);
  return false;
}