Put CSV import function under its own module
parent
ad4a06fca5
commit
ef8dc7272b
@ -1,27 +1,33 @@
|
|||||||
require "csv"
|
require "csv"
|
||||||
|
|
||||||
def parse_subscription_export_csv(csv_content : String)
|
struct Invidious::User
|
||||||
rows = CSV.new(csv_content, headers: true)
|
module Import
|
||||||
subscriptions = Array(String).new
|
extend self
|
||||||
|
|
||||||
# Counter to limit the amount of imports.
|
# Parse a youtube CSV subscription file
|
||||||
# This is intended to prevent DoS.
|
def parse_subscription_export_csv(csv_content : String)
|
||||||
row_counter = 0
|
rows = CSV.new(csv_content, headers: true)
|
||||||
|
subscriptions = Array(String).new
|
||||||
|
|
||||||
rows.each do |row|
|
# Counter to limit the amount of imports.
|
||||||
# Limit to 1200
|
# This is intended to prevent DoS.
|
||||||
row_counter += 1
|
row_counter = 0
|
||||||
break if row_counter > 1_200
|
|
||||||
|
|
||||||
# Channel ID is the first column in the csv export we can't use the header
|
rows.each do |row|
|
||||||
# name, because the header name is localized depending on the
|
# Limit to 1200
|
||||||
# language the user has set on their account
|
row_counter += 1
|
||||||
channel_id = row[0].strip
|
break if row_counter > 1_200
|
||||||
|
|
||||||
next if channel_id.empty?
|
# Channel ID is the first column in the csv export we can't use the header
|
||||||
|
# name, because the header name is localized depending on the
|
||||||
|
# language the user has set on their account
|
||||||
|
channel_id = row[0].strip
|
||||||
|
|
||||||
subscriptions << channel_id
|
next if channel_id.empty?
|
||||||
end
|
subscriptions << channel_id
|
||||||
|
end
|
||||||
|
|
||||||
return subscriptions
|
return subscriptions
|
||||||
|
end
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue