32 lines
762 B
Ruby
32 lines
762 B
Ruby
require "yaml"
|
|
require 'erb'
|
|
|
|
problem = YAML::load_file(ARGV[0] + "/desc.yml")
|
|
output = ERB.new(IO.read('template.rt'))
|
|
puts output.result()
|
|
|
|
if(ARGV[1] == 'table')
|
|
require 'terminal-table'
|
|
files = Dir[ARGV[0]+"/**/*/desc.yml"]
|
|
files.sort
|
|
entry = []
|
|
files.each do |file|
|
|
yaml = YAML::load_file(file)
|
|
nested = file.split('/')
|
|
count = 0
|
|
if( yaml['solution'] and yaml['solutions'])
|
|
yaml['solutions'].each do |sol|
|
|
count += 1
|
|
end
|
|
end
|
|
entry << {'title'=>yaml['title'],'code'=>nested[-2].to_s,'solved'=>count}
|
|
end
|
|
|
|
rows = []
|
|
entry.each do |e|
|
|
row = [e['code'],e['title'],e['solved']]
|
|
rows << row
|
|
end
|
|
table = Terminal::Table.new(:headings => ['Code','Title','Solved'],:rows => rows)
|
|
puts table
|
|
end |