feat(python): read version-file from manifest

googleapis · Nov 16, 2024 · b96be9b · b96be9b
1 parent 3e80797
commit b96be9b
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/strategies/base.ts b/src/strategies/base.ts
@@ -113,6 +113,7 @@ export abstract class BaseStrategy implements Strategy {
   readonly componentNoSpace?: boolean;
   readonly extraFiles: ExtraFile[];
   readonly extraLabels: string[];
+  readonly versionFile?: string;
 
   readonly changelogNotes: ChangelogNotes;
 
@@ -148,6 +149,7 @@ export abstract class BaseStrategy implements Strategy {
     this.extraFiles = options.extraFiles || [];
     this.initialVersion = options.initialVersion;
     this.extraLabels = options.extraLabels || [];
+    this.versionFile = options.versionFile;
   }
 
   /**

diff --git a/src/strategies/python.ts b/src/strategies/python.ts
@@ -105,7 +105,15 @@ export class Python extends BaseStrategy {
     if (!projectName) {
       this.logger.warn('No project/component found.');
     } else {
-      [projectName, projectName.replace(/-/g, '_')]
+      if (this.versionFile) {
+        const versionFilePath = this.addPath(this.versionFile);
+        updates.push({
+          path: versionFilePath,
+          createIfMissing: true,
+          updater: new PythonFileWithVersion({version}),
+        });
+      } else {
+        [projectName, projectName.replace(/-/g, '_')]
         .flatMap(packageName => [
           `${packageName}/__init__.py`,
           `src/${packageName}/__init__.py`,
@@ -117,6 +125,7 @@ export class Python extends BaseStrategy {
             updater: new PythonFileWithVersion({version}),
           })
         );
+      }
     }
 
     // There should be only one version.py, but foreach in case that is incorrect