neo4apis-activerecord 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: e872c76387db29333b1d6b009ea65045888f2fa2
+  data.tar.gz: 938e81267ab658e6857050cb20ee204127ae87a4
+SHA512:
+  metadata.gz: dca90c53f4e3cb387333e67ffbd2c0b2ebbf2a7ab088e04e3508507909d0a104340e991c381dda82c4311117f43d34d403d3cba1d65511da6d6337bd342b9c4b
+  data.tar.gz: 8c6f51d5a96207ab8e69f8679cd7c5e29a044ba860896f5de19fb564769f3309ce494971bbe6aed7fd815b8e8515e79149346bd7b3e5ddc300f4ff2ae586daa7

data/Gemfile

@@ -0,0 +1,17 @@
+source 'https://rubygems.org'
+
+gemspec
+
+group :development do
+  gem 'pry'
+  gem 'neo4apis', path: '../neo4apis'
+  gem 'rubocop'
+end
+
+group :test do
+  gem 'rspec'
+  gem 'sqlite3'
+
+  gem 'guard'
+  gem 'guard-rspec'
+end

data/lib/neo4apis-activerecord.rb

@@ -0,0 +1,4 @@
+require 'neo4apis'
+
+require 'neo4apis/activerecord'
+require 'neo4apis/cli/activerecord'

data/lib/neo4apis/activerecord.rb

@@ -0,0 +1,56 @@
+require 'neo4apis'
+require 'ostruct'
+
+module Neo4Apis
+  class ActiveRecord < Base
+    prefix nil
+
+    batch_size 1000
+
+    def self.model_importer(model_class)
+      uuid model_class.name.to_sym, model_class.primary_key
+
+      importer model_class.name.to_sym do |object|
+        node = add_model_node model_class, object
+
+        model_class.reflect_on_all_associations.each do |association_reflection|
+          case association_reflection.macro
+          when :belongs_to, :has_one
+            if options[:"import_#{association_reflection.macro}"]
+              referenced_object = object.send(association_reflection.name)
+              add_model_relationship association_reflection.name, node, referenced_object
+            end
+          when :has_many
+            if options[:import_has_many]
+              object.send(association_reflection.name).each do |referenced_object|
+                add_model_relationship association_reflection.name, node, referenced_object
+              end
+            end
+          end
+        end
+      end
+    end
+
+    def add_model_relationship(relationship_name, node, referenced_object)
+      referenced_class = referenced_object.class
+      referenced_node = add_model_node referenced_class, referenced_object
+
+      add_relationship relationship_name, node, referenced_node
+    end
+
+    def add_model_node(model_class, object)
+      object_data = OpenStruct.new
+
+      object.attributes.each do |column, value|
+        v = if coder = model_class.serialized_attributes[column]
+              coder.dump(value)
+            else
+              value
+            end
+        object_data.send("#{column}=", v)
+      end
+
+      add_node model_class.name.to_sym, object_data, model_class.column_names
+    end
+  end
+end

data/lib/neo4apis/cli/activerecord.rb

