1 files changed, 9 insertions, 0 deletions
diff --git a/app/models/user_custom_attribute.rb b/app/models/user_custom_attribute.rb
index 425f2cc062b..15d50071bf6 100644
--- a/app/models/user_custom_attribute.rb
+++ b/app/models/user_custom_attribute.rb
@@ -15,6 +15,7 @@ class UserCustomAttribute < ApplicationRecord
   UNBLOCKED_BY = 'unblocked_by'
   ARKOSE_RISK_BAND = 'arkose_risk_band'
   AUTO_BANNED_BY_ABUSE_REPORT_ID = 'auto_banned_by_abuse_report_id'
+  AUTO_BANNED_BY_SPAM_LOG_ID = 'auto_banned_by_spam_log_id'
   ALLOW_POSSIBLE_SPAM = 'allow_possible_spam'
   IDENTITY_VERIFICATION_PHONE_EXEMPT = 'identity_verification_phone_exempt'
 
@@ -45,6 +46,14 @@ class UserCustomAttribute < ApplicationRecord
       upsert_custom_attributes([custom_attribute])
     end
 
+    def set_banned_by_spam_log(spam_log)
+      return unless spam_log
+
+      custom_attribute = { user_id: spam_log.user_id, key: AUTO_BANNED_BY_SPAM_LOG_ID, value: spam_log.id }
+
+      upsert_custom_attributes([custom_attribute])
+    end
+
     private
 
     def blocked_users