dba 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b66a9e45cd9054a65b9453a5c981d6637d1060f2d0862d45874276d630bd8859
-  data.tar.gz: d1ddad863691a445c275cdbb580e6f463a8bc6a7bf4e6856b2ff3b110ac7c9f7
+  metadata.gz: 4212cdf60ed43e28c00a0bb815eabf6ff0912535713a27a425f077e71ef27f4d
+  data.tar.gz: a3da3be24d5e75207eb239fd814307c02457d22b286f1de548b5429c830dec60
 SHA512:
-  metadata.gz: 6592ef3d6089bb864d72c3eb1606b8a3907d1bcdbe114a29b426c253e7d8864ad90c6f7a23663184339c1040692d68b82ce73ee151fd18fba13c5d618a1b4c0b
-  data.tar.gz: 683c320b294abcb74b42d74de889bf0205f79075b1c51f541c1e64eabfe418e662c5fc88a33b0ef22cdcf558d07299a76352cfe21a870d009b36aa4b0102b356
+  metadata.gz: 84c04318233dc14cacafb8b5bd065fb1d44d713cbbf39f50463dd1c46e12bc49599560e637dc0179ccbcc44b8c2cc3bde45a9244daf5d28867a0d538762fe352
+  data.tar.gz: 0a62d78ddcd245fc5d8781f3af94dda6b463f5dac11623fe11be546339de7bdedd8827d08b68db6da85a1fb377bf2316f724bb83ac54331bf850be7f557c4d62

data/CHANGES.md

@@ -0,0 +1,11 @@
+# 1.1.0
+
+* Fixed sample command for postgres
+
+* Added dump command for dumping data to CSV, LDJSON, or YAML
+
+* Added load command for loading data from CSV, LDJSON, or YAML
+
+# 1.0.0
+
+* First version!

data/README.md

@@ -1,6 +1,6 @@
 # dba
 
-A developer tool for working with development databases.
+Ruby command line tool for working with development databases.
 
 
 ## Installation
