arql 0.1.32 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 48ce3b7b50ccdb8ab338b73e1895b73cd7ebec18ef779c18931603353e709d12
-  data.tar.gz: 3a9bf2b11cdf4be30ab58590aedb81ea2944f6341ca26fd2ab584c9a490cb512
+  metadata.gz: bf94862baf6c8e5ff36589e84a022e892b4b88664ecc300c51d313a541e982fd
+  data.tar.gz: baa0a4b30e5efeb4083347c18f2863faa764052505e46442e45dd9af519612b0
 SHA512:
-  metadata.gz: 1cb190b2e90d18b08ccc98878b5db934ebb1e008fc4aba6a850fd4e384e917d02f14837700a06aca8bc5bdc7799a8ea6ffd5c737bb38260304f60166db3a63ce
-  data.tar.gz: 3b9c6e813b91904cf39bd044c1ac86925802f52d6f0a4e7a7d3f10f2309db812e01cecb0476555bbcb751df73f640c941fe3e307a96c05f9101d854659e53a70
+  metadata.gz: aa3a4aba5288c64c7969efe368713e9f2a64ff9dfde3f41d47b9d8f3ec2f64fbfbe5479d6dc1c95281d82f68497616f44f78e84b635e312b6c6e40528871e6cf
+  data.tar.gz: cae343a1411c59e683dcc99ee309d1ca91100c2c8cf22cbff37ffe04cc02e338d38562e0623a752c5f0803926fba1b0f5a2332074ec3f98083dadd0b8439ab0e

data/Gemfile.lock

@@ -1,15 +1,17 @@
 PATH
   remote: .
   specs:
-    arql (0.1.32)
+    arql (0.2.3)
       activerecord (~> 6.0.3)
       activesupport (~> 6.0.3)
+      caxlsx (~> 3.0.2)
       mysql2 (~> 0.5.3)
       net-ssh-gateway (~> 2.0.0)
       pry (~> 0.13.1)
       pry-byebug (~> 3.9.0)
       pry-doc (~> 1.1.0)
       rainbow (~> 3.0.0)
+      roo (~> 2.8.3)
       sqlite3 (~> 1.4)
       table_print (~> 1.5.6)
       terminal-table (~> 1.8.0)
@@ -17,28 +19,38 @@ PATH
 GEM
   remote: https://rubygems.org/
   specs:
-    activemodel (6.0.3.1)
-      activesupport (= 6.0.3.1)
-    activerecord (6.0.3.1)
-      activemodel (= 6.0.3.1)
-      activesupport (= 6.0.3.1)
-    activesupport (6.0.3.1)
+    activemodel (6.0.3.3)
+      activesupport (= 6.0.3.3)
+    activerecord (6.0.3.3)
+      activemodel (= 6.0.3.3)
+      activesupport (= 6.0.3.3)
+    activesupport (6.0.3.3)
       concurrent-ruby (~> 1.0, >= 1.0.2)
       i18n (>= 0.7, < 2)
       minitest (~> 5.1)
       tzinfo (~> 1.1)
       zeitwerk (~> 2.2, >= 2.2.2)
     byebug (11.1.3)
+    caxlsx (3.0.2)
+      htmlentities (~> 4.3, >= 4.3.4)
+      mimemagic (~> 0.3)
+      nokogiri (~> 1.10, >= 1.10.4)
+      rubyzip (>= 1.3.0, < 3)
     coderay (1.1.3)
     concurrent-ruby (1.1.7)
+    htmlentities (4.3.4)
     i18n (1.8.5)
       concurrent-ruby (~> 1.0)
     method_source (1.0.0)
-    minitest (5.14.1)
+    mimemagic (0.3.5)
+    mini_portile2 (2.4.0)
+    minitest (5.14.2)
     mysql2 (0.5.3)
     net-ssh (6.1.0)
     net-ssh-gateway (2.0.0)
       net-ssh (>= 4.0.0)
+    nokogiri (1.10.10)
+      mini_portile2 (~> 2.4.0)
     pry (0.13.1)
       coderay (~> 1.1)
       method_source (~> 1.0)
@@ -50,8 +62,12 @@ GEM
       yard (~> 0.9.11)
     rainbow (3.0.0)
     rake (12.3.3)
+    roo (2.8.3)
+      nokogiri (~> 1)
+      rubyzip (>= 1.3.0, < 3.0.0)
+    rubyzip (2.3.0)
     sqlite3 (1.4.2)
-    table_print (1.5.6)
+    table_print (1.5.7)
     terminal-table (1.8.0)
       unicode-display_width (~> 1.1, >= 1.1.1)
     thread_safe (0.3.6)

data/arql.gemspec

