Add the solution to pb_entry when not in contest
[gruntmaster-data.git] / lib / Gruntmaster / Data.pm
CommitLineData
4ed3f8e7 1use utf8;
bbf8209c 2package Gruntmaster::Data;
4ed3f8e7
MG
3
4# Created by DBIx::Class::Schema::Loader
5# DO NOT MODIFY THE FIRST PART OF THIS FILE
6
7use strict;
bbf8209c 8use warnings;
014ee8a6 9
4ed3f8e7
MG
10use base 'DBIx::Class::Schema';
11
12__PACKAGE__->load_namespaces;
014ee8a6 13
f7386aab 14
4ed3f8e7
MG
15# Created by DBIx::Class::Schema::Loader v0.07039 @ 2014-03-05 13:11:39
16# DO NOT MODIFY THIS OR ANYTHING ABOVE! md5sum:dAEmtAexvUaNXLgYz2rNEg
17
347ea1e4 18our $VERSION = '5999.000_011';
4ed3f8e7
MG
19
20use Lingua::EN::Inflect qw/PL_N/;
91203bd5 21use JSON::MaybeXS qw/decode_json/;
15a021b7 22use List::Util qw/sum/;
3473324b 23use PerlX::Maybe qw/maybe/;
4ed3f8e7 24use Sub::Name qw/subname/;
014ee8a6 25
de625c9b
MG
26use constant PROBLEM_PUBLIC_COLUMNS => [qw/id author writer level name owner private statement timeout olimit value/];
27use constant USER_PUBLIC_COLUMNS => [qw/id admin name town university level/];
28use constant JOBS_PER_PAGE => 10;
29
014ee8a6 30sub dynsub{
fb6a4e3d 31 our ($name, $sub) = @_;
dcd72991 32 no strict 'refs'; ## no critic (Strict)
fb6a4e3d 33 *$name = subname $name => $sub
014ee8a6
MG
34}
35
36BEGIN {
4ed3f8e7
MG
37 for my $rs (qw/contest contest_problem job open problem user/) {
38 my $rsname = ucfirst $rs;
dcd72991 39 $rsname =~ s/_([a-z])/\u$1/gs;
4ed3f8e7
MG
40 dynsub PL_N($rs) => sub { $_[0]->resultset($rsname) };
41 dynsub $rs => sub { $_[0]->resultset($rsname)->find($_[1]) };
014ee8a6
MG
42 }
43}
44
15a021b7
MG
45sub calc_score{
46 my ($mxscore, $time, $tries, $totaltime) = @_;
47 my $score = $mxscore;
48 $time = 0 if $time < 0;
49 $time = 300 if $time > $totaltime;
50 $score = ($totaltime - $time) / $totaltime * $score;
51 $score -= $tries / 10 * $mxscore;
52 $score = $mxscore * 3 / 10 if $score < $mxscore * 3 / 10;
53 int $score + 0.5
54}
55
56sub standings {
57 my ($self, $ct) = @_;
58 $ct &&= $self->contest($ct);
59
60 my @problems = map { $_->problem } $self->contest_problems->search({contest => $ct && $ct->id}, {qw/join problem order_by problem.level/});
61 my (%scores, %tries);
62 for my $job ($self->jobs->search({contest => $ct && $ct->id}, {order_by => 'id'})) {
63 if ($ct) {
64 my $open = $self->opens->find($ct->id, $job->problem->id, $job->owner->id);
65 my $time = $job->date - ($open ? $open->time : $ct->start);
66 next if $time < 0;
aaa9eb7d 67 my $value = $job->problem->value;
15a021b7 68 my $factor = $job->result ? 0 : 1;
dcd72991 69 $factor = $1 / 100 if $job->result_text =~ /^(\d+ )/s;
15a021b7
MG
70 $scores{$job->owner->id}{$job->problem->id} = int ($factor * calc_score ($value, $time, $tries{$job->owner->id}{$job->problem->id}++, $ct->stop - $ct->start));
71 } else {
dcd72991 72 no warnings 'numeric'; ## no critic (ProhibitNoWarnings)
15a021b7
MG
73 $scores{$job->owner->id}{$job->problem->id} = 0 + $job->result_text || ($job->result ? 0 : 100)
74 }
75 }
76
dcd72991 77 my @st = sort { $b->{score} <=> $a->{score} or $a->{user}->id cmp $b->{user}->id} map { ## no critic (ProhibitReverseSortBlock)
15a021b7
MG
78 my $user = $_;
79 +{
80 user => $self->user($user),
35596ea8 81 score => sum (values %{$scores{$user}}),
15a021b7
MG
82 scores => [map { $scores{$user}{$_->id} // '-'} @problems],
83 problems => $ct,
84 }
85 } keys %scores;
86
87 $st[0]->{rank} = 1;
88 $st[$_]->{rank} = $st[$_ - 1]->{rank} + ($st[$_]->{score} < $st[$_ - 1]->{score}) for 1 .. $#st;
89 @st
90}
91
de625c9b
MG
92sub user_list {
93 my $rs = $_[0]->users->search(undef, {order_by => 'name', columns => USER_PUBLIC_COLUMNS});
dcd72991 94 [ map { { $_->get_columns } } $rs->all ]
de625c9b
MG
95}
96
97sub user_entry {
98 my ($self, $id) = @_;
99 +{ $self->users->find($id, {columns => USER_PUBLIC_COLUMNS})->get_columns }
100}
101
102sub problem_list {
103 my ($self, %args) = @_;
104 my $rs = $self->problems->search(undef, {order_by => 'me.name', columns => PROBLEM_PUBLIC_COLUMNS, prefetch => 'owner'});
105 $rs = $rs->search({-or => ['contest_problems.contest' => undef, 'contest.stop' => {'<=', time}], 'me.private' => 0}, {join => {'contest_problems' => 'contest'}, distinct => 1}) unless $args{contest};
106 $rs = $rs->search({'contest_problems.contest' => $args{contest}}, {join => 'contest_problems'}) if $args{contest};
107 $rs = $rs->search({'me.owner' => $args{owner}}) if $args{owner};
108 my %params;
109 $params{contest} = $args{contest} if $args{contest};
110 for ($rs->all) {
111 $params{$_->level} //= [];
35596ea8 112 push @{$params{$_->level}}, {$_->get_columns, owner_name => $_->owner->name} ;
de625c9b
MG
113 }
114 \%params
115}
116
117sub problem_entry {
118 my ($self, $id, $contest, $user) = @_;
de625c9b 119 my $running = $contest && $self->contest($contest)->is_running;
8bd66733
MG
120 my $columns = PROBLEM_PUBLIC_COLUMNS;
121 push @$columns, 'solution' unless $running;
122 my $pb = $self->problems->find($id, {columns => $columns, prefetch => 'owner'});
dcd72991 123 eval { ## no critic (RequireCheckingReturnValueOfEval)
de625c9b
MG
124 $self->opens->create({
125 contest => $contest,
126 problem => $id,
127 owner => $user,
128 time => time,
129 })
130 } if $running;
2431260b
MG
131 $contest &&= $self->contest($contest);
132 +{ $pb->get_columns, owner_name => $pb->owner->name, cansubmit => $contest ? $running : 1, $running ? (contest_start => $contest->start, contest_stop => $contest->stop) : () }
de625c9b
MG
133}
134
135sub contest_list {
136 my ($self, %args) = @_;
137 my $rs = $self->contests->search(undef, {order_by => {-desc => 'start'}, prefetch => 'owner'});
138 $rs = $rs->search({owner => $args{owner}}) if $args{owner};
139 my %params;
140 for ($rs->all) {
141 my $state = $_->is_pending ? 'pending' : $_->is_running ? 'running' : 'finished';
142 $params{$state} //= [];
35596ea8 143 push @{$params{$state}}, { $_->get_columns, started => !$_->is_pending, owner_name => $_->owner->name };
de625c9b
MG
144 }
145 \%params
146}
147
148sub contest_entry {
149 my ($self, $id) = @_;
150 my $ct = $self->contest($id);
151 +{ $ct->get_columns, started => !$ct->is_pending, owner_name => $ct->owner->name }
152}
153
154sub job_list {
155 my ($self, %args) = @_;
156 $args{page} //= 1;
3473324b 157 my $rs = $self->jobs->search(undef, {order_by => {-desc => 'me.id'}, prefetch => ['problem', 'owner'], rows => JOBS_PER_PAGE, page => $args{page}});
526b9e80
MG
158 $rs = $rs->search({'me.owner' => $args{owner}}) if $args{owner};
159 $rs = $rs->search({contest => $args{contest}}) if $args{contest};
160 $rs = $rs->search({problem => $args{problem}}) if $args{problem};
3473324b
MG
161 return {
162 log => [map {
163 my %params = $_->get_columns;
164 $params{owner_name} = $_->owner->name;
165 $params{problem_name} = $_->problem->name;
166 $params{results} &&= decode_json $params{results};
167 $params{size} = length $params{source};
168 delete $params{source};
169 \%params
170 } $rs->all],
171 current_page => $rs->pager->current_page,
172 maybe previous_page => $rs->pager->previous_page,
173 maybe next_page => $rs->pager->next_page,
174 maybe last_page => $rs->pager->last_page,
175 }
de625c9b
MG
176}
177
178sub job_entry {
179 my ($self, $id) = @_;
180 my $job = $self->jobs->find($id, {prefetch => ['problem', 'owner']});
181 my %params = $job->get_columns;
182 $params{owner_name} = $job->owner->name;
183 $params{problem_name} = $job->problem->name;
184 $params{results} &&= decode_json $params{results};
185 $params{size} = length $params{source};
186 delete $params{source};
187 \%params
188}
189
f7386aab 1901;
4a8747ef
MG
191
192__END__
193
194=encoding utf-8
195
196=head1 NAME
197
198Gruntmaster::Data - Gruntmaster 6000 Online Judge -- database interface and tools
199
200=head1 SYNOPSIS
201
202 my $db = Gruntmaster::Data->connect('dbi:Pg:');
cbb36c78
MG
203
204 my $problem = $db->problem('my_problem');
205 $problem->update({timeout => 2.5}); # Set time limit to 2.5 seconds
206 $problem->rerun; # And rerun all jobs for this problem
207
208 # ...
209
210 my $contest = $db->contests->create({ # Create a new contest
211 id => 'my_contest',
212 name => 'My Awesome Contest',
213 start => time + 100,
214 end => time + 1900,
215 });
216 $db->contest_problems->create({ # Add a problem to the contest
217 contest => 'my_contest',
218 problem => 'my_problem',
219 });
220
221 say 'The contest has not started yet' if $contest->is_pending;
222
223 # ...
224
225 my @jobs = $db->jobs->search({contest => 'my_contest', owner => 'MGV'})->all;
226 $_->rerun for @jobs; # Rerun all jobs sent by MGV in my_contest
4a8747ef
MG
227
228=head1 DESCRIPTION
229
cbb36c78
MG
230Gruntmaster::Data is the interface to the Gruntmaster 6000 database. Read the L<DBIx::Class> documentation for usage information.
231
232In addition to the typical DBIx::Class::Schema methods, this module contains several convenience methods:
233
234=over
235
236=item contests
237
238Equivalent to C<< $schema->resultset('Contest') >>
239
240=item contest_problems
241
242Equivalent to C<< $schema->resultset('ContestProblem') >>
243
244=item jobs
245
246Equivalent to C<< $schema->resultset('Job') >>
247
248=item problems
249
250Equivalent to C<< $schema->resultset('Problem') >>
251
252=item users
253
254Equivalent to C<< $schema->resultset('User') >>
255
256=item contest($id)
257
258Equivalent to C<< $schema->resultset('Contest')->find($id) >>
259
260=item job($id)
261
262Equivalent to C<< $schema->resultset('Job')->find($id) >>
263
264=item problem($id)
265
266Equivalent to C<< $schema->resultset('Problem')->find($id) >>
267
268=item user($id)
269
270Equivalent to C<< $schema->resultset('User')->find($id) >>
271
de625c9b
MG
272=item user_list
273
274Returns a list of users as an arrayref containing hashrefs.
275
276=item user_entry($id)
277
278Returns a hashref with information about the user $id.
279
280=item problem_list([%args])
281
282Returns a list of problems grouped by level. A hashref with levels as keys.
283
284Takes the following arguments:
285
286=over
287
288=item owner
289
290Only show problems owned by this user
291
292=item contest
293
294Only show problems in this contest
295
296=back
297
298=item problem_entry($id, [$contest, $user])
299
300Returns a hashref with information about the problem $id. If $contest and $user are present, problem open data is updated.
301
302=item contest_list([%args])
303
304Returns a list of contests grouped by state. A hashref with the following keys:
305
306=over
307
308=item pending
309
310An arrayref of hashrefs representing pending contests
311
312=item running
313
314An arrayref of hashrefs representing running contests
315
316=item finished
317
318An arrayref of hashrefs representing finished contests
319
320=back
321
322Takes the following arguments:
323
324=over
325
326=item owner
327
328Only show contests owned by this user.
329
330=back
331
332=item contest_entry($id)
333
334Returns a hashref with information about the contest $id.
335
336=item job_list([%args])
337
338Returns a list of jobs as an arrayref containing hashrefs. Takes the following arguments:
339
340=over
341
342=item owner
343
344Only show jobs submitted by this user.
345
346=item contest
347
348Only show jobs submitted in this contest.
349
350=item problem
351
352Only show jobs submitted for this problem.
353
354=item page
355
356Show this page of results. Defaults to 1. Pages have 10 entries, and the first page has the most recent jobs.
357
358=back
359
360=item job_entry($id)
361
362Returns a hashref with information about the job $id.
363
cbb36c78 364=back
4a8747ef
MG
365
366=head1 AUTHOR
367
368Marius Gavrilescu E<lt>marius@ieval.roE<gt>
369
370=head1 COPYRIGHT AND LICENSE
371
372Copyright (C) 2014 by Marius Gavrilescu
373
9d2e740e
MG
374This library is free software; you can redistribute it and/or modify
375it under the same terms as Perl itself, either Perl version 5.18.1 or,
376at your option, any later version of Perl 5 you may have available.
4a8747ef
MG
377
378
379=cut
This page took 0.035014 seconds and 4 git commands to generate.