neo4apis-twitter 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: f9ac29686dd644899a2f184a99f653f1b15d1b78
-  data.tar.gz: cb662fae49dd85790bdd8af896f87ea3f3ebe7a0
+  metadata.gz: 43816b4c7d84582347f3d469727ff75ab9219cad
+  data.tar.gz: b7fb544aa5507bec0c2094723479baa4078a39f7
 SHA512:
-  metadata.gz: 33f5489c963868868b654557145dfa2858d6419977a42031a04ced7e93a2286ddf600d0552ee98629c2a9f5e9fde2a36f8ad914a7b1b52ddf8e9c4c09f5eb2f5
-  data.tar.gz: c9eca1f76eaf51cabec63e3c87ee15418ae134a189e2c6cc6a75227e4a3a86273cce39c5931460efafad402f590b76f4e3c1cad4882349b3a27dd520b820529f
+  metadata.gz: fec12fd33e54c2601938c8b5e816ca99a26f8e359027aa8dcc1153d7b51b01fcb306024a8628b3c0d6f9e852fa53528a18fb19e85f43503903c01559e1a322a3
+  data.tar.gz: e00ac6020eed2fdc76fd9fe02303c2f66fe43c7870ce61b52aee6a9b8112ec78a04f4f26a844578ba298c8dbac267e7912ed4a5f7d2ef17c6ae0bbb284dea18f

data/README.md

@@ -1,16 +1,23 @@
 
 neo4apis-twitter is a ruby gem for making importing data from twitter to neo4j easy
 
-The `twitter` gem is used and thus the methods in this API try to match that API as closely as possible
+This adapter supports objects created from the `twitter` gem.
 
 ```ruby
 
 twitter_client = Twitter::REST::Client.new ...
 
-neo4japis_twitter = Neo4Apis::Twitter.new(Neo4j::Session.open, twitter_client: twitter_client)
-
-neo4japis_twitter.batch do
-  neo4japis_twitter.import_search("ebola", :result_type => "recent")
+neo4japis_twitter = Neo4Apis::Twitter.new(Neo4j::Session.open, import_retweets: true)
+
+neo4japis_twitter.batch do 
+  twitter_client.search("ebola", :result_type => "recent").take(100).each do |tweet|
+    # Imports:
+    #  * The tweet
+    #  * The tweeter, 
+    #  * The original tweet if a retweet
+    #  * The user for the original tweet
+    import :Tweet, tweet
+  end
 end
 
 ```

data/lib/neo4apis/twitter.rb

@@ -2,32 +2,13 @@ require 'neo4apis'
 
 module Neo4Apis
   class Twitter < Base
-    PREFIX = 'twitter'
+    uuid :Tweet, :id
+    uuid :User, :id
 
-    def initialize(neo4j_client, options = {})
-      @client = options[:twitter_client]
+    importer :Tweet do |tweet|
+      user_node = import :User, tweet.user
 
-      raise ArgumentError, "Invalid client: #{@client.inspect}" if not @client.is_a?(::Twitter::REST::Client)
-
-      options[:uuids] = (options[:uuids] || {}).merge({
-        Tweet: :id,
-        User: :id
-      })
-
-      super(neo4j_client, options)
-    end
-
-    def import_search(*args)
-      @client.search(*args).take(100).each do |tweet|
-        add_tweet(tweet)
-      end
-    end
-
-    private
-    
-    def add_tweet(tweet)
-      user_node = add_user(tweet.user)
-      retweeted_tweet_node = add_tweet(tweet.retweeted_tweet) if tweet.retweeted_tweet?
+      retweeted_tweet_node = import :Tweet, tweet.retweeted_tweet if options[:import_retweets] && tweet.retweeted_tweet?
 
       node = add_node :Tweet, {
         id: tweet.id,
@@ -35,12 +16,12 @@ module Neo4Apis
       }
 
       add_relationship(:tweeted, user_node, node)
-      add_relationship(:retweets, node, retweeted_tweet_node) if tweet.retweeted_tweet?
+      add_relationship(:retweets, node, retweeted_tweet_node) if options[:import_retweets] && tweet.retweeted_tweet?
 
       node
     end
 
-    def add_user(user)
+    importer :User do |user|
       add_node :User, {
         id: user.id,
         screen_name: user.screen_name,

data/lib/neo4apis/twitter/version.rb

@@ -1,6 +1,8 @@
+require 'neo4apis'
+
 module Neo4Apis
-  class Twitter
-    VERSION = '0.0.1'
+  class Twitter < Base
+    VERSION = '0.0.2'
   end
 end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: neo4apis-twitter
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.2
 platform: ruby
 authors:
 - Brian Underwood
@@ -62,4 +62,3 @@ signing_key:
 specification_version: 4
 summary: An ruby gem to import twitter data to neo4j
 test_files: []
-has_rdoc: