sigil-stitch 0.2.0

Type-safe, import-aware, width-aware code generation for multiple languages
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
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
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102

use sigil_stitch::code_block::CodeBlock;
use sigil_stitch::lang::python::Python;
use sigil_stitch::spec::file_spec::FileSpec;
use sigil_stitch::type_name::TypeName;

mod golden;

// === Function with imports ===

#[test]
fn test_python_function_with_imports() {
    let json_dumps = TypeName::<Python>::importable("json", "dumps");
    let http_server = TypeName::<Python>::importable("http.server", "HTTPServer");

    let mut b = CodeBlock::<Python>::builder();
    b.add("def start_server():", ());
    b.add_line();
    b.add("%>", ());
    b.add_statement("srv = %T()", (http_server,));
    b.add_statement("data = %T(srv)", (json_dumps,));
    b.add("%<", ());
    let block = b.build().unwrap();

    let mut fb = FileSpec::builder_with("server.py", Python::new());
    fb.add_code(block);
    let file = fb.build().unwrap();

    let output = file.render(80).unwrap();
    golden::assert_golden("python/function_with_imports.py", &output);
}

// === Import grouping: stdlib vs third-party ===

#[test]
fn test_python_import_grouping() {
    let json_dumps = TypeName::<Python>::importable("json", "dumps");
    let json_loads = TypeName::<Python>::importable("json", "loads");
    let flask = TypeName::<Python>::importable("flask", "Flask");
    let os_path = TypeName::<Python>::importable("os.path", "join");

    let mut b = CodeBlock::<Python>::builder();
    b.add_statement("_ = %T()", (json_dumps,));
    b.add_statement("_ = %T()", (json_loads,));
    b.add_statement("_ = %T()", (flask,));
    b.add_statement("_ = %T()", (os_path,));
    let block = b.build().unwrap();

    let mut fb = FileSpec::builder_with("app.py", Python::new());
    fb.add_code(block);
    let file = fb.build().unwrap();

    let output = file.render(80).unwrap();
    golden::assert_golden("python/import_grouping.py", &output);
}

// === Control flow (if/elif/else with colon) ===

#[test]
fn test_python_control_flow() {
    let mut b = CodeBlock::<Python>::builder();
    b.add("def classify(x: int) -> str:", ());
    b.add_line();
    b.add("%>", ());
    b.begin_control_flow("if x > 0", ());
    b.add_statement("return 'positive'", ());
    b.next_control_flow("elif x < 0", ());
    b.add_statement("return 'negative'", ());
    b.next_control_flow("else", ());
    b.add_statement("return 'zero'", ());
    b.end_control_flow();
    b.add("%<", ());
    let block = b.build().unwrap();

    let mut fb = FileSpec::builder_with("classify.py", Python::new());
    fb.add_code(block);
    let file = fb.build().unwrap();

    let output = file.render(80).unwrap();
    golden::assert_golden("python/control_flow.py", &output);
}

// === Simple class with fields ===

#[test]
fn test_python_class_basic() {
    let mut b = CodeBlock::<Python>::builder();
    b.add("class Config:", ());
    b.add_line();
    b.add("%>", ());
    b.add_statement("name: str", ());
    b.add_statement("port: int", ());
    b.add_statement("debug: bool = False", ());
    b.add("%<", ());
    let block = b.build().unwrap();

    let mut fb = FileSpec::builder_with("config.py", Python::new());
    fb.add_code(block);
    let file = fb.build().unwrap();

    let output = file.render(80).unwrap();
    golden::assert_golden("python/class_basic.py", &output);
}