polars-df 0.1.0 → 0.1.1

data/lib/polars/struct_expr.rb

@@ -0,0 +1,27 @@
+module Polars
+  class StructExpr
+    attr_accessor :_rbexpr
+
+    def initialize(expr)
+      self._rbexpr = expr._rbexpr
+    end
+
+    def [](item)
+      if item.is_a?(String)
+        field(item)
+      elsif item.is_a?(Integer)
+        Utils.wrap_expr(_rbexpr.struct_field_by_index(item))
+      else
+        raise ArgumentError, "expected type Integer or String, got #{item.class.name}"
+      end
+    end
+
+    def field(name)
+      Utils.wrap_expr(_rbexpr.struct_field_by_name(name))
+    end
+
+    def rename_fields(names)
+      Utils.wrap_expr(_rbexpr.struct_rename_fields(names))
+    end
+  end
+end

data/lib/polars/utils.rb

@@ -1,5 +1,7 @@
 module Polars
   module Utils
+    DTYPE_TEMPORAL_UNITS = ["ns", "us", "ms"]
+
     def self.wrap_s(s)
       Series._from_rbseries(s)
     end
@@ -43,5 +45,30 @@ module Polars
     def self.format_path(path)
       File.expand_path(path)
     end
+
+    # TODO fix
+    def self.is_polars_dtype(data_type)
+      true
+    end
+
+    # TODO fix
+    def self.rb_type_to_dtype(dtype)
+      dtype.to_s
+    end
+
+    def self.scale_bytes(sz, to:)
+      scaling_factor = {
+          "b" => 1,
+          "k" => 1024,
+          "m" => 1024 ** 2,
+          "g" => 1024 ** 3,
+          "t" => 1024 ** 4,
+      }[to[0]]
+      if scaling_factor > 1
+        sz / scaling_factor.to_f
+      else
+        sz
+      end
+    end
   end
 end

data/lib/polars/version.rb

@@ -1,3 +1,3 @@
 module Polars
-  VERSION = "0.1.0"
+  VERSION = "0.1.1"
 end

data/lib/polars.rb

@@ -2,15 +2,20 @@
 require "polars/polars"
 
 # modules
+require "polars/cat_expr"
 require "polars/data_frame"
+require "polars/date_time_expr"
 require "polars/expr"
 require "polars/functions"
+require "polars/io"
 require "polars/lazy_frame"
 require "polars/lazy_functions"
 require "polars/lazy_group_by"
-require "polars/io"
+require "polars/list_expr"
+require "polars/meta_expr"
 require "polars/series"
 require "polars/string_expr"
+require "polars/struct_expr"
 require "polars/utils"
 require "polars/version"
 require "polars/when"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: polars-df
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.1.1
 platform: ruby
 authors:
 - Andrew Kane
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2022-11-22 00:00:00.000000000 Z
+date: 2022-11-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rb_sys
@@ -32,6 +32,8 @@ extensions:
 extra_rdoc_files: []
 files:
 - CHANGELOG.md
+- Cargo.lock
+- Cargo.toml
 - LICENSE.txt
 - README.md
 - ext/polars/Cargo.toml
@@ -42,21 +44,27 @@ files:
 - ext/polars/src/file.rs
 - ext/polars/src/lazy/dataframe.rs
 - ext/polars/src/lazy/dsl.rs
+- ext/polars/src/lazy/meta.rs
 - ext/polars/src/lazy/mod.rs
 - ext/polars/src/lazy/utils.rs
 - ext/polars/src/lib.rs
 - ext/polars/src/series.rs
 - lib/polars-df.rb
 - lib/polars.rb
+- lib/polars/cat_expr.rb
 - lib/polars/data_frame.rb
+- lib/polars/date_time_expr.rb
 - lib/polars/expr.rb
 - lib/polars/functions.rb
 - lib/polars/io.rb
 - lib/polars/lazy_frame.rb
 - lib/polars/lazy_functions.rb
 - lib/polars/lazy_group_by.rb
+- lib/polars/list_expr.rb
+- lib/polars/meta_expr.rb
 - lib/polars/series.rb
 - lib/polars/string_expr.rb
+- lib/polars/struct_expr.rb
 - lib/polars/utils.rb
 - lib/polars/version.rb
 - lib/polars/when.rb