You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
124 lines
4.5 KiB
124 lines
4.5 KiB
use clap::{arg, ArgAction, Command};
|
|
|
|
// Generate command line options for the csv command
|
|
pub fn gen_csv_command() -> Command {
|
|
Command::new("csv")
|
|
.about("Read csv, output arrow stream")
|
|
.arg(arg!([path] "Path to CSV file"))
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
|
|
.arg(
|
|
arg!(-i --stdin ... "Read from stdin")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!(-q --query <String> "Execute query on the file").required(false))
|
|
.arg(
|
|
arg!(-s --summary ... "Summarize the data")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(
|
|
arg!(-t --text ... "Output text instead of binary")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!(-P --parquet <String> "Write output as a parquet file").required(false))
|
|
.arg(
|
|
arg!(-a --head ... "Print the header of the table")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
}
|
|
|
|
// Generate command line options for the schema command
|
|
pub fn gen_schema_command() -> Command {
|
|
Command::new("schema")
|
|
.about("Several table schema related utilities")
|
|
.arg(
|
|
arg!(-i --stdin ... "Read from stdin")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
|
|
.arg(arg!(-n --name <String> "Table name").required(false))
|
|
.arg(arg!(-l --strlen <String> "Default length for string columns").required(false))
|
|
.arg(
|
|
arg!(-s --summary ... "Summarize the schema")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(
|
|
arg!(-p --postgresql ... "Create a postgresql table with schema")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
}
|
|
|
|
// Generate command line options for the sql command
|
|
pub fn gen_sql_command() -> Command {
|
|
Command::new("sql")
|
|
.about("Runs a sql statement on the file")
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
|
|
.arg(arg!([statement] "SQL statement"))
|
|
.arg(
|
|
arg!(-t --text ... "Input text instead of binary")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter").required(false))
|
|
}
|
|
|
|
// Generate command line options for the rpq command
|
|
pub fn gen_rpq_command() -> Command {
|
|
Command::new("rpq")
|
|
.about("Read parquet file")
|
|
.arg(arg!([path] "Path to the parquet file"))
|
|
.arg(arg!(-q --query <String> "Execute query on the file").required(false))
|
|
.arg(
|
|
arg!(-s --summary ... "Summarize the data")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(
|
|
arg!(-i --stdin ... "Read from stdin instead than from a file")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(
|
|
arg!(-t --text ... "Output text instead of binary")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!(-P --parquet <String> "Write the result as a parquet file").required(false))
|
|
.arg(
|
|
arg!(-a --head ... "Print the header of the table")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
}
|
|
|
|
// Generate command line options for the wpq command
|
|
pub fn gen_wpq_command() -> Command {
|
|
Command::new("wpq")
|
|
.about("Write to a paquet file")
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
|
|
.arg(
|
|
arg!(-t --text ... "Input text instead of binary")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
.arg(arg!([path] "Path to the new parquet file"))
|
|
}
|
|
|
|
// Generate command line options for the print command
|
|
pub fn gen_print_command() -> Command {
|
|
Command::new("print")
|
|
.about("Pretty prints the table")
|
|
.arg(arg!(-d --delimiter <String> "Column delimiter. Assume ,").required(false))
|
|
.arg(
|
|
arg!(-t --text ... "Inputs csv instead of binary")
|
|
.required(false)
|
|
.action(ArgAction::SetTrue),
|
|
)
|
|
}
|