Contents

require 'atcoder_greedy'
require 'atcoder_greedy/command'
require 'atcoder_greedy/lib/contest'

module AtcoderGreedy
  class Command < Thor
    desc 'create [CONTESTURL] [OPTION]', 'create contest templates for [CONTESTURL]'
    option :no_templates, type: :boolean, default: false, aliases: "\--nt"
    option :no_input, type: :boolean, default: false, aliases: "\--ni"
    option :select_problem, aliases: '-p', default: nil, desc: 'create only select problem'
    option :select_directory, aliases: '-d', default: '', desc: 'select target directory'
    option :select_language, aliases: '-l', default: '', desc: 'select language'
    option :select_template, aliases: '-t', default: '', desc: 'select generate template'

    def create(contest_url)
      user_options = {
          no: {input: false, template: false},
          problems: [],
          directory: options[:select_directory],
          language: options[:select_language],
          template: options[:select_template]
      }

      user_options[:no][:input] = true if options[:no_input]
      user_options[:no][:template] = true if options[:no_templates]
      user_options[:problems] = options[:select_problem].split unless options[:select_problem].nil?

      contest = Contest.new(contest_url, user_options)
      # TODO: contest_infoが存在したときの処理
      File.open("#{contest.dir}/.contest_info.yml", 'w') do |f|
        info = {
            name: contest.name,
            url: contest.url,
            task: {}
        }
        contest.problems.each do |p|
          info[:task][:"#{p[:name]}"] = {
              id: p[:task_id]
          }
        end

        f.puts info.to_yaml
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
atcoder_greedy-0.5.0	lib/atcoder_greedy/command/create.rb