mirror of
https://github.com/RustPython/RustPython.git
synced 2026-06-09 22:49:57 +09:00
106 lines
3.4 KiB
Rust
106 lines
3.4 KiB
Rust
#![recursion_limit = "128"]
|
|
#![doc(html_logo_url = "https://raw.githubusercontent.com/RustPython/RustPython/main/logo.png")]
|
|
#![doc(html_root_url = "https://docs.rs/rustpython-derive/")]
|
|
|
|
extern crate proc_macro;
|
|
|
|
#[macro_use]
|
|
extern crate maplit;
|
|
|
|
#[macro_use]
|
|
mod error;
|
|
#[macro_use]
|
|
mod util;
|
|
|
|
mod compile_bytecode;
|
|
mod doc;
|
|
mod from_args;
|
|
mod pyclass;
|
|
mod pymodule;
|
|
mod pystructseq;
|
|
mod pyvalue;
|
|
|
|
use error::{extract_spans, Diagnostic};
|
|
use proc_macro::TokenStream;
|
|
use proc_macro2::TokenStream as TokenStream2;
|
|
use quote::ToTokens;
|
|
use syn::{parse_macro_input, AttributeArgs, DeriveInput, Item};
|
|
|
|
fn result_to_tokens(result: Result<TokenStream2, Diagnostic>) -> TokenStream {
|
|
result.unwrap_or_else(ToTokens::into_token_stream).into()
|
|
}
|
|
|
|
#[proc_macro_derive(FromArgs, attributes(pyarg))]
|
|
pub fn derive_from_args(input: TokenStream) -> TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
result_to_tokens(from_args::impl_from_args(input))
|
|
}
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn pyclass(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let attr = parse_macro_input!(attr as AttributeArgs);
|
|
let item = parse_macro_input!(item as Item);
|
|
result_to_tokens(pyclass::impl_pyclass(attr, item))
|
|
}
|
|
|
|
/// This macro serves a goal of generating multiple
|
|
/// `BaseException` / `Exception`
|
|
/// subtypes in a uniform and convenient manner.
|
|
/// It looks like `SimpleExtendsException` in `CPython`.
|
|
/// https://github.com/python/cpython/blob/main/Objects/exceptions.c
|
|
///
|
|
/// We need `ctx` to be ready to add
|
|
/// `properties` / `custom` constructors / slots / methods etc.
|
|
/// So, we use `extend_class!` macro as the second
|
|
/// step in exception type definition.
|
|
#[proc_macro]
|
|
pub fn define_exception(input: TokenStream) -> TokenStream {
|
|
let exc_def = parse_macro_input!(input as pyclass::PyExceptionDef);
|
|
result_to_tokens(pyclass::impl_define_exception(exc_def))
|
|
}
|
|
|
|
/// Helper macro to define `Exception` types.
|
|
/// More-or-less is an alias to `pyclass` macro.
|
|
#[proc_macro_attribute]
|
|
pub fn pyexception(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let attr = parse_macro_input!(attr as AttributeArgs);
|
|
let item = parse_macro_input!(item as Item);
|
|
result_to_tokens(pyclass::impl_pyexception(attr, item))
|
|
}
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn pyimpl(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let attr = parse_macro_input!(attr as AttributeArgs);
|
|
let item = parse_macro_input!(item as Item);
|
|
result_to_tokens(pyclass::impl_pyimpl(attr, item))
|
|
}
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn pymodule(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let attr = parse_macro_input!(attr as AttributeArgs);
|
|
let item = parse_macro_input!(item as Item);
|
|
result_to_tokens(pymodule::impl_pymodule(attr, item))
|
|
}
|
|
|
|
#[proc_macro_derive(PyStructSequence)]
|
|
pub fn pystruct_sequence(input: TokenStream) -> TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
result_to_tokens(pystructseq::impl_pystruct_sequence(input))
|
|
}
|
|
|
|
#[proc_macro]
|
|
pub fn py_compile(input: TokenStream) -> TokenStream {
|
|
result_to_tokens(compile_bytecode::impl_py_compile(input.into()))
|
|
}
|
|
|
|
#[proc_macro]
|
|
pub fn py_freeze(input: TokenStream) -> TokenStream {
|
|
result_to_tokens(compile_bytecode::impl_py_freeze(input.into()))
|
|
}
|
|
|
|
#[proc_macro_derive(PyValue)]
|
|
pub fn pyvalue(input: TokenStream) -> TokenStream {
|
|
let input = parse_macro_input!(input as DeriveInput);
|
|
result_to_tokens(pyvalue::impl_pyvalue(input))
|
|
}
|