mackuba.eu
+4
-2
lib/tasks/import.rake
+4
-2
lib/tasks/import.rake
···
9
9
raise "Required USER parameter missing"
10
10
end
11
11
12
+
user = User.find_or_create_by!(did: ENV['USER'])
13
+
12
14
case ENV['COLLECTION']
13
15
when 'likes'
14
-
queue = ItemQueue.new(Like.where(post: nil).to_a)
16
+
queue = ItemQueue.new(user.likes.where(post: nil).to_a)
15
17
importer = LikesImporter.new(ENV['USER'])
16
18
when 'reposts'
17
-
queue = ItemQueue.new(Repost.where(post: nil).to_a)
19
+
queue = ItemQueue.new(user.reposts.where(post: nil).to_a)
18
20
importer = RepostsImporter.new(ENV['USER'])
19
21
when nil
20
22
raise "Required COLLECTION parameter missing"