@@ -37,4 +37,6 @@ Gem::Specification.new do |spec|
   spec.add_dependency 'rainbow', '~> 3.0.0'
   spec.add_dependency 'terminal-table', '~> 1.8.0'
   spec.add_dependency 'table_print', '~> 1.5.6'
+  spec.add_dependency 'roo', '~> 2.8.3'
+  spec.add_dependency 'caxlsx', '~> 3.0.2'
 end

data/lib/arql.rb

@@ -1,4 +1,7 @@
 require 'table_print'
+require 'roo'
+require 'caxlsx'
+require 'csv'
 require "arql/version"
 require 'arql/id'
 require 'arql/multi_io'

data/lib/arql/app.rb

@@ -4,7 +4,7 @@ module Arql
   class App
 
     class << self
-      attr_accessor :log_io, :env, :prompt
+      attr_accessor :log_io, :env, :prompt, :instance
 
       def config
         @@effective_config
@@ -29,6 +29,7 @@ module Arql
       Connection.open(connect_options)
       @definition = Definition.new(effective_config)
       load_initializer!
+      App.instance = self
     end
 
     def connect_options

data/lib/arql/definition.rb

@@ -26,6 +26,14 @@ module Arql
       self.class.to_upsert_sql([self])
     end
 
+    def write_csv(filename, *fields, **options)
+      [self].write_csv(filename, *fields, **options)
+    end
+
+    def write_excel(filename, *fields, **options)
+      [self].write_excel(filename, *fields, **options)
+    end
+
     included do
     end
 
@@ -116,6 +124,8 @@ module Arql
             end
           end
         end
+
+        App.instance&.load_initializer!
       end
     end
 
@@ -168,13 +178,26 @@ module Arql
     end
 
     ::ActiveRecord::Relation.class_eval do
-      def t
-        records.t
+      def t(*attrs, **options)
+        records.t(*attrs, **options)
       end
 
       def v
         records.v
       end
+
+      def a
+        to_a
+      end
+
+      def write_csv(filename, *fields, **options)
+        records.write_csv(filename, *fields, **options)
+      end
+
+      def write_excel(filename, *fields, **options)
+        records.write_excel(filename, *fields, **options)
+      end
+
     end
   end
 end

data/lib/arql/ext/array.rb

@@ -13,8 +13,16 @@ class Array
     end.join("\n")
   end
 
