polars-df 0.13.0-x64-mingw-ucrt

data/lib/polars/dynamic_group_by.rb

@@ -0,0 +1,52 @@
+module Polars
+  # A dynamic grouper.
+  #
+  # This has an `.agg` method which allows you to run all polars expressions in a
+  # group by context.
+  class DynamicGroupBy
+    def initialize(
+      df,
+      index_column,
+      every,
+      period,
+      offset,
+      truncate,
+      include_boundaries,
+      closed,
+      by,
+      start_by
+    )
+      period = Utils.parse_as_duration_string(period)
+      offset = Utils.parse_as_duration_string(offset)
+      every = Utils.parse_as_duration_string(every)
+
+      @df = df
+      @time_column = index_column
+      @every = every
+      @period = period
+      @offset = offset
+      @truncate = truncate
+      @include_boundaries = include_boundaries
+      @closed = closed
+      @by = by
+      @start_by = start_by
+    end
+
+    def agg(*aggs, **named_aggs)
+      @df.lazy
+        .group_by_dynamic(
+          @time_column,
+          every: @every,
+          period: @period,
+          offset: @offset,
+          truncate: @truncate,
+          include_boundaries: @include_boundaries,
+          closed: @closed,
+          by: @by,
+          start_by: @start_by
+        )
+        .agg(*aggs, **named_aggs)
+        .collect(no_optimization: true, string_cache: false)
+    end
+  end
+end

data/lib/polars/exceptions.rb

@@ -0,0 +1,38 @@
+module Polars
+  # @private
+  # Base class for all Polars errors.
+  class Error < StandardError; end
+
+  # @private
+  # Exception raised when an operation is not allowed (or possible) against a given object or data structure.
+  class InvalidOperationError < Error; end
+
+  # @private
+  # Exception raised when an unsupported testing assert is made.
+  class InvalidAssert < Error; end
+
+  # @private
+  # Exception raised when the number of returned rows does not match expectation.
+  class RowsException < Error; end
+
+  # @private
+  # Exception raised when no rows are returned, but at least one row is expected.
+  class NoRowsReturned < RowsException; end
+
+  # @private
+  # Exception raised when more rows than expected are returned.
+  class TooManyRowsReturned < RowsException; end
+
+  # @private
+  class AssertionError < Error; end
+
+  # @private
+  class ComputeError < Error; end
+
+  # @private
+  class Todo < Error
+    def message
+      "not implemented yet"
+    end
+  end
+end