Git-Mirrors/invidious
1
0
Fork 0
mirror of https://github.com/iv-org/invidious.git synced 2024-12-18 20:24:36 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add helper function parse_subscription_export_csv() which parses the csv format returned by the subscription exporter

Browse Source
This commit is contained in:
bbielsa 2021-10-26 21:19:20 -04:00 committed by Samantaz Fox
parent 3fba342ed2
commit 7cbd79fee5
No known key found for this signature in database
GPG Key ID: F42821059186176E
1 changed files with 22 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
src/invidious/helpers/utils.cr
View File

@ -1,3 +1,5 @@
require "csv"
# See http://www.evanmiller.org/how-not-to-sort-by-average-rating.html # See http://www.evanmiller.org/how-not-to-sort-by-average-rating.html
def ci_lower_bound(pos, n) def ci_lower_bound(pos, n)
if n == 0 if n == 0
@ -367,3 +369,23 @@ def fetch_random_instance
return filtered_instance_list.sample(1)[0] return filtered_instance_list.sample(1)[0]
end end
def parse_subscription_export_csv(csv_content : String)
rows = CSV.new(csv_content, headers: true)
subscriptions = Array(String).new
rows.each do |row|
# Channel ID is the first column in the csv export we can't use the header
# name, because I believe the header name is localized depending on the
# language the user has set on their account
channel_id = row[0].strip
if channel_id.empty?
next
end
subscriptions << channel_id
end
subscriptions
end