Fix bug in category endpoint & stats command

2024-06-29 02:14:37 -05:00 · 2024-06-29 02:14:37 -05:00 · fa36795dde
commit fa36795dde
parent f84eef5e5f
2 changed files with 2 additions and 2 deletions
--- a/app/controllers/yiff_rest/api_v2_controller.rb
+++ b/app/controllers/yiff_rest/api_v2_controller.rb
@ -76,7 +76,7 @@ module YiffRest
      render(json: {
        success: true,
        data:    {
-          **APIImage.categories.find { |c| c[:db] == category },
+          **APIImage.categories.find { |c| c.db == category },
          count: count,
        },
      })
--- a/app/models/api_key.rb
+++ b/app/models/api_key.rb
@ -325,7 +325,7 @@ class APIKey < ApplicationRecord
  include WebhookMethods

  def self.stats(ip: nil, key: nil)
-    categories = APIImage.categories.pluck(:db)
+    categories = APIImage.categories.map(&:db)

    keys = [
      "yiffy2:stats:images:total",