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

openapi_nexus/generators/python/requests/
codegen.rs
