popeye-cli 1.0.1 → 1.2.0

package/src/generators/python.ts

@@ -31,6 +31,22 @@ export interface GenerationResult {
   error?: string;
 }
 
+/**
+ * Python generator options for workspace/monorepo support
+ */
+export interface PythonGeneratorOptions {
+  /** Base directory for project (defaults to outputDir/projectName) */
+  baseDir?: string;
+  /** Override auto-derived package name */
+  packageName?: string;
+  /** Adjust paths for monorepo structure */
+  workspaceMode?: boolean;
+  /** Skip Docker files (fullstack uses root docker-compose) */
+  skipDocker?: boolean;
+  /** Skip README (fullstack has root README) */
+  skipReadme?: boolean;
+}
+
 /**
  * Create a directory if it doesn't exist
  */
@@ -57,15 +73,27 @@ function toPythonPackageName(name: string): string {
  *
  * @param spec - Project specification
  * @param outputDir - Output directory
+ * @param options - Generator options for workspace/monorepo support
  * @returns Generation result
  */
 export async function generatePythonProject(
   spec: ProjectSpec,
-  outputDir: string
+  outputDir: string,
+  options: PythonGeneratorOptions = {}
 ): Promise<GenerationResult> {
+  const {
+    baseDir,
+    packageName: customPackageName,
+    workspaceMode = false,
+    skipDocker = false,
+    skipReadme = false,
+  } = options;
+
   const projectName = spec.name || 'my-project';
-  const packageName = toPythonPackageName(projectName);
-  const projectDir = path.join(outputDir, projectName);
+  const packageName = customPackageName || toPythonPackageName(projectName);
+
+  // In workspace mode with baseDir, use it directly; otherwise create subdirectory
+  const projectDir = baseDir || path.join(outputDir, projectName);
   const filesCreated: string[] = [];
 
   try {
@@ -73,7 +101,12 @@ export async function generatePythonProject(
     await ensureDir(projectDir);
     await ensureDir(path.join(projectDir, 'src', packageName));
     await ensureDir(path.join(projectDir, 'tests'));
-    await ensureDir(path.join(projectDir, 'data'));
+
+    // Only create data/docs dirs in standalone mode
+    if (!workspaceMode) {
+      await ensureDir(path.join(projectDir, 'data'));
+      await ensureDir(path.join(projectDir, 'docs'));
+    }
 
     // Generate and write files
     const files: Array<{ path: string; content: string }> = [
@@ -94,22 +127,10 @@ export async function generatePythonProject(
         path: path.join(projectDir, '.env.example'),
         content: generateEnvExample(),
       },
-      {
-        path: path.join(projectDir, 'README.md'),
-        content: generateReadme(projectName, spec.idea),
-      },
       {
         path: path.join(projectDir, 'Makefile'),
         content: generateMakefile(projectName),
       },
-      {
-        path: path.join(projectDir, 'Dockerfile'),
-        content: generateDockerfile(projectName),
-      },
-      {
-        path: path.join(projectDir, 'docker-compose.yml'),
-        content: generateDockerCompose(projectName),
-      },
 
       // Source files
       {
@@ -138,13 +159,37 @@ export async function generatePythonProject(
         path: path.join(projectDir, 'tests', 'test_main.py'),
         content: generateTestMain(projectName),
       },
+    ];
 
-      // Data placeholder
-      {
+    // Add README if not skipped
+    if (!skipReadme) {
+      files.push({
+        path: path.join(projectDir, 'README.md'),
+        content: generateReadme(projectName, spec.idea),
+      });
+    }
+
+    // Add Docker files if not skipped
+    if (!skipDocker) {
+      files.push(
+        {
+          path: path.join(projectDir, 'Dockerfile'),
+          content: generateDockerfile(projectName),
+        },
+        {
+          path: path.join(projectDir, 'docker-compose.yml'),
+          content: generateDockerCompose(projectName),
+        }
+      );
+    }
+
+    // Add data placeholder in standalone mode
+    if (!workspaceMode) {
+      files.push({
         path: path.join(projectDir, 'data', '.gitkeep'),
         content: '',
-      },
-    ];
+      });
+    }
 
     // Write all files
     for (const file of files) {