pyo3cls 0.1.0

Proc macros for PyO3 package
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

// Copyright (c) 2017-present PyO3 Project and Contributors
use syn;
use quote::{Tokens, ToTokens};


pub fn print_err(msg: String, t: Tokens) {
    println!("Error: {} in '{}'", msg, t.to_string());
}

pub fn for_err_msg(i: &ToTokens) -> String {
    let mut tokens = Tokens::new();

    i.to_tokens(&mut tokens);
    tokens.as_str().to_string()
}

pub fn get_doc(attrs: &Vec<syn::Attribute>, null_terminated: bool) -> syn::Lit {
    let mut doc = Vec::new();

    for attr in attrs.iter() {
        match attr.value {
            syn::MetaItem::NameValue(ref ident, ref lit) => {
                if ident.as_ref() == "doc" {
                    let s = quote!{ #lit }.to_string();
                    let mut s = s[1..s.len()-1].to_string();
                    if s.starts_with("/// ") {
                        // Remove leading whitespace and ///
                        s = s[4..].to_string();
                    } else {
                        // Remove only ///
                        s = s[3..].to_string();
                    }
                    doc.push(s)
                }
            }
            _ => (),
        }
    }
    let doc = doc.join("\n");
    if null_terminated {
        syn::Lit::Str(format!("{}\0", doc), syn::StrStyle::Cooked)
    } else {
        syn::Lit::Str(doc, syn::StrStyle::Cooked)
    }
}