openapi_nexus/generators/python/requests/
codegen.rs

1//! Python requests code generator: IR → idiomatic Python + requests (synchronous).
2
3use std::error::Error;
4use std::sync::OnceLock;
5
6use heck::ToKebabCase as _;
7
8use super::config::PythonRequestsConfig;
9use super::runtime::runtime_files;
10use super::{emit_api, emit_models, project_files};
11use crate::codegen::traits::code_generator::CodeGenerator;
12use crate::codegen::traits::file_writer::{FileInfo, FileWriter};
13use crate::codegen::{GeneratorType, Language};
14use crate::ir::types::{IrInfo, IrSpec};
15
16/// Python requests code generator (synchronous).
17#[derive(Debug)]
18pub struct PythonRequestsCodeGenerator {
19    config: PythonRequestsConfig,
20    resolved_package_name: OnceLock<String>,
21}
22
23impl Clone for PythonRequestsCodeGenerator {
24    fn clone(&self) -> Self {
25        Self {
26            config: self.config.clone(),
27            resolved_package_name: OnceLock::new(),
28        }
29    }
30}
31
32impl PythonRequestsCodeGenerator {
33    pub fn new(config: toml::value::Table) -> Self {
34        Self {
35            config: PythonRequestsConfig::from(config),
36            resolved_package_name: OnceLock::new(),
37        }
38    }
39
40    fn package_name(&self, info: &IrInfo) -> String {
41        self.config
42            .package_name
43            .clone()
44            .unwrap_or_else(|| info.title.to_kebab_case().replace('-', "_"))
45    }
46
47    fn generate_ir(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
48        let package_name = self.package_name(&ir.info);
49        let _ = self.resolved_package_name.set(package_name.clone());
50        let header = render_file_header(&ir.info);
51
52        let mut files = Vec::new();
53
54        files.extend(
55            emit_models::generate_model_files(ir, &header).map_err(|msg| {
56                Box::<dyn Error + Send + Sync>::from(format!("emit_models: {msg}"))
57            })?,
58        );
59
60        files.extend(
61            emit_api::generate_api_files(ir, &header)
62                .map_err(|msg| Box::<dyn Error + Send + Sync>::from(format!("emit_api: {msg}")))?,
63        );
64
65        files.extend(runtime_files(&header));
66
67        files.extend(project_files::generate_project_files(
68            &ir.info,
69            &package_name,
70            &header,
71            &ir.schemas,
72            &ir.operations,
73        ));
74
75        Ok(files)
76    }
77}
78
79impl CodeGenerator for PythonRequestsCodeGenerator {
80    fn language(&self) -> Language {
81        Language::Python
82    }
83
84    fn generator_type(&self) -> GeneratorType {
85        GeneratorType::PythonRequests
86    }
87
88    fn generate(&self, ir: &IrSpec) -> Result<Vec<FileInfo>, Box<dyn Error + Send + Sync>> {
89        self.generate_ir(ir)
90    }
91}
92
93impl FileWriter for PythonRequestsCodeGenerator {
94    fn source_dir(&self) -> Option<&str> {
95        self.resolved_package_name.get().map(|s| s.as_str())
96    }
97}
98
99fn render_file_header(info: &IrInfo) -> String {
100    let mut out = String::new();
101    out.push_str("# Code generated by openapi-nexus. DO NOT EDIT.\n");
102    out.push_str("#\n");
103    out.push_str(&format!("# {} — {}\n", info.title, info.version));
104    if let Some(desc) = &info.description {
105        for line in desc.lines() {
106            out.push_str(&format!("# {line}\n"));
107        }
108    }
109    out.push('\n');
110    out
111}
openapi_nexus/generators/python/requests/codegen.rs

openapi_nexus/generators/python/requests/
codegen.rs
