Discourse/discourse
Discourse/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-09-05 08:59:27 +08:00
Code Packages Activity Actions 11

FEATURE: my_bb import supports avatars (#17617)

Browse source
This commit is contained in:
Jay Pfaffman 2022-07-25 09:22:25 -04:00 • committed by GitHub
parent 9103081eb7
commit 7ab5dcf82f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

20
script/import_scripts/mybb.rb
View file

@ -11,6 +11,7 @@ export DB_NAME="mybb"
export DB_PW="" export DB_PW=""
export DB_USER="root" export DB_USER="root"
export TABLE_PREFIX="mybb_" export TABLE_PREFIX="mybb_"
export UPLOADS_DIR="/data/youruploads"
export BASE="" # export BASE="" #
=end =end
@ -23,6 +24,7 @@ class ImportScripts::MyBB < ImportScripts::Base
DB_PW ||= ENV['DB_PW'] || "" DB_PW ||= ENV['DB_PW'] || ""
DB_USER ||= ENV['DB_USER'] || "root" DB_USER ||= ENV['DB_USER'] || "root"
TABLE_PREFIX ||= ENV['TABLE_PREFIX'] || "mybb_" TABLE_PREFIX ||= ENV['TABLE_PREFIX'] || "mybb_"
UPLOADS_DIR ||= ENV['UPLOADS'] || '/data/limelightgaming/uploads'
BATCH_SIZE = 1000 BATCH_SIZE = 1000
BASE = "" BASE = ""
QUIET = true QUIET = true
@ -59,7 +61,7 @@ class ImportScripts::MyBB < ImportScripts::Base
batches(BATCH_SIZE) do |offset| batches(BATCH_SIZE) do |offset|
results = mysql_query( results = mysql_query(
"SELECT uid id, email email, username, regdate, g.title `group` "SELECT uid id, email email, username, regdate, g.title `group`, avatar
FROM #{TABLE_PREFIX}users u FROM #{TABLE_PREFIX}users u
JOIN #{TABLE_PREFIX}usergroups g ON g.gid = u.usergroup JOIN #{TABLE_PREFIX}usergroups g ON g.gid = u.usergroup
WHERE g.title != 'Banned' WHERE g.title != 'Banned'
@ -72,12 +74,26 @@ class ImportScripts::MyBB < ImportScripts::Base
next if all_records_exist? :users, results.map { |u| u["id"].to_i } next if all_records_exist? :users, results.map { |u| u["id"].to_i }
create_users(results, total: total_count, offset: offset) do |user| create_users(results, total: total_count, offset: offset) do |user|
avatar_url = user['avatar'].match(/^http/) ? user['avatar'].gsub(/\?.*/, "") : nil
{ id: user['id'], { id: user['id'],
email: user['email'], email: user['email'],
username: user['username'], username: user['username'],
created_at: Time.zone.at(user['regdate']), created_at: Time.zone.at(user['regdate']),
moderator: user['group'] == 'Super Moderators', moderator: user['group'] == 'Super Moderators',
admin: user['group'] == 'Administrators' } admin: user['group'] == 'Administrators' ,
avatar_url: avatar_url,
post_create_action: proc do |newuser|
if !user["avatar"].blank?
avatar = user["avatar"].gsub(/\?.*/, "")
if avatar.match(/^http.*/)
UserAvatar.import_url_for_user(avatar, newuser)
else
filename = File.join(UPLOADS_DIR, avatar)
@uploader.create_avatar(newuser, filename) if File.exists?(filename)
end
end
end
}
end end
end end
end end