extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-05-10 04:54:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

make polars commands propagate metadata

Browse Source
This commit is contained in:
Julian Amarilla 2025-04-25 02:01:35 -03:00
parent bc946f012a
commit b8e6dde4ff
9 changed files with 126 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
crates/nu_plugin_polars/src/dataframe/command/aggregation/agg_groups.rs
View File

@ -60,6 +60,20 @@ impl PluginCommand for ExprAggGroups {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl ExprAggGroups {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let value = input.into_value(call.head)?;
match PolarsPluginObject::try_from_value(plugin, &value)? {

14
crates/nu_plugin_polars/src/dataframe/command/aggregation/aggregate.rs
View File

@ -126,6 +126,20 @@ impl PluginCommand for LazyAggregate {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl LazyAggregate {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let vals: Vec<Value> = call.rest(0)?;
let value = Value::list(vals, call.head);

14
crates/nu_plugin_polars/src/dataframe/command/aggregation/count.rs
View File

@ -55,6 +55,20 @@ impl PluginCommand for ExprCount {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl ExprCount {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let value = input.into_value(call.head)?;
match PolarsPluginObject::try_from_value(plugin, &value)? {

14
crates/nu_plugin_polars/src/dataframe/command/aggregation/implode.rs
View File

@ -52,6 +52,20 @@ impl PluginCommand for ExprImplode {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl ExprImplode {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let value = input.into_value(call.head)?;
match PolarsPluginObject::try_from_value(plugin, &value)? {

14
crates/nu_plugin_polars/src/dataframe/command/core/summary.rs
View File

@ -111,6 +111,20 @@ impl PluginCommand for Summary {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl Summary {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
command(plugin, engine, call, input).map_err(LabeledError::from)
}

14
crates/nu_plugin_polars/src/dataframe/command/data/first.rs
View File

@ -96,6 +96,20 @@ impl PluginCommand for FirstDF {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl FirstDF {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let value = input.into_value(call.head)?;
match PolarsPluginObject::try_from_value(plugin, &value)? {

14
crates/nu_plugin_polars/src/dataframe/command/data/join.rs
View File

@ -240,6 +240,20 @@ impl PluginCommand for LazyJoin {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl LazyJoin {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let left = call.has_flag("left")?;
let full = call.has_flag("full")?;

14
crates/nu_plugin_polars/src/dataframe/command/data/len.rs
View File

@ -55,6 +55,20 @@ impl PluginCommand for ExprLen {
plugin: &Self::Plugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl ExprLen {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
_input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let res: NuExpression = polars::prelude::len().into();

14
crates/nu_plugin_polars/src/dataframe/command/string/str_slice.rs
View File

@ -116,6 +116,20 @@ impl PluginCommand for StrSlice {
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let metadata = input.metadata();
self.run_inner(plugin, engine, call, input)
.map(|pd| pd.set_metadata(metadata))
}
}
impl StrSlice {
fn run_inner(
&self,
plugin: &PolarsPlugin,
engine: &EngineInterface,
call: &EvaluatedCall,
input: PipelineData,
) -> Result<PipelineData, LabeledError> {
let value = input.into_value(call.head)?;
match PolarsPluginObject::try_from_value(plugin, &value)? {