Updated csv reading as per feedback and ran

Signed-off-by: thtmnisamnstr <gavinj1984@gmail.com>
This commit is contained in:
thtmnisamnstr 2023-04-03 17:07:58 -07:00
parent 3341929060
commit fffdaa1410

View File

@ -40,7 +40,7 @@ struct Invidious::User
title = csv_head[4] title = csv_head[4]
description = csv_head[5] description = csv_head[5]
visibility = csv_head[6] visibility = csv_head[6]
if visibility.compare("Public", case_insensitive: true) == 0 if visibility.compare("Public", case_insensitive: true) == 0
privacy = PlaylistPrivacy::Public privacy = PlaylistPrivacy::Public
else else
@ -51,34 +51,33 @@ struct Invidious::User
Invidious::Database::Playlists.update_description(playlist.id, description) Invidious::Database::Playlists.update_description(playlist.id, description)
# Add each video to the playlist from the body content # Add each video to the playlist from the body content
CSV.each_row(raw_body) do |row| csv_body = CSV.new(raw_body, headers: true)
if row.size >= 1 csv_body.each do |row|
video_id = row[0] video_id = row[0]
if playlist if playlist
next if !video_id next if !video_id
next if video_id == "Video Id" next if video_id == "Video Id"
begin begin
video = get_video(video_id) video = get_video(video_id)
rescue ex rescue ex
next next
end
playlist_video = PlaylistVideo.new({
title: video.title,
id: video.id,
author: video.author,
ucid: video.ucid,
length_seconds: video.length_seconds,
published: video.published,
plid: playlist.id,
live_now: video.live_now,
index: Random::Secure.rand(0_i64..Int64::MAX),
})
Invidious::Database::PlaylistVideos.insert(playlist_video)
Invidious::Database::Playlists.update_video_added(playlist.id, playlist_video.index)
end end
playlist_video = PlaylistVideo.new({
title: video.title,
id: video.id,
author: video.author,
ucid: video.ucid,
length_seconds: video.length_seconds,
published: video.published,
plid: playlist.id,
live_now: video.live_now,
index: Random::Secure.rand(0_i64..Int64::MAX),
})
Invidious::Database::PlaylistVideos.insert(playlist_video)
Invidious::Database::Playlists.update_video_added(playlist.id, playlist_video.index)
end end
end end