繁体   English   中英

StructOpt - 如何为 Vec 提供默认值?

[英]StructOpt - how to provide a default value for a Vec?

我正在寻找一种默认情况下用多个项目初始化structopt Vec字段的方法。 我可以为单个项目做到这一点:

use structopt::StructOpt;

#[derive(Debug, StructOpt)]
pub struct Cli {
    #[structopt(default_value = "foo")]
    foo: Vec<String>,
}

fn main() {
    let cli = Cli::from_iter(Vec::<String>::new());
    assert_eq!(cli.foo, vec!["foo"]);
}

但是如何使cli.foo相等,让我们说vec!["foo", "bar"]默认情况下?

我不认为你可以这样做:虽然 StructOpt 有一些关于默认值的技巧,但我希望这仍然以默认值被注入到 CLI 解析中结束,就好像它已经被明确提供一样,这意味着可能没有办法提供多个默认值(尽管我肯定是错的)。

您可能希望在应用程序级别处理这个问题,例如在解析 CLI 之后,检查foo并在它为空时更新它。

我遵循了 L. Riemer 的建议,似乎仅实现FromStr就足够了:

use structopt::StructOpt;

#[derive(Debug, PartialEq)]
struct Foo(Vec<String>);

impl std::str::FromStr for Foo {
    type Err = Box<dyn std::error::Error>;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        Ok(Foo(s.split(",").map(|x| x.trim().to_owned()).collect()))
    }
}

#[derive(StructOpt)]
pub struct Cli {
    #[structopt(long, default_value = "foo, bar")]
    foo: Foo,
}

fn main() {
    let cli = Cli::from_iter(Vec::<String>::new());
    assert_eq!(cli.foo, Foo(vec!["foo".into(), "bar".into()]));

    let cli = Cli::from_iter(vec!["", "--foo", "foo"]);
    assert_eq!(cli.foo, Foo(vec!["foo".into()]));

    let cli = Cli::from_iter(vec!["", "--foo", "foo,bar,baz"]);
    assert_eq!(cli.foo, Foo(vec!["foo".into(), "bar".into(), "baz".into()]));
}

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM