github-mirror/discourse
2
0
Fork 0
mirror of https://github.com/discourse/discourse.git synced 2025-05-22 19:52:43 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fluxbb.rb: optional table prefix env variable added

Browse Source
This commit is contained in:
Vinoth Kannan
2016-10-07 10:29:56 +05:30
parent 470529d4c8
commit ae8088ab99
1 changed files with 16 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
script/import_scripts/fluxbb.rb
View File

@ -5,25 +5,29 @@ require File.expand_path(File.dirname(__FILE__) + "/base.rb")
# Before running this script, paste these lines into your shell, # Before running this script, paste these lines into your shell,
# then use arrow keys to edit the values # then use arrow keys to edit the values
=begin =begin
export FLUXBB_USER="root" export FLUXBB_HOST="localhost"
export FLUXBB_DB="fluxbb" export FLUXBB_DB="fluxbb"
export FLUXBB_USER="root"
export FLUXBB_PW="" export FLUXBB_PW=""
export FLUXBB_PREFIX=""
=end =end
# Call it like this: # Call it like this:
# RAILS_ENV=production bundle exec ruby script/import_scripts/fluxbb.rb # RAILS_ENV=production bundle exec ruby script/import_scripts/fluxbb.rb
class ImportScripts::FluxBB < ImportScripts::Base class ImportScripts::FluxBB < ImportScripts::Base
FLUXBB_HOST ||= ENV['FLUXBB_HOST'] || "localhost"
FLUXBB_DB ||= ENV['FLUXBB_DB'] || "fluxbb" FLUXBB_DB ||= ENV['FLUXBB_DB'] || "fluxbb"
BATCH_SIZE ||= 1000 BATCH_SIZE ||= 1000
FLUXBB_PW ||= ENV['FLUXBB_PW'] || ""
FLUXBB_USER ||= ENV['FLUXBB_USER'] || "root" FLUXBB_USER ||= ENV['FLUXBB_USER'] || "root"
FLUXBB_PW ||= ENV['FLUXBB_PW'] || ""
FLUXBB_PREFIX ||= ENV['FLUXBB_PREFIX'] || ""
def initialize def initialize
super super
@client = Mysql2::Client.new( @client = Mysql2::Client.new(
host: "localhost", host: FLUXBB_HOST,
username: FLUXBB_USER, username: FLUXBB_USER,
password: FLUXBB_PW, password: FLUXBB_PW,
database: FLUXBB_DB database: FLUXBB_DB
@ -43,7 +47,7 @@ class ImportScripts::FluxBB < ImportScripts::Base
results = mysql_query( results = mysql_query(
"SELECT g_id id, g_title name, g_user_title title "SELECT g_id id, g_title name, g_user_title title
FROM groups") FROM #{FLUXBB_PREFIX}groups")
create_groups(results) do |group| create_groups(results) do |group|
{ id: group['id'], { id: group['id'],
@ -62,7 +66,7 @@ class ImportScripts::FluxBB < ImportScripts::Base
"SELECT id, username, realname name, url website, email email, registered created_at, "SELECT id, username, realname name, url website, email email, registered created_at,
registration_ip registration_ip_address, last_visit last_visit_time, registration_ip registration_ip_address, last_visit last_visit_time,
last_email_sent last_emailed_at, location, group_id last_email_sent last_emailed_at, location, group_id
FROM users FROM #{FLUXBB_PREFIX}users
LIMIT #{BATCH_SIZE} LIMIT #{BATCH_SIZE}
OFFSET #{offset};") OFFSET #{offset};")
@ -103,7 +107,7 @@ class ImportScripts::FluxBB < ImportScripts::Base
categories = mysql_query(" categories = mysql_query("
SELECT id, cat_name name, disp_position position SELECT id, cat_name name, disp_position position
FROM categories FROM #{FLUXBB_PREFIX}categories
ORDER BY id ASC ORDER BY id ASC
").to_a ").to_a
@ -118,7 +122,7 @@ class ImportScripts::FluxBB < ImportScripts::Base
children_categories = mysql_query(" children_categories = mysql_query("
SELECT id, forum_name name, forum_desc description, disp_position position, cat_id parent_category_id SELECT id, forum_name name, forum_desc description, disp_position position, cat_id parent_category_id
FROM forums FROM #{FLUXBB_PREFIX}forums
ORDER BY id ORDER BY id
").to_a ").to_a
@ -135,7 +139,7 @@ class ImportScripts::FluxBB < ImportScripts::Base
def import_posts def import_posts
puts "", "creating topics and posts" puts "", "creating topics and posts"
total_count = mysql_query("SELECT count(*) count from posts").first["count"] total_count = mysql_query("SELECT count(*) count from #{FLUXBB_PREFIX}posts").first["count"]
batches(BATCH_SIZE) do |offset| batches(BATCH_SIZE) do |offset|
results = mysql_query(" results = mysql_query("
@ -147,8 +151,8 @@ class ImportScripts::FluxBB < ImportScripts::Base
p.poster_id user_id, p.poster_id user_id,
p.message raw, p.message raw,
p.posted created_at p.posted created_at
FROM posts p, FROM #{FLUXBB_PREFIX}posts p,
topics t #{FLUXBB_PREFIX}topics t
WHERE p.topic_id = t.id WHERE p.topic_id = t.id
ORDER BY p.posted ORDER BY p.posted
LIMIT #{BATCH_SIZE} LIMIT #{BATCH_SIZE}
@ -190,11 +194,11 @@ class ImportScripts::FluxBB < ImportScripts::Base
banned = 0 banned = 0
failed = 0 failed = 0
total = mysql_query("SELECT count(*) count FROM bans").first['count'] total = mysql_query("SELECT count(*) count FROM #{FLUXBB_PREFIX}bans").first['count']
system_user = Discourse.system_user system_user = Discourse.system_user
mysql_query("SELECT username, email FROM bans").each do |b| mysql_query("SELECT username, email FROM #{FLUXBB_PREFIX}bans").each do |b|
user = User.find_by_email(b['email']) user = User.find_by_email(b['email'])
if user if user
user.suspended_at = Time.now user.suspended_at = Time.now