polars-df 0.10.0-x86_64-linux-musl

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._timedelta_to_pl_duration(period)
+      offset = Utils._timedelta_to_pl_duration(offset)
+      every = Utils._timedelta_to_pl_duration(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)
+      @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)
+        .collect(no_optimization: true, string_cache: false)
+    end
+  end
+end

data/lib/polars/exceptions.rb

@@ -0,0 +1,31 @@
+module Polars
+  # @private
+  # Base class for all Polars errors.
+  class Error < StandardError; 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 Todo < Error
+    def message
+      "not implemented yet"
+    end
+  end
+end