cafe-grader-web

Commit r345:d3662f88d75c public

parent | child
Description:
let testdata importer set the full score
Commit status:
[Not Reviewed]
References:
default
Diff options:
Raw Diff | Patch Diff | Download Diff | Show whitespace | Increase context
Comments:
0 Commit comments 0 Inline Comments
Unresolved TODOs:
There are no unresolved TODOs
Side by Side Unified
Expand All Files Collapse All Files Wide Mode Diff
Add another comment
Browse Files

r345:d3662f88d75c - - 1 file changed: 11 inserted, 0 deleted

Show file before | Show file after | Raw diff | Download diff | Show whitespace Increase context Show commentsHide comments
@@ -1,172 +1,183
1 1 require 'tmpdir'
2 2
3 3 class TestdataImporter
4 4
5 5 attr :log_msg
6 6
7 7 def initialize(problem)
8 8 @problem = problem
9 9 end
10 10
11 11 def import_from_file(tempfile,
12 12 time_limit,
13 13 memory_limit,
14 14 import_to_db=false)
15 15
16 16 dirname = extract(tempfile)
17 17 return false if not dirname
18 18 if not import_to_db
19 19 @log_msg = GraderScript.call_import_problem(@problem.name,
20 20 dirname,
21 21 time_limit,
22 22 memory_limit)
23 23 else
24 24 # Import test data to test pairs.
25 25
26 26 @problem.test_pairs.clear
27 27 if import_test_pairs(dirname)
28 28 test_pair_count = TestPair.count :conditions => "problem_id = #{@problem.id}"
29 29 @log_msg = "Importing test pair successful. (#{test_pair_count} test pairs imported)"
30 30 else
31 31 @log_msg = "Importing test pair failed. (0 test pairs imported)"
32 32 end
33 33 end
34 34
35 35 @log_msg << import_problem_description(dirname)
36 36 @log_msg << import_problem_pdf(dirname)
37 + @log_msg << import_full_score(dirname)
37 38
38 39 return true
39 40 end
40 41
41 42 protected
42 43
43 44 def self.long_ext(filename)
44 45 i = filename.index('.')
45 46 len = filename.length
46 47 return filename.slice(i..len)
47 48 end
48 49
49 50 def extract(tempfile)
50 51 testdata_filename = save_testdata_file(tempfile)
51 52 ext = TestdataImporter.long_ext(tempfile.original_filename)
52 53
53 54 extract_dir = File.join(GraderScript.raw_dir, @problem.name)
54 55 if File.exists? extract_dir
55 56 backup_count = 0
56 57 begin
57 58 backup_count += 1
58 59 backup_dirname = "#{extract_dir}.backup.#{backup_count}"
59 60 end while File.exists? backup_dirname
60 61 File.rename(extract_dir, backup_dirname)
61 62 end
62 63 Dir.mkdir extract_dir
63 64
64 65 if ext=='.tar.gz' or ext=='.tgz'
65 66 cmd = "tar -zxvf #{testdata_filename} -C #{extract_dir}"
66 67 elsif ext=='.tar'
67 68 cmd = "tar -xvf #{testdata_filename} -C #{extract_dir}"
68 69 elsif ext=='.zip'
69 70 cmd = "unzip -o #{testdata_filename} -d #{extract_dir}"
70 71 else
71 72 return nil
72 73 end
73 74
74 75 system(cmd)
75 76
76 77 files = Dir["#{extract_dir}/**/*1*.in"]
77 78 return nil if files.length==0
78 79
79 80 File.delete(testdata_filename)
80 81
81 82 return File.dirname(files[0])
82 83 end
83 84
84 85 def save_testdata_file(tempfile)
85 86 ext = TestdataImporter.long_ext(tempfile.original_filename)
86 87 testdata_filename = File.join(Dir.tmpdir,"#{@problem.name}#{ext}")
87 88
88 89 return nil if tempfile==""
89 90
90 91 if tempfile.instance_of?(Tempfile)
91 92 tempfile.close
92 93 FileUtils.move(tempfile.path,testdata_filename)
93 94 else
94 95 File.open(testdata_filename, "wb") do |f|
95 96 f.write(tempfile.read)
96 97 end
97 98 end
98 99
99 100 return testdata_filename
100 101 end
101 102
102 103 def import_test_pairs(dirname)
103 104 test_num = 1
104 105 while FileTest.exists? "#{dirname}/#{test_num}.in"
105 106 in_filename = "#{dirname}/#{test_num}.in"
106 107 sol_filename = "#{dirname}/#{test_num}.sol"
107 108
108 109 break if not FileTest.exists? sol_filename
109 110
110 111 test_pair = TestPair.new(:input => open(in_filename).read,
111 112 :solution => open(sol_filename).read,
112 113 :problem => @problem)
113 114 break if not test_pair.save
114 115
115 116 test_num += 1
116 117 end
117 118 return test_num > 1
118 119 end
119 120
120 121 def import_problem_description(dirname)
121 122 html_files = Dir["#{dirname}/*.html"]
122 123 markdown_files = Dir["#{dirname}/*.md"] + Dir["#{dirname}/*.markdown"]
123 124 if (html_files.length != 0) or (markdown_files.length != 0)
124 125 description = @problem.description || Description.new
125 126
126 127 if html_files.length != 0
127 128 filename = html_files[0]
128 129 description.markdowned = false
129 130 else
130 131 filename = markdown_files[0]
131 132 description.markdowned = true
132 133 end
133 134
134 135 description.body = open(filename).read
135 136 description.save
136 137 @problem.description = description
137 138 @problem.save
138 139 return "\nProblem description imported from #{filename}."
139 140 else
140 141 return ''
141 142 end
142 143 end
143 144
144 145 def import_problem_pdf(dirname)
145 146 pdf_files = Dir["#{dirname}/*.pdf"]
146 147 puts "CHECKING... #{dirname}"
147 148 if pdf_files.length != 0
148 149 puts "HAS PDF FILE"
149 150 filename = pdf_files[0]
150 151
151 152 @problem.save if not @problem.id
152 153 out_dirname = "#{Problem.download_file_basedir}/#{@problem.id}"
153 154 if not FileTest.exists? out_dirname
154 155 Dir.mkdir out_dirname
155 156 end
156 157
157 158 out_filename = "#{out_dirname}/#{@problem.name}.pdf"
158 159
159 160 if FileTest.exists? out_filename
160 161 File.delete out_filename
161 162 end
162 163
163 164 File.rename(filename, out_filename)
164 165 @problem.description_filename = "#{@problem.name}.pdf"
165 166 @problem.save
166 167 return "\nProblem pdf imported from #{filename}."
167 168 else
168 169 return ""
169 170 end
170 171 end
171 172
173 + #just set the full score to the total number of test case
174 + #it is not perfect but works on most normal use case
175 + def import_full_score(dirname)
176 + in_file = Dir["#{dirname}/*.in"]
177 + full_score =in_file.length * 10
178 + @problem.full_score = full_score
179 + @problem.save
180 + return "\nFull score is set to #{full_score}."
172 181 end
182 +
183 + end
You need to be logged in to leave comments. Login now