reproto-backend-python 0.3.36

Rethinking Protocol Generators 
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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

use genco::{Cons, Element, IntoTokens, Python, Quoted, Tokens};
use std::fmt;

pub struct BlockComment<'el>(pub &'el [String]);

impl<'el> IntoTokens<'el, Python<'el>> for BlockComment<'el> {
    fn into_tokens(self) -> Tokens<'el, Python<'el>> {
        let c: Tokens<'el, Python<'el>> = self.0
            .iter()
            .map(|c| Element::Literal(c.as_str().into()))
            .collect();

        let mut toks = Tokens::new();

        toks.push("\"\"\"");
        toks.push(c.join(Element::Line));
        toks.push("\"\"\"");

        toks
    }
}

pub struct IfNoneThen<C, D>(pub C, pub D);

impl<'el, C, D> IntoTokens<'el, Python<'el>> for IfNoneThen<C, D>
where
    C: Into<Tokens<'el, Python<'el>>>,
    D: Into<Tokens<'el, Python<'el>>>,
{
    fn into_tokens(self) -> Tokens<'el, Python<'el>> {
        let mut toks = Tokens::new();

        let cond = self.0.into();
        let def = self.1.into();

        toks.push(toks!["if ", cond.clone(), " is None:"]);
        toks.nested(toks![cond, " = ", def]);

        toks
    }
}

#[derive(Clone)]
pub struct IfNoneRaise<C, M>(pub C, pub M);

impl<'el, C, M> IntoTokens<'el, Python<'el>> for IfNoneRaise<C, M>
where
    C: Into<Tokens<'el, Python<'el>>>,
    M: Clone + Into<Cons<'el>>,
{
    fn into_tokens(self) -> Tokens<'el, Python<'el>> {
        let IfNoneRaise(var, m) = self;

        let mut t = Tokens::new();
        push!(t, "if ", var.into(), " is None:");
        nested!(t, "raise ", Exception(m));
        t
    }
}

#[derive(Clone)]
pub struct Exception<M>(pub M);

impl<'el, M> From<Exception<M>> for Element<'el, Python<'el>>
where
    M: Into<Cons<'el>>,
{
    fn from(value: Exception<M>) -> Self {
        toks!["Exception(", value.0.into().quoted(), ")"].into()
    }
}

/// Utilities for handling differences between python versions.
pub trait VersionHelper: fmt::Debug {
    /// Check if the given variable is a string.
    ///
    /// In Python 3, strings are `str` objects.
    /// In Python 2, strings would be `unicode` objects.
    fn is_string<'el>(&self, var: Cons<'el>) -> Tokens<'el, Python<'el>>;
}