polars-df 0.9.0 → 0.10.0

data/lib/polars/whenthen.rb

@@ -0,0 +1,83 @@
+module Polars
+  # @private
+  class When
+    attr_accessor :_when
+
+    def initialize(rbwhen)
+      self._when = rbwhen
+    end
+
+    def then(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Then.new(_when.then(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class Then < Expr
+    attr_accessor :_then
+
+    def initialize(rbthen)
+      self._then = rbthen
+    end
+
+    def self._from_rbexpr(rbexpr)
+      Utils.wrap_expr(rbexpr)
+    end
+
+    def _rbexpr
+      _then.otherwise(Polars.lit(nil)._rbexpr)
+    end
+
+    def when(*predicates, **constraints)
+      condition_rbexpr = Utils.parse_when_inputs(*predicates, **constraints)
+      ChainedWhen.new(_then.when(condition_rbexpr))
+    end
+
+    def otherwise(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Utils.wrap_expr(_then.otherwise(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class ChainedWhen
+    attr_accessor :_chained_when
+
+    def initialize(chained_when)
+      self._chained_when = chained_when
+    end
+
+    def then(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      ChainedThen.new(_chained_when.then(statement_rbexpr))
+    end
+  end
+
+  # @private
+  class ChainedThen < Expr
+    attr_accessor :_chained_then
+
+    def initialize(chained_then)
+      self._chained_then = chained_then
+    end
+
+    def self._from_rbexpr(rbexpr)
+      Utils.wrap_expr(rbexpr)
+    end
+
+    def _rbexpr
+      _chained_then.otherwise(Polars.lit(nil)._rbexpr)
+    end
+
+    def when(*predicates, **constraints)
+      condition_rbexpr = Utils.parse_when_inputs(*predicates, **constraints)
+      ChainedWhen.new(_chained_then.when(condition_rbexpr))
+    end
+
+    def otherwise(statement)
+      statement_rbexpr = Utils.parse_as_expression(statement)
+      Utils.wrap_expr(_chained_then.otherwise(statement_rbexpr))
+    end
+  end
+end

data/lib/polars.rb

@@ -63,8 +63,7 @@ require_relative "polars/struct_name_space"
 require_relative "polars/testing"
 require_relative "polars/utils"
 require_relative "polars/version"
-require_relative "polars/when"
-require_relative "polars/when_then"
+require_relative "polars/whenthen"
 
 module Polars
   extend Convert

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.9.0
+  version: 0.10.0
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-03-03 00:00:00.000000000 Z
+date: 2024-05-03 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bigdecimal
@@ -157,8 +157,7 @@ files:
 - lib/polars/testing.rb
 - lib/polars/utils.rb
 - lib/polars/version.rb
-- lib/polars/when.rb
-- lib/polars/when_then.rb
+- lib/polars/whenthen.rb
 homepage: https://github.com/ankane/polars-ruby
 licenses:
 - MIT
@@ -178,7 +177,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.5.3
+rubygems_version: 3.5.9
 signing_key:
 specification_version: 4
 summary: Blazingly fast DataFrames for Ruby

data/lib/polars/when.rb

@@ -1,16 +0,0 @@
-module Polars
-  # @private
-  class When
-    attr_accessor :_rbwhen
-
-    def initialize(rbwhen)
-      self._rbwhen = rbwhen
-    end
-
-    def then(expr)
-      expr = Utils.expr_to_lit_or_expr(expr)
-      rbwhenthen = _rbwhen._then(expr._rbexpr)
-      WhenThen.new(rbwhenthen)
-    end
-  end
-end

data/lib/polars/when_then.rb

@@ -1,19 +0,0 @@
-module Polars
-  # @private
-  class WhenThen
-    attr_accessor :_rbwhenthen
-
-    def initialize(rbwhenthen)
-      self._rbwhenthen = rbwhenthen
-    end
-
-    def when(predicate)
-      WhenThenThen.new(_rbwhenthen.when(predicate._rbexpr))
-    end
-
-    def otherwise(expr)
-      expr = Utils.expr_to_lit_or_expr(expr)
-      Utils.wrap_expr(_rbwhenthen.otherwise(expr._rbexpr))
-    end
-  end
-end