mirror of
https://github.com/iv-org/invidious
synced 2024-11-24 12:03:26 +01:00
Clean up Video class
This commit is contained in:
parent
aa0864f87d
commit
f0b33b2975
@ -5,12 +5,6 @@ class Video
|
|||||||
end
|
end
|
||||||
end
|
end
|
||||||
|
|
||||||
module XMLConverter
|
|
||||||
def self.from_rs(rs)
|
|
||||||
XML.parse_html(rs.read(String))
|
|
||||||
end
|
|
||||||
end
|
|
||||||
|
|
||||||
def initialize(id, info, updated, title, views, likes, dislikes, wilson_score, published, description)
|
def initialize(id, info, updated, title, views, likes, dislikes, wilson_score, published, description)
|
||||||
@id = id
|
@id = id
|
||||||
@info = info
|
@info = info
|
||||||
@ -105,10 +99,8 @@ def get_client(pool)
|
|||||||
end
|
end
|
||||||
|
|
||||||
def fetch_video(id, client)
|
def fetch_video(id, client)
|
||||||
begin
|
|
||||||
info = client.get("/get_video_info?video_id=#{id}&el=detailpage&ps=default&eurl=&gl=US&hl=en").body
|
info = client.get("/get_video_info?video_id=#{id}&el=detailpage&ps=default&eurl=&gl=US&hl=en").body
|
||||||
html = client.get("/watch?v=#{id}").body
|
html = client.get("/watch?v=#{id}").body
|
||||||
end
|
|
||||||
|
|
||||||
html = XML.parse_html(html)
|
html = XML.parse_html(html)
|
||||||
info = HTTP::Params.parse(info)
|
info = HTTP::Params.parse(info)
|
||||||
@ -175,22 +167,21 @@ def get_video(id, client, db, refresh = true)
|
|||||||
# If record was last updated over an hour ago, refresh (expire param in response lasts for 6 hours)
|
# If record was last updated over an hour ago, refresh (expire param in response lasts for 6 hours)
|
||||||
if refresh && Time.now - video.updated > 1.hours
|
if refresh && Time.now - video.updated > 1.hours
|
||||||
video = fetch_video(id, client)
|
video = fetch_video(id, client)
|
||||||
db.exec("UPDATE videos SET info = $2, updated = $3,\
|
db.exec("DELETE FROM videos * WHERE id = $1", id)
|
||||||
title = $4, views = $5, likes = $6, dislikes = $7, wilson_score = $8,\
|
args = arg_array(video.to_a)
|
||||||
published = $9, description = $10 WHERE id = $1", video.to_a)
|
db.exec("INSERT INTO videos VALUES (#{args})", video.to_a)
|
||||||
end
|
end
|
||||||
else
|
else
|
||||||
video = fetch_video(id, client)
|
video = fetch_video(id, client)
|
||||||
db.exec("INSERT INTO videos VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)", video.to_a)
|
args = arg_array(video.to_a)
|
||||||
|
db.exec("INSERT INTO videos VALUES (#{args})", video.to_a)
|
||||||
end
|
end
|
||||||
|
|
||||||
return video
|
return video
|
||||||
end
|
end
|
||||||
|
|
||||||
def search(query, client)
|
def search(query, client)
|
||||||
begin
|
|
||||||
html = client.get("https://www.youtube.com/results?q=#{query}&sp=EgIQAVAU").body
|
html = client.get("https://www.youtube.com/results?q=#{query}&sp=EgIQAVAU").body
|
||||||
end
|
|
||||||
|
|
||||||
html = XML.parse_html(html)
|
html = XML.parse_html(html)
|
||||||
|
|
||||||
@ -330,3 +321,12 @@ end
|
|||||||
def number_with_separator(number)
|
def number_with_separator(number)
|
||||||
number.to_s.reverse.gsub(/(\d{3})(?=\d)/, "\\1,").reverse
|
number.to_s.reverse.gsub(/(\d{3})(?=\d)/, "\\1,").reverse
|
||||||
end
|
end
|
||||||
|
|
||||||
|
def arg_array(array)
|
||||||
|
args = [] of String
|
||||||
|
(1..array.size).each { |i| args << "($#{i})," }
|
||||||
|
args = args.join("")
|
||||||
|
args = args.chomp(",")
|
||||||
|
|
||||||
|
return args
|
||||||
|
end
|
||||||
|
@ -150,11 +150,8 @@ spawn do
|
|||||||
top = rank_videos(PG_DB, 40)
|
top = rank_videos(PG_DB, 40)
|
||||||
client = get_client(youtube_pool)
|
client = get_client(youtube_pool)
|
||||||
|
|
||||||
args = [] of String
|
|
||||||
if top.size > 0
|
if top.size > 0
|
||||||
(1..top.size).each { |i| args << "($#{i})," }
|
args = arg_array(top)
|
||||||
args = args.join("")
|
|
||||||
args = args.chomp(",")
|
|
||||||
else
|
else
|
||||||
next
|
next
|
||||||
end
|
end
|
||||||
|
Loading…
Reference in New Issue
Block a user