diff --git a/import_problem b/import_problem --- a/import_problem +++ b/import_problem @@ -38,14 +38,22 @@ SCRIPT_DIR = File.dirname(__FILE__) # print usage -if ARGV.length < 3 +if ARGV.length < 4 puts < 1, :mem_limit => 16} process_options(options) # start working puts "creating directories" - system("mkdir #{problem}") system("mkdir #{problem}/script") system("mkdir #{problem}/test_cases") @@ -85,6 +92,15 @@ cfg_file.puts all_test_cfg.result cfg_file.close + +# copy check script +if File.exists?(SCRIPT_DIR + "/templates/check.#{check_script}") + check_script_fname = SCRIPT_DIR + "/templates/check.#{check_script}" +else + check_script_fname = check_script +end +system("cp #{check_script_fname} #{problem}/script/check") + # generating test_request directory puts "generating test_request template" FileUtils.mkdir_p("test_request/#{problem}/script")