aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-17 22:14:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Properly load schemas specified as strings (#943)

Browse Source 
* Properly load schemas specified by their key string
* Add test for UserSchema 
* Prevent users without teams from interacting with challenges if the CTF is in Team Mode
This commit is contained in:
Kevin Chung
2019-04-08 01:47:26 -04:00
committed by GitHub
parent 7c60c697ee
commit c0a32a836b
16 changed files with 111 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
CTFd/schemas/notifications.py
View File

@@ -3,6 +3,7 @@ from marshmallow import fields, post_load
from marshmallow import validate, ValidationError
from marshmallow_sqlalchemy import field_for
from CTFd.models import ma, Notifications
from CTFd.utils import string_types
class NotificationSchema(ma.ModelSchema):
@@ -13,9 +14,9 @@ class NotificationSchema(ma.ModelSchema):
def __init__(self, view=None, *args, **kwargs):
if view:
if type(view) == str:
if isinstance(view, string_types):
kwargs['only'] = self.views[view]
elif type(view) == list:
elif isinstance(view, list):
kwargs['only'] = view
super(NotificationSchema, self).__init__(*args, **kwargs)