arql 0.2.14 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: acc076bc57fba6651a6d3b5c5232e002d0ad3c23dffb4c11b09a5f4697ec65b5
-  data.tar.gz: 9a868a87e0d9658800de3740b6b42697cc65c07535b76c67bc88d38c37c82433
+  metadata.gz: d63702948cb453c9e8fe42743024faa3cb1c7102a2ba592076d7a1005f64392b
+  data.tar.gz: 42c4e35504a6334990283aa0cc433a507339d4dd1bfb804bb81454d57b7187eb
 SHA512:
-  metadata.gz: 6dbaa27d969c85aea66fc13d443d91080cbfa3550a4728a20440385927a6527ff758dc23591f1b220e0b2c0f9ed5f4012609ce00c655251be64f22605f2f39c8
-  data.tar.gz: 06cb755475b4f6cffec68aa681fea2706577f01f0c0771024f79b7f9f92a039cfb35e815ed251d50198229a908051b1c103828d773481f29d11a33c5b5417a5d
+  metadata.gz: 668284267a0469734710a9b5c7490116af4ebeb75de5c6754d8f9e733c409b4798a5d4b3216eb55f8d7b50569502d84fb22cc2a1bdf896d330d4ae0ce7ca4af6
+  data.tar.gz: bdf6f3900ba0b936cfa2a0d48dccc981215f2f2f1a6397b907d4e81de430f99d87bf2d22faa6b3f0fb16281b5a1b83ccbfd03e0f6f7b7d4783fc1090fc796b55

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    arql (0.2.14)
+    arql (0.3.0)
       activerecord (~> 6.0.3)
       activesupport (~> 6.0.3)
       caxlsx (~> 3.0.2)
@@ -32,7 +32,7 @@ GEM
       tzinfo (~> 1.1)
       zeitwerk (~> 2.2, >= 2.2.2)
     byebug (11.1.3)
-    caxlsx (3.0.2)
+    caxlsx (3.0.4)
       htmlentities (~> 4.3, >= 4.3.4)
       mimemagic (~> 0.3)
       nokogiri (~> 1.10, >= 1.10.4)
@@ -74,10 +74,10 @@ GEM
     terminal-table (1.8.0)
       unicode-display_width (~> 1.1, >= 1.1.1)
     thread_safe (0.3.6)
-    tzinfo (1.2.8)
+    tzinfo (1.2.9)
       thread_safe (~> 0.1)
     unicode-display_width (1.7.0)
-    yard (0.9.25)
+    yard (0.9.26)
     zeitwerk (2.4.2)
 
 PLATFORMS
@@ -88,4 +88,4 @@ DEPENDENCIES
   rake (~> 12.0)
 
 BUNDLED WITH
-   2.1.4
+   2.2.3

data/arql.gemspec

@@ -10,7 +10,7 @@ Gem::Specification.new do |spec|
   spec.description   = %{Use ActiveRecord and Pry as your favorite SQL query editor.}
   spec.homepage      = "https://github.com/lululau/arql"
   spec.license       = "MIT"
-  spec.required_ruby_version = Gem::Requirement.new(">= 2.3.0")
+  spec.required_ruby_version = Gem::Requirement.new(">= 2.6.0")
 
 
 

data/lib/arql.rb

@@ -10,6 +10,7 @@ require "arql/repl"
 require "arql/ssh_proxy"
 require "arql/app"
 require "arql/cli"
+require "arql/mysqldump"
 
 module Arql
 end

data/lib/arql/app.rb

@@ -4,7 +4,7 @@ module Arql
   class App
 
     class << self
-      attr_accessor :log_io, :env, :prompt, :instance
+      attr_accessor :log_io, :env, :prompt, :instance, :connect_options
 
       def config
         @@effective_config
@@ -27,7 +27,8 @@ module Arql
       require "arql/definition"
       @options = options
       App.env = @options.env
-      Connection.open(connect_options)
+      App.connect_options = connect_options
+      Connection.open(App.connect_options)
       @definition = Definition.new(effective_config)
       load_initializer!
       App.instance = self

data/lib/arql/connection.rb

@@ -4,6 +4,9 @@ module Arql
       def open(options)
         ActiveRecord::Base.establish_connection(options)
         $C = ActiveRecord::Base.connection
+        $C.define_singleton_method(:dump) do |filename, no_create_db=false|
+          Arql::Mysqldump.new.dump_database(filename, no_create_db)
+        end
       end
     end
   end

