mirror of
https://gitea.invidious.io/iv-org/invidious-copy-2022-08-14.git
synced 2024-08-15 00:53:20 +00:00
Put CSV import function under its own module
This commit is contained in:
parent
ad4a06fca5
commit
ef8dc7272b
1 changed files with 24 additions and 18 deletions
|
@ -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?
|
||||||
|
subscriptions << channel_id
|
||||||
|
end
|
||||||
|
|
||||||
|
return subscriptions
|
||||||
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
return subscriptions
|
|
||||||
end
|
end
|
||||||
|
|
Loading…
Reference in a new issue