cafe-grader-web

Commit r274:25b80877ebce public

parent | child
Description:
fixed import error (thanks K.Siththa)
Commit status:
[Not Reviewed]
References:
default
Diff options:
Raw Diff | Patch Diff | Download Diff | Ignore 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

r274:25b80877ebce - - 1 file changed: 2 inserted, 0 deleted

Show file before | Show file after | Raw diff | Download diff | Ignore whitespace Increase context Show commentsHide comments
@@ -1,151 +1,153
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 37
38 38 return true
39 39 end
40 40
41 41 protected
42 42
43 43 def self.long_ext(filename)
44 44 i = filename.index('.')
45 45 len = filename.length
46 46 return filename.slice(i..len)
47 47 end
48 48
49 49 def extract(tempfile)
50 50 testdata_filename = save_testdata_file(tempfile)
51 51 ext = TestdataImporter.long_ext(tempfile.original_filename)
52 52
53 53 extract_dir = File.join(GraderScript.raw_dir, @problem.name)
54 54 begin
55 55 Dir.mkdir extract_dir
56 56 rescue Errno::EEXIST
57 57 end
58 58
59 59 if ext=='.tar.gz' or ext=='.tgz'
60 60 cmd = "tar -zxvf #{testdata_filename} -C #{extract_dir}"
61 61 elsif ext=='.tar'
62 62 cmd = "tar -xvf #{testdata_filename} -C #{extract_dir}"
63 63 elsif ext=='.zip'
64 64 cmd = "unzip -o #{testdata_filename} -d #{extract_dir}"
65 65 else
66 66 return nil
67 67 end
68 68
69 69 system(cmd)
70 70
71 71 files = Dir["#{extract_dir}/**/*1*.in"]
72 72 return nil if files.length==0
73 73
74 74 File.delete(testdata_filename)
75 75
76 76 return File.dirname(files[0])
77 77 end
78 78
79 79 def save_testdata_file(tempfile)
80 80 ext = TestdataImporter.long_ext(tempfile.original_filename)
81 81 testdata_filename = File.join(Dir.tmpdir,"#{@problem.name}#{ext}")
82 82
83 83 return nil if tempfile==""
84 84
85 85 if tempfile.instance_of?(Tempfile)
86 86 tempfile.close
87 87 FileUtils.move(tempfile.path,testdata_filename)
88 88 else
89 89 File.open(testdata_filename, "wb") do |f|
90 90 f.write(tempfile.read)
91 91 end
92 92 end
93 93
94 94 return testdata_filename
95 95 end
96 96
97 97 def import_test_pairs(dirname)
98 98 test_num = 1
99 99 while FileTest.exists? "#{dirname}/#{test_num}.in"
100 100 in_filename = "#{dirname}/#{test_num}.in"
101 101 sol_filename = "#{dirname}/#{test_num}.sol"
102 102
103 103 break if not FileTest.exists? sol_filename
104 104
105 105 test_pair = TestPair.new(:input => open(in_filename).read,
106 106 :solution => open(sol_filename).read,
107 107 :problem => @problem)
108 108 break if not test_pair.save
109 109
110 110 test_num += 1
111 111 end
112 112 return test_num > 1
113 113 end
114 114
115 115 def import_problem_description(dirname)
116 116 html_files = Dir["#{dirname}/*.html"]
117 117 markdown_files = Dir["#{dirname}/*.md"] + Dir["#{dirname}/*.markdown"]
118 118 if (html_files.length != 0) or (markdown_files.length != 0)
119 119 description = @problem.description || Description.new
120 120
121 121 if html_files.length != 0
122 122 filename = html_files[0]
123 123 description.markdowned = false
124 124 else
125 125 filename = markdown_files[0]
126 126 description.markdowned = true
127 127 end
128 128
129 129 description.body = open(filename).read
130 130 description.save
131 131 @problem.description = description
132 132 @problem.save
133 133 return "\nProblem description imported from #{filename}."
134 134 else
135 135 return ''
136 136 end
137 137 end
138 138
139 139 def import_problem_pdf(dirname)
140 140 pdf_files = Dir["#{dirname}/*.pdf"]
141 141 if pdf_files.length != 0
142 142 filename = pdf_files[0]
143 143 out_filename = "#{Problem.download_file_basedir}/#{@problem.name}.pdf"
144 144 File.rename(filename, out_filename)
145 145 @problem.description_filename = "#{@problem.name}.pdf"
146 146 @problem.save
147 147 return "\nProblem pdf imported from #{filename}."
148 + else
149 + return ""
148 150 end
149 151 end
150 152
151 153 end
You need to be logged in to leave comments. Login now