def regen_all(cls, context): """重算所有缓存,只有通过命令行提权后才能调用""" User.test_permission(context) for obj in models.Submission.objects.all(): try: obj.group = User.get(context, obj.user).group obj.save() except NotFound: pass for obj in models.FlagClear.objects.all(): try: user = User.get(context, obj.user) obj.group = user.group challenge = Challenge.get(context, obj.challenge) if obj.flag not in range(len(challenge.flags)): raise NotFound obj.save() except NotFound: obj.delete() for challenge in Challenge.get_all(context): cls._regen_challenge_clear(challenge) models.ChallengeFirst.objects.all().delete() models.FlagFirst.objects.all().delete() cls._refill_first() cls._regen_score()
def update(self, **kwargs): User.test_permission(self._context, 'challenge.full') old = self._json_all self._update(**kwargs) new = self._json_all for subscriber in self.subscribers: subscriber(old, new)
def get_log(cls, context, *, after=None, before=None, limit=None, challenge=None, group=None, match=None): if after is not None and before is not None: raise WrongArguments() User.test_permission(context, 'submission.full', 'submission.view') queryset = models.Submission.objects.order_by('-pk') if after is not None: queryset = queryset.filter(pk__lt=after) if before is not None: queryset = queryset.filter(pk__gt=before).reverse() if challenge is not None: queryset = queryset.filter(challenge=challenge) if group is not None: queryset = queryset.filter(group=group) if match is not None: queryset = queryset.filter(flagclear__isnull=not match) queryset = list( queryset.values( 'pk', 'user', 'challenge', 'text', 'time', flag=models.models.F('flagclear__flag'), )[slice(limit)]) if before is not None: queryset.reverse() return queryset
def get_all(cls, context): queryset = models.Trigger.objects.order_by('time') try: User.test_permission(context, 'trigger.full') except PermissionRequired: queryset = queryset.filter(time__lte=context.time) return [cls(context, obj) for obj in queryset]
def delete(self): User.test_permission(self._context, 'challenge.full') old = self._json_all self._obj.expr_set.all().delete() self._obj.delete() self._obj = None for subscriber in self.subscribers: subscriber(old, None)
def create(cls, context, **kwargs): User.test_permission(context, 'trigger.full') self = cls(context, models.Trigger()) self._update(**kwargs) new = self._json_all for subscriber in self.subscribers: subscriber(None, new) return self
def get_board(cls, context, *, limit=None, category=None, group=None): if group in User.no_board_groups: User.test_permission(context, 'submission.full', 'submission.view') return list( cls._filter_group(models.Score.objects, group).filter(category=category).order_by( '-score', 'time').values('user', 'score', 'time')[slice(limit)])
def get_enabled(cls, context): User.test_authenticated(context) Terms.test_agreed_enabled(context) User.test_profile(context) try: User.test_permission(context, 'challenge.full', 'challenge.view') except PermissionRequired: Trigger.test_can_view_challenges(context) queryset = models.Challenge.objects.filter(enabled=True) return [cls(context, obj) for obj in queryset]
def create(cls, context, **kwargs): User.test_permission(context, 'challenge.full') self = cls(context, models.Challenge()) flags = kwargs.pop('flags') self._update(**kwargs) self._update(flags=flags) new = self._json_all for subscriber in self.subscribers: subscriber(None, new) return self
def flags(self): flags = json.loads(self._obj.flags) try: User.test_permission(self._context, 'challenge.full', 'challenge.view') return flags except PermissionRequired: return [{ 'name': flag['name'], 'score': flag['score'], } for flag in flags]
def get_violations(cls, context, *, challenge=None, group=None): User.test_permission(context, 'submission.full', 'submission.view') queryset = models.FlagViolation.objects if challenge is not None: queryset = queryset.filter(submission__challenge=challenge) if group is not None: queryset = queryset.filter(submission__group=group) return list(queryset.values( 'violation_flag', 'violation_user', user=models.models.F('submission__user'), challenge=models.models.F('submission__challenge'), time=models.models.F('submission__time'), ))
def get_first(cls, context, *, group=None): if group in User.no_board_groups: User.test_permission(context, 'submission.full', 'submission.view') return { 'challenges': list( models.ChallengeFirst.objects.filter(group=group).values( 'challenge', 'user', 'time')), 'flags': list( models.FlagFirst.objects.filter(group=group).values( 'challenge', 'flag', 'user', 'time')), }
def get(cls, context, pk): queryset = models.Challenge.objects.all() try: User.test_permission(context, 'challenge.full', 'challenge.view') except PermissionRequired: User.test_authenticated(context) Terms.test_agreed_enabled(context) User.test_profile(context) Trigger.test_can_view_challenges(context) queryset = queryset.filter(enabled=True) try: return cls(context, queryset.get(pk=pk)) except models.Challenge.DoesNotExist: raise NotFound('题目不存在')
def regen_all(cls, context): """重算所有缓存,只有通过命令行提权后才能调用""" User.test_permission(context) models.User.objects.all().delete() models.Expr.objects.all().delete() models.ExprFlag.objects.all().delete() for challenge in cls.get_all(context): for i, flag in enumerate(challenge.flags): if flag['type'] != 'expr': continue challenge._obj.expr_set.create(flag_index=i, expr=flag['flag']) for user in User.get_all(context): if not user.token: continue cls._add_user(user.pk) models.User.objects.create(user=user.pk)
def get_all(cls, context): User.test_permission(context, 'challenge.full', 'challenge.view') return [cls(context, obj) for obj in models.Challenge.objects.all()]
def submit(cls, context, user, challenge, text): if context.user.pk != user: User.test_permission(context) if len(text) > 200: raise WrongFormat('Flag 不应超过 200 个字符') user = User.get(context, user) challenge = Challenge.get(context, challenge) try: latest = (models.Submission.objects.filter( user=user.pk, challenge=challenge.pk).latest('time')) except models.Submission.DoesNotExist: pass else: if latest.time + timedelta(seconds=10) > context.time: raise SlowDown('提交过于频繁,请 10 秒后再试') obj = models.Submission.objects.create( user=user.pk, group=user.group, challenge=challenge.pk, text=text, time=context.time, ) matches, violations = challenge.check_flag_with_violations(text) queryset = models.FlagClear.objects.filter(user=user.pk, challenge=challenge.pk) flags = {i.flag for i in queryset} match_flags = {i['index'] for i in matches} for flag in match_flags - flags: models.FlagClear.objects.create( submission=obj, user=user.pk, group=user.group, challenge=challenge.pk, flag=flag, time=context.time, ) if user.group not in User.no_score_groups: models.FlagFirst.objects.get_or_create( challenge=challenge.pk, flag=flag, group=None, defaults={ 'user': user.pk, 'time': context.time }, ) models.FlagFirst.objects.get_or_create( challenge=challenge.pk, flag=flag, group=user.group, defaults={ 'user': user.pk, 'time': context.time }, ) for f, u in violations: models.FlagViolation.objects.create( submission=obj, violation_flag=f['index'], violation_user=u, ) if match_flags - flags: if (flags | match_flags).issuperset(range(len(challenge.flags))): models.ChallengeClear.objects.create( user=user.pk, group=user.group, challenge=challenge.pk, time=context.time, ) if user.group not in User.no_score_groups: models.ChallengeFirst.objects.get_or_create( challenge=challenge.pk, group=None, defaults={ 'user': user.pk, 'time': context.time }, ) models.ChallengeFirst.objects.get_or_create( challenge=challenge.pk, group=user.group, defaults={ 'user': user.pk, 'time': context.time }, ) score = sum(i['score'] for i in matches if i['index'] not in flags) cls._add_score(user.pk, user.group, context.time, score, challenge.category) return matches
def delete(self): User.test_permission(self._context, 'trigger.full') self._obj.delete() self._obj = None
def get(cls, context, pk): User.test_permission(context, 'trigger.full') try: return cls(context, models.Trigger.objects.get(pk=pk)) except models.Trigger.DoesNotExist: raise NotFound()