embulk-input-twitterarchive 0.1.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a29d2bbc465d3f797cc20e0edafc783e5d395c42
+  data.tar.gz: 00ea0f9c339f4e9509d2efa7bd3511594a804733
+SHA512:
+  metadata.gz: a1de1efcd5f17097af462c92977cdaf818dfad2922cfea4bf183508a00ac00b6b035e573c778946eb6c0f6193fffc87324df6719b58b3e98783cc07dbdecd9d2
+  data.tar.gz: eb836861de9cac44f2179471e7de6a433dec3b77eca2b7a32028ea743d833a3eb49050e7f5ea084bc3a330a51f20c25e6750434aea7c8415e4af55794dc483f5

data/.gitignore

@@ -0,0 +1,4 @@
+/.bundle/
+/Gemfile.lock
+/pkg/
+/.ruby-version

data/Gemfile

@@ -0,0 +1,2 @@
+source 'https://rubygems.org/'
+gemspec

data/LICENSE.txt

@@ -0,0 +1,21 @@
+
+MIT License
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.md

@@ -0,0 +1,32 @@
+# Twitterarchive input plugin for Embulk
+
+TODO: Write short description here and embulk-input-twitterarchive.gemspec file.
+
+## Overview
+
+* **Plugin type**: input
+* **Resume supported**: yes
+* **Cleanup supported**: yes
+* **Guess supported**: no
+
+## Configuration
+
+- **option1**: description (integer, required)
+- **option2**: description (string, default: `"myvalue"`)
+- **option3**: description (string, default: `null`)
+
+## Example
+
+```yaml
+in:
+  type: twitterarchive
+  option1: example1
+  option2: example2
+```
+
+
+## Build
+
+```
+$ rake
+```

data/Rakefile

@@ -0,0 +1,3 @@
+require "bundler/gem_tasks"
+
+task default: :build

data/embulk-input-twitterarchive.gemspec

@@ -0,0 +1,18 @@
+
+Gem::Specification.new do |spec|
+  spec.name          = "embulk-input-twitterarchive"
+  spec.version       = "0.1.0"
+  spec.authors       = ["takuya sato"]
+  spec.summary       = "Twitter Archive input plugin for Embulk"
+  spec.description   = "Loads records from Twitterarchive."
+  spec.email         = ["takuya0219@gmail.com"]
+  spec.licenses      = ["MIT"]
+  spec.homepage      = "https://github.com/nazo/embulk-input-twitterarchive"
+
+  spec.files         = `git ls-files`.split("\n") + Dir["classpath/*.jar"]
+  spec.test_files    = spec.files.grep(%r{^(test|spec)/})
+  spec.require_paths = ["lib"]
+
+  spec.add_development_dependency 'bundler', ['~> 1.0']
+  spec.add_development_dependency 'rake', ['>= 10.0']
+end

data/lib/embulk/input/twitterarchive.rb

@@ -0,0 +1,64 @@
+require 'json'
+require 'time'
+
+module Embulk
+  module Input
+
+    class TwitterarchiveInputPlugin < InputPlugin
+      # input plugin file name must be: embulk/input/<name>.rb
+      Plugin.register_input('twitterarchive', self)
+
+      def self.transaction(config, &control)
+        directory = config.param('directory', :string, default: nil)
+        index_file = File.read(File.join(directory, 'data/js/tweet_index.js'))
+        files = []
+        JSON.parse(index_file.gsub(/var tweet_index =  /, '')).each do |file_meta|
+          files.push(file_meta['file_name'])
+        end
+        task = {
+          'files' => files,
+          'directory' => directory
+        }
+
+        columns = [
+          Column.new(0, 'id', :long),
+          Column.new(1, 'text', :string),
+          Column.new(2, 'source', :string),
+          Column.new(3, 'in_reply_to_status_id', :long),
+          Column.new(4, 'created_at', :timestamp),
+        ]
+
+        resume(task, columns, files.length, &control)
+      end
+
+      def self.resume(task, columns, count, &control)
+        puts "Twitter Archive input started."
+        commit_reports = yield(task, columns, count)
+        puts "Twitter Archive input finished. Commit reports = #{commit_reports.to_json}"
+
+        next_config_diff = {}
+        return next_config_diff
+      end
+
+      def initialize(task, schema, index, page_builder)
+        super
+        @file = task['files'][index]
+        @directory = task['directory']
+      end
+
+      def run
+        puts "Twitter Archive input thread #{@index}..."
+
+        tweet_file = File.read(File.join(@directory, @file))
+        JSON.parse(tweet_file.gsub(/Grailbird\.data\.tweets_[0-9]+_[0-9]+ = /, '')).each do |tweet|
+          @page_builder.add([tweet['id'], tweet['text'], tweet['source'], tweet['in_reply_to_status_id'], Time.parse(tweet['created_at'])])
+        end
+        @page_builder.finish  # don't forget to call finish :-)
+
+        commit_report = {}
+        return commit_report
+      end
+    end
+
+  end
+end

metadata

@@ -0,0 +1,79 @@
+--- !ruby/object:Gem::Specification
+name: embulk-input-twitterarchive
+version: !ruby/object:Gem::Version
+  version: 0.1.0
+platform: ruby
+authors:
+- takuya sato
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-08-02 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: bundler
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '10.0'
+description: Loads records from Twitterarchive.
+email:
+- takuya0219@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- ".gitignore"
+- Gemfile
+- LICENSE.txt
+- README.md
+- Rakefile
+- embulk-input-twitterarchive.gemspec
+- lib/embulk/input/twitterarchive.rb
+homepage: https://github.com/nazo/embulk-input-twitterarchive
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Twitter Archive input plugin for Embulk
+test_files: []