xlsimport/afterfix.rb

32 lines
968 B
Ruby

#!/usr/bin/ruby
require 'yaml'
require 'mysql2'
require 'mysql2-cs-bind'
require './sql.rb'
config = YAML::load_file(File.join(__dir__, 'config.yaml'))
host = config["db_host"]
username = config["db_user"]
password = config["db_password"]
database = config["db_database"]
$database = Mysql2::Client.new(
:host => host,
:username => username,
:password => password,
:database => database
)
$debug = true
brands = []
dbres = $database.query("select brand from parts group by brand having count(*) < 2")
dbres.each do| result_row |
brand_id = result_row["brand"].to_i
brands.push(brand_id)
end
$database.xquery("UPDATE parts SET brand = NULL WHERE brand IN (?)", [brands])
dbres = $database.query("SELECT brands.id, COUNT(*) FROM brands JOIN parts ON parts.brand = brands.id GROUP BY brands.id HAVING count(*) < 2;")
dbres.each do| result_row |
brand_id = result_row["brands.id"].to_i
$database.query("DELETE FROM brands WHERE id = '#{brand_id}';")
end