data/lib/arql/definition.rb

@@ -35,6 +35,10 @@ module Arql
       [self].write_excel(filename, *fields, **options)
     end
 
+    def dump(filename, batch_size=500)
+      [self].dump(filename, batch_size)
+    end
+
     included do
     end
 
@@ -66,6 +70,10 @@ module Arql
       def to_create_sql
         ActiveRecord::Base.connection.exec_query("show create table #{table_name}").rows.last.last
       end
+
+      def dump(filename, no_create_table=false)
+        Arql::Mysqldump.new.dump_table(filename, table_name, no_create_table)
+      end
     end
   end
 
@@ -237,6 +245,9 @@ module Arql
         records.write_excel(filename, *fields, **options)
       end
 
+      def dump(filename, batch_size=500)
+        records.dump(filename, batch_size)
+      end
     end
   end
 end

data/lib/arql/ext/array.rb

@@ -127,4 +127,13 @@ class Array
       end
     end
   end
+
+  def dump(filename, batch_size=500)
+    File.open(File.expand_path(filename), 'w') do |file|
+      group_by(&:class).each do |(klass, records)|
+        file.puts(klass.to_upsert_sql(records, batch_size))
+      end
+    end
+    {size: size, file: File.expand_path(filename)}
+  end
 end

data/lib/arql/ext/kernel.rb

@@ -95,7 +95,7 @@ module Kernel
   def generate_excel(filename)
     Axlsx::Package.new do |package|
       yield(package.workbook)
-      package.serialize(filename)
+      package.serialize(File.expand_path(filename))
     end
   end
 

data/lib/arql/ext/string.rb

@@ -2,4 +2,12 @@ class String
   def p
     puts self
   end
+
+  def expa
+    File.expand_path(self)
+  end
+
+  def f
+    expa
+  end
 end

data/lib/arql/mysqldump.rb

@@ -0,0 +1,44 @@
+module Arql
+  class Mysqldump
+
+    def initialize(options = nil)
+      options ||= App.connect_options
+      @options = options
+      if options[:socket]
+        port_or_sock = "-S #{options[:socket]}"
+      else
+        port_or_sock = "-P #{options[:port] || 3306}"
+      end
+      @base_dump_cmd = "mysqldump -u %{user} -h %{host} %{port_or_sock} %{password} --skip-lock-tables " % {
+        user: options[:username],
+        host: options[:host],
+        password: options[:password].presence.try { |e| "-p#{e}" } || '',
+        port_or_sock: port_or_sock
+      }
+    end
+
+    def dump_table(filename, table_name, no_create_table = false)
+      system dump_table_cmd(table_name, no_create_table) + " > #{filename}"
+    end
+
+    def dump_database(filename, no_create_db = false)
+      system dump_database_cmd(no_create_db) + " > #{filename}"
+    end
+
+    def dump_table_cmd(table_name, no_create_table = false)
+      @base_dump_cmd + " " + if no_create_table
+                               "--no-create-info #{@options[:database]} #{table_name}"
+                             else
+                               "--add-drop-table #{@options[:database]} #{table_name}"
+                             end
+    end
+
+    def dump_database_cmd(no_create_db = false)
+      @base_dump_cmd + " " + if no_create_db
+                               "--no-create-db --add-drop-database --databases #{@options[:database]}"
+                             else
+                               "--add-drop-database --add-drop-table --databases #{@options[:database]}"
+                             end
+    end
+  end
+end

data/lib/arql/version.rb

@@ -1,3 +1,3 @@
 module Arql
-  VERSION = "0.2.14"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: arql
 version: !ruby/object:Gem::Version
-  version: 0.2.14
+  version: 0.3.0
 platform: ruby
 authors:
 - Liu Xiang
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2021-01-03 00:00:00.000000000 Z
+date: 2021-01-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: mysql2
@@ -251,6 +251,7 @@ files:
 - lib/arql/ext/time.rb
 - lib/arql/id.rb
 - lib/arql/multi_io.rb
+- lib/arql/mysqldump.rb
 - lib/arql/repl.rb
 - lib/arql/ssh_proxy.rb
 - lib/arql/ssh_proxy_patch.rb
@@ -267,14 +268,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 2.3.0
+      version: 2.6.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
+rubygems_version: 3.2.3
 signing_key:
 specification_version: 4
 summary: Rails ActiveRecord + Pry is the best SQL query editor