summaryrefslogtreecommitdiffstats
path: root/test cases/rust/11 generated main/gen.py
blob: c8cfe76ec00989673ef3004837e074bd6d75fe01 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
#!/usr/bin/env python3

import argparse


def main() -> None:
    parser = argparse.ArgumentParser()
    parser.add_argument('out')
    parser.add_argument('--mode', choices=['main', 'lib'], default='main')
    args = parser.parse_args()

    with open(args.out, 'w') as f:
        if args.mode == 'main':
            f.write('fn main() { println!("I prefer tarnish, actually.") }')
        elif args.mode == 'lib':
            f.write('pub fn libfun() { println!("I prefer tarnish, actually.") }')


if __name__ == "__main__":
    main()