-  def t(*attrs)
-    if attrs.present? && present? && first.is_a?(ActiveRecord::Base)
+  def t(*attrs, **options)
+    if (attrs.present? || options.present? && options[:except]) && present? && first.is_a?(ActiveRecord::Base)
+      if options.present? && options[:except]
+        attrs = first.attribute_names.map(&:to_sym) if attrs.empty?
+        if options[:except].is_a?(Regexp)
+          attrs.reject! { |e| e =~ options[:except] }
+        else
+          attrs -= [options[:except]].flatten
+        end
+      end
       puts Terminal::Table.new { |t|
         t << attrs
         t << :separator
@@ -62,4 +70,47 @@ class Array
     end
     t
   end
+
+  def write_csv(filename, *fields, **options)
+    generate_csv(filename, **options) do |csv|
+      if size > 0 && first.is_a?(ActiveRecord::Base)
+        if fields.empty?
+          fields = first.attributes.keys
+        else
+          fields = fields.map(&:to_s)
+        end
+        csv << fields
+      end
+      each do |row|
+        if row.is_a?(Array)
+          csv << row.map(&:to_s)
+        else
+          csv << row.slice(fields).values.map(&:to_s)
+        end
+      end
+    end
+  end
+
+  def write_excel(filename, *fields, **options)
+    sheet_name = options[:sheet_name] || 'Sheet1'
+    generate_excel(filename) do |workbook|
+      workbook.add_worksheet(name: sheet_name) do |sheet|
+        if size > 0 && first.is_a?(ActiveRecord::Base)
+          if fields.empty?
+            fields = first.attributes.keys
+          else
+            fields = fields.map(&:to_s)
+          end
+          sheet.add_row(fields, types: [:string] * fields.size)
+        end
+        each do |row|
+          if row.is_a?(Array)
+            sheet.add_row(row.map(&:to_s), types: [:string] * row.size)
+          else
+            sheet.add_row(row.slice(fields).values.map(&:to_s), types: [:string] * fields.size)
+          end
+        end
+      end
+    end
+  end
 end

data/lib/arql/ext/hash.rb

@@ -0,0 +1,25 @@
+class Hash
+  def write_excel(filename)
+    generate_excel(filename) do |workbook|
+      each do |sheet_name, sheet_data|
+        workbook.add_worksheet(name: sheet_name) do |sheet|
+          if sheet_data.is_a?(Hash) && sheet_data[:fields].present?
+              fields = sheet_data[:fields].map(&:to_s)
+            else
+              fields = sheet_data[:data].first.attributes.keys
+            end
+            sheet.add_row(fields, types: [:string] * fields.size)
+            sheet_data = sheet_data[:data]
+          end
+          sheet_data.each do |row|
+            if row.is_a?(Array)
+              sheet.add_row(row.map(&:to_s), types: [:string] * row.size)
+            else
+              sheet.add_row(row.slice(fields).values.map(&:to_s), types: [:string] * fields.size)
+            end
+          end
+        end
+      end
+    end
+  end
+end

data/lib/arql/ext/kernel.rb

@@ -1,4 +1,6 @@
 module Kernel
+  CSV_BOM = "\xef\xbb\xbf"
+
   def sql(sql)
     ActiveRecord::Base.connection.exec_query(sql)
   end
@@ -62,4 +64,45 @@ module Kernel
 
     outputs.each { |out| puts out; puts }
   end
+
+  def generate_csv(filename, **options, &block)
+    opts = {
+      col_sep: "\t",
+      row_sep: "\r\n"
+    }
+    opts.merge!(options.except(:encoding))
+    encoding = options[:encoding] || 'UTF-16LE'
+    File.open(File.expand_path(filename), "w:#{encoding}") do |file|
+      file.write(CSV_BOM)
+      file.write CSV.generate(**opts, &block)
+    end
+  end
+
+  def parse_csv(filename, **options)
+    encoding = options[:encoding] || 'UTF-16'
+    opts = {
+      headers: false,
+      col_sep: "\t",
+      row_sep: "\r\n"
+    }
+    opts.merge!(options.except(:encoding))
+    CSV.parse(IO.read(File.expand_path(filename), encoding: encoding, binmode: true).encode('UTF-8'), **opts).to_a
+  end
+
+  def generate_excel(filename)
+    Axlsx::Package.new do |package|
+      yield(package.workbook)
+      package.serialize(filename)
+    end
+  end
+
+  def parse_excel(filename)
+    xlsx = Roo::Excelx.new(File.expand_path(filename))
+    xlsx.sheets.each_with_object({}) do |sheet_name, result|
+      begin
+        result[sheet_name] = xlsx.sheet(sheet_name).to_a
+      rescue
+      end
+    end
+  end
 end

data/lib/arql/ext/object.rb

@@ -15,4 +15,8 @@ class Object
   def jjp
     puts jj
   end
+
+  def a
+    [self]
+  end
 end

data/lib/arql/repl.rb

@@ -7,7 +7,7 @@ module Arql
   class Repl
     def initialize
       Pry.config.prompt = Pry::Prompt.new("", "", prompt)
-      main_object.pry
+      Pry.start
     end
 
     def main_object
@@ -25,13 +25,17 @@ module Arql
     end
 
     def prompt
-      [proc do |obj, nest_level, _|
+      [proc do |obj, nest_level, pry_instance|
          if obj == main_object && nest_level == 0
            nest_level_prompt = ''
          else
-           nest_level_prompt = "(#{obj}:#{nest_level})"
+           nest_level_prompt = if nest_level.zero?
+                                 "(#{obj})"
+                               else
+                                 "(#{obj}:#{nest_level})"
+                               end
          end
-         "%s#{Rainbow('@').green}%s#{nest_level_prompt} %s " % [Rainbow('ARQL').red, Rainbow(App.prompt).yellow, Rainbow('❯').green]
+         "%s#{Rainbow('@').green}%s#{nest_level_prompt} [%d] %s " % [Rainbow('ARQL').red, Rainbow(App.prompt).yellow, pry_instance.input_ring.count, Rainbow('❯').green]
        end]
     end
   end

data/lib/arql/version.rb

@@ -1,3 +1,3 @@
 module Arql
-  VERSION = "0.1.32"
+  VERSION = "0.2.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.1.32
+  version: 0.2.3
 platform: ruby
 authors:
 - Liu Xiang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2020-08-26 00:00:00.000000000 Z
+date: 2020-09-24 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -164,6 +164,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 1.5.6
+- !ruby/object:Gem::Dependency
+  name: roo
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.3
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 2.8.3
+- !ruby/object:Gem::Dependency
+  name: caxlsx
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.2
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 3.0.2
 description: Use ActiveRecord and Pry as your favorite SQL query editor.
 email:
 - liuxiang921@gmail.com
@@ -199,6 +227,7 @@ files:
 - lib/arql/definition.rb
 - lib/arql/ext.rb
 - lib/arql/ext/array.rb
+- lib/arql/ext/hash.rb
 - lib/arql/ext/kernel.rb
 - lib/arql/ext/object.rb
 - lib/arql/ext/string.rb