dawarich/app/services/photoprism/request_photos.rb

81 lines
1.8 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
class Photoprism::RequestPhotos
attr_reader :user, :photoprism_api_base_url, :photoprism_api_key, :start_date, :end_date
def initialize(user, start_date: '1970-01-01', end_date: nil)
@user = user
@photoprism_api_base_url = "#{user.settings['photoprism_url']}/api/v1/photos"
@photoprism_api_key = user.settings['photoprism_api_key']
@start_date = start_date
@end_date = end_date
end
def call
raise ArgumentError, 'Photoprism API key is missing' if photoprism_api_key.blank?
raise ArgumentError, 'Photoprism URL is missing' if user.settings['photoprism_url'].blank?
data = retrieve_photoprism_data
time_framed_data(data)
end
private
def retrieve_photoprism_data
data = []
offset = 0
while offset < 1_000_000
response = HTTParty.get(
photoprism_api_base_url,
headers: headers,
query: request_params(offset)
)
break if response.code != 200
photoprism_data = JSON.parse(response.body)
data << photoprism_data
break if photoprism_data.empty?
offset += 1000
end
data
end
def headers
{
'Authorization' => "Bearer #{photoprism_api_key}",
'accept' => 'application/json'
}
end
def request_params(offset = 0)
params = {
q: '',
public: true,
quality: 3,
after: start_date,
offset: offset,
count: 1000
}
params.delete(:offset) if offset.zero?
params[:before] = end_date if end_date.present?
params
end
def time_framed_data(data)
data.flatten.select do |photo|
taken_at = DateTime.parse(photo['TakenAtLocal'])
end_date ||= Time.current
taken_at.between?(start_date.to_datetime, end_date.to_datetime)
end
end
end