mirror of
https://github.com/bitcoinresearchkit/brk.git
synced 2026-06-08 22:21:56 -07:00
nym21
60 lines
2.2 KiB
Rust
60 lines
2.2 KiB
Rust
//! Python client generation.
|
|
//!
|
|
//! This module generates a Python client with type hints for the BRK API.
|
|
|
|
pub mod api;
|
|
pub mod client;
|
|
pub mod tree;
|
|
pub mod types;
|
|
|
|
use std::{fmt::Write, io, path::Path};
|
|
|
|
use super::write_if_changed;
|
|
use crate::{ClientMetadata, Endpoint, TypeSchemas};
|
|
|
|
/// Generate Python client from metadata and OpenAPI endpoints.
|
|
///
|
|
/// `output_path` is the full path to the output file (e.g., "packages/brk_client/__init__.py").
|
|
pub fn generate_python_client(
|
|
metadata: &ClientMetadata,
|
|
endpoints: &[Endpoint],
|
|
schemas: &TypeSchemas,
|
|
output_path: &Path,
|
|
) -> io::Result<()> {
|
|
let mut output = String::new();
|
|
|
|
writeln!(output, "# Auto-generated BRK Python client").unwrap();
|
|
writeln!(output, "# Do not edit manually\n").unwrap();
|
|
writeln!(output, "from __future__ import annotations").unwrap();
|
|
writeln!(output, "from dataclasses import dataclass").unwrap();
|
|
writeln!(
|
|
output,
|
|
"from typing import TypeVar, Generic, Any, Dict, Optional, List, Iterator, Literal, TypedDict, Union, Protocol, overload, Tuple, TYPE_CHECKING"
|
|
)
|
|
.unwrap();
|
|
writeln!(
|
|
output,
|
|
"from http.client import HTTPSConnection, HTTPConnection"
|
|
)
|
|
.unwrap();
|
|
writeln!(output, "from urllib.parse import urlparse").unwrap();
|
|
writeln!(output, "from datetime import date, datetime, timedelta, timezone").unwrap();
|
|
writeln!(output, "import json\n").unwrap();
|
|
writeln!(output, "if TYPE_CHECKING:").unwrap();
|
|
writeln!(output, " import pandas as pd # type: ignore[import-not-found]").unwrap();
|
|
writeln!(output, " import polars as pl # type: ignore[import-not-found]\n").unwrap();
|
|
writeln!(output, "T = TypeVar('T')\n").unwrap();
|
|
|
|
types::generate_type_definitions(&mut output, schemas);
|
|
client::generate_base_client(&mut output);
|
|
client::generate_endpoint_class(&mut output);
|
|
client::generate_index_accessors(&mut output, &metadata.index_set_patterns);
|
|
client::generate_structural_patterns(&mut output, &metadata.structural_patterns, metadata);
|
|
tree::generate_tree_classes(&mut output, &metadata.catalog, metadata);
|
|
api::generate_main_client(&mut output, endpoints);
|
|
|
|
write_if_changed(output_path, &output)?;
|
|
|
|
Ok(())
|
|
}
|