@@ -19,9 +19,11 @@ You can connect to any database supported by [sequel](https://rubygems.org/gems/
     Usage: dba COMMAND
 
       dba diff URL
+      dba dump TABLE EXTENSION
       dba edit TABLE IDENTIFIER
       dba find TABLE IDENTIFIER
       dba indexes [TABLE]
+      dba load PATH
       dba pull TABLE URL
       dba sample TABLE [COLUMN]
       dba schema [TABLE]

data/dba.gemspec

@@ -1,14 +1,14 @@
 Gem::Specification.new do |s|
   s.name = 'dba'
-  s.version = '1.0.0'
+  s.version = '1.1.0'
   s.license = 'GPL-3.0'
   s.platform = Gem::Platform::RUBY
   s.authors = ['Tim Craft']
   s.email = ['mail@timcraft.com']
   s.homepage = 'https://github.com/readysteady/dba'
-  s.description = 'A developer tool for working with development databases'
+  s.description = 'Ruby command line tool for working with development databases'
   s.summary = 'See description'
-  s.files = Dir.glob('lib/**/*.rb') + %w[LICENSE.txt README.md dba.gemspec]
+  s.files = Dir.glob('lib/**/*.rb') + %w[CHANGES.md LICENSE.txt README.md dba.gemspec]
   s.required_ruby_version = '>= 2.3.0'
   s.add_dependency('zeitwerk', '~> 2', '>= 2.2')
   s.add_dependency('sequel', '~> 5')

data/lib/dba.rb

@@ -1,4 +1,4 @@
-# Copyright (c) 2019 TIMCRAFT
+# Copyright (c) 2019-2020 TIMCRAFT
 #
 # This program is free software: you can redistribute it and/or modify
 # it under the terms of the GNU General Public License as published by
@@ -17,7 +17,12 @@ module DBA
 
   loader = Zeitwerk::Loader.new
   loader.tag = File.basename(__FILE__, '.rb')
-  loader.inflector.inflect('dba' => 'DBA')
+  loader.inflector.inflect({
+    'dba' => 'DBA',
+    'csv' => 'CSV',
+    'ldjson' => 'LDJSON',
+    'yaml' => 'YAML'
+  })
   loader.push_dir(__dir__)
   loader.setup
 end

data/lib/dba/csv.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+require 'csv'
+
+module DBA::CSV
+  def self.dump(database, table_name, path)
+    columns = nil
+
+    ::CSV.open(path, 'wb') do |csv|
+      database[table_name].each_with_index do |row, index|
+        if index.zero?
+          columns = row.keys
+
+          csv << columns
+        end
+
+        csv << row.values_at(*columns)
+      end
+    end
+  end
+
+  def self.load(path, database, table_name)
+    dataset = database[table_name]
+
+    rows = ::CSV.read(path)
+
+    headers = rows.shift
+
+    dataset.import(headers, rows)
+  end
+end

data/lib/dba/dump.rb

@@ -0,0 +1,26 @@
+# frozen_string_literal: true
+
+class DBA::Dump < DBA::Command
+  ADAPTERS = {
+    'csv' => :CSV,
+    'ldjson' => :LDJSON,
+    'yml' => :YAML,
+    'yaml' => :YAML
+  }
+
+  def call(table, extension)
+    self.table_name = table
+
+    output_path = "#{table_name}.#{extension}"
+
+    adapter = ADAPTERS.fetch(extension) { raise DBA::Error, 'unsupported file extension' }
+
+    adapter = DBA.const_get(adapter)
+
+    rows = database[table_name].count
+
+    return if rows.zero?
+
+    adapter.dump(database, table_name, output_path)
+  end
+end

data/lib/dba/ldjson.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+require 'json'
+
+module DBA::LDJSON
+  def self.dump(database, table_name, path)
+    File.open(path, 'w+') do |io|
+      database[table_name].each do |row|
+        io.puts ::JSON.generate(row.compact)
+      end
+    end
+  end
+
+  def self.load(path, database, table_name)
+    dataset = database[table_name]
+
+    database.transaction do
+      File.readlines(path).each do |line|
+        dataset.insert(::JSON.parse(line))
+      end
+    end
+  end
+end

data/lib/dba/load.rb

@@ -0,0 +1,34 @@
+# frozen_string_literal: true
+
+class DBA::Load < DBA::Command
+  ADAPTERS = {
+    '.csv' => :CSV,
+    '.ldjson' => :LDJSON,
+    '.yml' => :YAML,
+    '.yaml' => :YAML
+  }
+
+  def call(path)
+    file_list(path).each do |file|
+      extension = File.extname(file)
+
+      adapter = ADAPTERS.fetch(extension) { raise DBA::Error, 'unsupported file extension' }
+
+      adapter = DBA.const_get(adapter)
+
+      table_name = File.basename(file, extension).to_sym
+
+      adapter.load(file, database, table_name)
+    end
+  end
+
+  private
+
+  def file_list(path)
+    if File.directory?(path)
+      Dir.glob(File.join(path, '*.{csv,ldjson,yml,yaml}'))
+    else
+      [path]
+    end
+  end
+end

data/lib/dba/sample.rb

@@ -4,14 +4,13 @@ class DBA::Sample < DBA::Command
 
     column_name = column.to_sym if column
 
-    random_rows = database[table_name].order(random_function)
-
     if column_name
-      random_rows.distinct.select(column_name).limit(20).each do |row|
+      dataset = database[table_name].distinct.select(column_name)
+      dataset.from_self.order(random_function).limit(20).each do |row|
         puts row[column_name]
       end
     else
-      random_rows.first(3).each do |row|
+      database[table_name].order(random_function).first(3).each do |row|
         printer.print(row)
         printer.print_line
       end

data/lib/dba/shell.rb

@@ -35,9 +35,11 @@ module DBA::Shell
   def commands
     {
       'diff' => :Diff,
+      'dump' => :Dump,
       'edit' => :Edit,
       'find' => :Find,
       'indexes' => :Indexes,
+      'load' => :Load,
       'pull' => :Pull,
       'sample' => :Sample,
       'schema' => :Schema,
@@ -52,7 +54,7 @@ module DBA::Shell
   end
 
   def program_name
-    File.basename($0)
+    'dba'
   end
 
   def print_usage

data/lib/dba/yaml.rb

@@ -0,0 +1,20 @@
+# frozen_string_literal: true
+require 'yaml'
+
+module DBA::YAML
+  def self.dump(database, table_name, path)
+    File.open(path, 'w+') do |io|
+      io.write ::YAML.dump(database[table_name].all.map(&:compact))
+    end
+  end
+
+  def self.load(path, database, table_name)
+    dataset = database[table_name]
+
+    database.transaction do
+      ::YAML.load_file(path).each do |row|
+        dataset.insert(row)
+      end
+    end
+  end
+end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: dba
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Tim Craft
-autorequire: 
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2019-10-22 00:00:00.000000000 Z
+date: 2020-10-13 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: zeitwerk
@@ -58,7 +58,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0'
-description: A developer tool for working with development databases
+description: Ruby command line tool for working with development databases
 email:
 - mail@timcraft.com
 executables:
@@ -66,17 +66,22 @@ executables:
 extensions: []
 extra_rdoc_files: []
 files:
+- CHANGES.md
 - LICENSE.txt
 - README.md
 - bin/dba
 - dba.gemspec
 - lib/dba.rb
 - lib/dba/command.rb
+- lib/dba/csv.rb
 - lib/dba/database.rb
 - lib/dba/diff.rb
+- lib/dba/dump.rb
 - lib/dba/edit.rb
 - lib/dba/find.rb
 - lib/dba/indexes.rb
+- lib/dba/ldjson.rb
+- lib/dba/load.rb
 - lib/dba/printer.rb
 - lib/dba/pull.rb
 - lib/dba/row_command.rb
@@ -88,11 +93,12 @@ files:
 - lib/dba/table_command.rb
 - lib/dba/table_schema.rb
 - lib/dba/tables.rb
+- lib/dba/yaml.rb
 homepage: https://github.com/readysteady/dba
 licenses:
 - GPL-3.0
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -107,8 +113,8 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.0.3
-signing_key: 
+rubygems_version: 3.1.4
+signing_key:
 specification_version: 4
 summary: See description
 test_files: []