XML в CSV рубин - PullRequest
       9

XML в CSV рубин

0 голосов
/ 14 января 2019

У меня есть несколько файлов примеров XML, и я хотел бы преобразовать их в CSV, но будет несколько разных атрибутов / узлов для разных файлов XML, поэтому я не хочу жестко кодировать разные атрибуты. Я хотел бы, чтобы выходные данные отображали заголовок столбца в качестве первой строки, а затем каждый узел / запись отображался вертикально, как традиционная таблица столбцов и строк. Вот один из примеров XML:

<?xml version="1.0" encoding="UTF-8"?>
<sd:root xmlns:wd="urn:com.sample/bsvc" sd:version="v31.0">
  <sd:Put_Job_Profile_Request sd:Add_Only="0">
    <sd:Job_Profile_Data>
      <sd:Job_Code>30000</sd:Job_Code>
      <sd:Effective_Date>1900-01-01</sd:Effective_Date>
      <sd:Job_Profile_Basic_Data>
        <sd:Job_Title>Chief Executive Officer</sd:Job_Title>
      </sd:Job_Profile_Basic_Data>
    </sd:Job_Profile_Data>
  </sd:Put_Job_Profile_Request>
  <sd:Put_Job_Profile_Request sd:Add_Only="0">
    <sd:Job_Profile_Data>
      <sd:Job_Code>30100</sd:Job_Code>
      <sd:Effective_Date>1900-01-01</sd:Effective_Date>
      <sd:Job_Profile_Basic_Data>
        <sd:Job_Title>Administrator Job Profile</sd:Job_Title>
      </sd:Job_Profile_Basic_Data>
    </sd:Job_Profile_Data>
  </sd:Put_Job_Profile_Request>
  <sd:Put_Job_Profile_Request sd:Add_Only="0">
    <sd:Job_Profile_Data>
      <sd:Job_Code>30200</sd:Job_Code>
      <sd:Effective_Date>1900-01-01</sd:Effective_Date>
      <sd:Job_Profile_Basic_Data>
        <sd:Inactive>0</sd:Inactive>
        <sd:Job_Title>Facilities &amp; Grounds Maintenance Attendant</sd:Job_Title>
        <sd:Include_Job_Code_in_Name>0</sd:Include_Job_Code_in_Name>
        <sd:Job_Profile_Private_Title>Maintenance Job Title</sd:Job_Profile_Private_Title>
        <sd:Job_Profile_Summary>Maintain cleanliness of the campus building throughout the day and fulfill special requests as needed.</sd:Job_Profile_Summary>
        <sd:Job_Description>&lt;p>Job Description&lt;b> rich text!&lt;/b>&lt;/p></sd:Job_Description>
        <sd:Additional_Job_Description>&lt;p>&lt;b>&lt;i>&lt;span class="emphasis-2">&lt;u>Additional&lt;/u>&lt;/span>&lt;/i>&lt;/b> Job Description&lt;b> rich text!&lt;/b>&lt;/p></sd:Additional_Job_Description>
        <sd:Work_Shift_Required>0</sd:Work_Shift_Required>
        <sd:Public_Job>1</sd:Public_Job>    
      </sd:Job_Profile_Basic_Data>
    </sd:Job_Profile_Data>
  </sd:Put_Job_Profile_Request>
  <sd:Put_Job_Profile_Request sd:Add_Only="0">
    <sd:Job_Profile_Data>
      <sd:Job_Code>30300</sd:Job_Code>
      <sd:Effective_Date>1900-01-01</sd:Effective_Date>
      <sd:Job_Profile_Basic_Data>
        <sd:Inactive>0</sd:Inactive>
        <sd:Job_Title>Sample_Job_Title</sd:Job_Title>
        <sd:Include_Job_Code_in_Name>0</sd:Include_Job_Code_in_Name>
        <sd:Job_Profile_Summary>Sample Job Profile Summary</sd:Job_Profile_Summary>
        <sd:Job_Description>Sample Job Description</sd:Job_Description>
        <sd:Additional_Job_Description>Sample Additional Job Description</sd:Additional_Job_Description>
        <sd:Work_Shift_Required>1</sd:Work_Shift_Required>
      </sd:Job_Profile_Basic_Data>
    </sd:Job_Profile_Data>
  </sd:Put_Job_Profile_Request>
</sd:root>

Код, который я использовал, но вышел неправильно:

require 'csv'
require 'nokogiri'

file = File.read('jobProfile.xml')
doc = Nokogiri::XML(file)
a = []

CSV.open('xmloutput.csv', 'wb') do |csv|
  csv << doc.at('.').search('*').map(&:name)
  doc.search('.').each do |x|
    csv << x.search('*').map(&:text)
  end
end

Каждый набор записей с заголовками столбцов и данными, установленными горизонтально. Но я хотел бы перебрать данные и сохранить одну строку заголовков столбцов. Я не уверен, как это сделать без жесткого кодирования каждого из атрибутов: / Пожалуйста, помогите, так как я все еще новичок в программировании, и я пытался найти решение в течение недели: (

снимок экрана с выводом csv

1 Ответ

0 голосов
/ 15 января 2019

Сначала необходимо создать массив хэшей и извлечь ключи в виде заголовков, а затем поместить значения в правый столбец, все узлы сведены в столбцы, игнорируя корневой ключ и ключи записи.

Примерно так

require 'nokogiri'
require 'set'

file    = File.read('jobProfile.xml')
doc     = Nokogiri::XML(file)
record  = {}
keys    = Set.new
records = []
csv     = ""

doc.traverse do |node| 
  value = node.text.gsub(/\n +/, '')
  if node.name
    if node.name != "text" # skip these nodes
      if value.length > 0 # skip empty nodes
        key = node.name.gsub(/sd:/,'').to_sym
        # if a new and not empty record, add to our records collection
        if key == :Job_Profile_Data && !record.empty?
          records << record
          record = {}
        elsif key[/Job_Profile|^root$|^document$/]
          # neglect these keys
        else
          key = node.name.gsub(/sd:/,'').to_sym
          # in case our value is html instead of text
          record[key] = Nokogiri::HTML.parse(value).text
          # add to our key set only if not allready in the set
          keys << key
        end
      end
    end
  end
end

# build our csv
File.open('./xmloutput.csv', 'w') do |file|
  file.puts %Q{"#{keys.to_a.join('","')}"}
  records.each do |record|
    keys.each do |key|
      file.write %Q{"#{record[key]}",}
    end
    file.write "\n"
  end
end

Что дает в нашем CSV-файле следующее

"Job_Code","Effective_Date","Job_Title","Inactive","Include_Job_Code_in_Name","Job_Description","Additional_Job_Description","Work_Shift_Required","Public_Job"
"30000","1900-01-01","Chief Executive Officer","","","","","","",
"30100","1900-01-01","Administrator Job Profile","","","","","","",
"30200","1900-01-01","Facilities & Grounds Maintenance Attendant","0","0","Job Description rich text!","Additional Job Description rich text!","0","1",
"30300","1900-01-01","Sample_Job_Title","0","0","Sample Job Description","Sample Additional Job Description","1","",
...