@@ -0,0 +1,100 @@
+require 'active_record'
+require 'active_support/inflector'
+require 'thor'
+require 'colorize'
+
+module Neo4Apis
+  module CLI
+    class ActiveRecord < Thor
+      class_option :config_path, type: :string,  default: 'config/database.yml'
+
+      class_option :import_associations, type: :boolean, default: false
+      class_option :import_belongs_to, type: :boolean, default: nil
+      class_option :import_has_one, type: :boolean, default: nil
+      class_option :import_has_many, type: :boolean, default: nil
+
+      class_option :guess_associations, type: :boolean, default: false
+
+      class_option :active_record_config_path, type: :string, default: './config/database.yml'
+      class_option :active_record_environment, type: :string, default: 'development'
+
+      desc 'models MODELS_OR_TABLE_NAMES', 'Import SQL tables via ActiveRecord models'
+      def models(*models_or_table_names)
+        setup
+
+        puts 'models_or_table_names', models_or_table_names.inspect
+        model_classes = models_or_table_names.map(&method(:get_model))
+
+        model_classes.each do |model_class|
+          ::Neo4Apis::ActiveRecord.model_importer(model_class)
+        end
+
+        neo4apis_client.batch do
+          model_classes.each do |model_class|
+            model_class.find_each do |object|
+              neo4apis_client.import model_class.name.to_sym, object
+            end
+          end
+        end
+      end
+
+      private
+
+      def setup
+        if File.exist?('config/environment.rb')
+          # Rails
+          require './config/environment'
+        else
+          puts 'active_record_config', active_record_config.inspect
+          ::ActiveRecord::Base.establish_connection(active_record_config)
+        end
+      end
+
+      NEO4APIS_CLIENT_CLASS = ::Neo4Apis::ActiveRecord
+
+      def neo4apis_client
+        @neo4apis_client ||= NEO4APIS_CLIENT_CLASS.new(Neo4j::Session.open(:server_db, parent_options[:neo4j_url]),
+                                                       import_belongs_to: import_association?(:belongs_to),
+                                                       import_has_one: import_association?(:has_one),
+                                                       import_has_many: import_association?(:has_many))
+      end
+
+      def import_association?(type)
+        options[:"import_#{type}"].nil? ? options[:import_associations] : options[:"import_#{type}"]
+      end
+
+      def get_model(model_or_table_name)
+        model_class = model_or_table_name
+        model_class = model_or_table_name.classify unless model_or_table_name.match(/^[A-Z]/)
+        model_class.constantize
+      rescue NameError
+        Object.const_set(model_class, Class.new(::ActiveRecord::Base)).tap do |model_class|
+          apply_guessed_model_associations!(model_class) if options[:guess_associations]
+        end
+      end
+
+      def apply_guessed_model_associations!(model_class)
+        model_class.columns.each do |column|
+          next if not column.name.match(/_id$/)
+
+          begin
+            base = column.name.humanize.tableize.split(' ').join('_')
+
+            model_class.belongs_to base.singularize.to_sym
+          rescue NameError
+          end
+        end
+      end
+
+      def active_record_config
+        require 'yaml'
+        YAML.load(File.read(options[:active_record_config_path]))[options[:active_record_environment]]
+      end
+    end
+
+    class Base < Thor
+      desc 'activerecord SUBCOMMAND ...ARGS', 'methods of importing data automagically from Twitter'
+      subcommand 'activerecord', CLI::ActiveRecord
+    end
+  end
+end

data/neo4apis-activerecord.gemspec

@@ -0,0 +1,23 @@
+lib = File.expand_path('../lib/', __FILE__)
+$LOAD_PATH.unshift lib unless $LOAD_PATH.include?(lib)
+
+Gem::Specification.new do |s|
+  s.name     = 'neo4apis-activerecord'
+  s.version  = '0.1.0'
+  s.required_ruby_version = '>= 1.9.1'
+
+  s.authors  = 'Brian Underwood'
+  s.email    = 'public@brian-underwood.codes'
+  s.homepage = 'https://github.com/neo4jrb/neo4apis-activerecord/'
+  s.summary = 'An ruby gem to import SQL data to neo4j using activerecord'
+  s.license = 'MIT'
+  s.description = <<-EOF
+A ruby gem using neo4apis to make importing SQL data to neo4j easy
+  EOF
+
+  s.require_path = 'lib'
+  s.files = Dir.glob('{bin,lib,config}/**/*') + %w(README.md Gemfile neo4apis-activerecord.gemspec)
+
+  s.add_dependency('neo4apis', '~> 0.4.0')
+  s.add_dependency('activerecord', '~> 4.0')
+end

metadata

@@ -0,0 +1,78 @@
+--- !ruby/object:Gem::Specification
+name: neo4apis-activerecord
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- Brian Underwood
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-01-07 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: neo4apis
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.4.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.4.0
+- !ruby/object:Gem::Dependency
+  name: activerecord
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '4.0'
+description: |
+  A ruby gem using neo4apis to make importing SQL data to neo4j easy
+email: public@brian-underwood.codes
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- README.md
+- lib/neo4apis-activerecord.rb
+- lib/neo4apis/activerecord.rb
+- lib/neo4apis/cli/activerecord.rb
+- neo4apis-activerecord.gemspec
+homepage: https://github.com/neo4jrb/neo4apis-activerecord/
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: 1.9.1
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.3
+signing_key: 
+specification_version: 4
+summary: An ruby gem to import SQL data to neo4j using activerecord
+test_files: []