xuyue/OnlineJudge
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
yuetsh
2024-06-25 13:35:30 +00:00
parent f9cccdab34
commit fe97bae2a5
2 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
announcement/views/oj.py
View File

@@ -36,13 +36,15 @@ class MessageAPI(APIView):
except Message.DoesNotExist: except Message.DoesNotExist:
return self.error("Message does not exist") return self.error("Message does not exist")
else: else:
messages = Message.objects.filter(recipient=request.user) messages = Message.objects.select_related("recipient","sender", "submission").filter(recipient=request.user)
return self.success(self.paginate_data(request, messages, MessageListSerializer)) return self.success(self.paginate_data(request, messages, MessageListSerializer))
@validate_serializer(CreateMessageSerializer) @validate_serializer(CreateMessageSerializer)
@super_admin_required @super_admin_required
def post(self, request): def post(self, request):
data = request.data data = request.data
if data["recipient"] == request.user.id:
return self.error("Can not send a message to youself")
try: try:
recipient = User.objects.get(id=data["recipient"], is_disabled=False) recipient = User.objects.get(id=data["recipient"], is_disabled=False)
except User.DoesNotExist: except User.DoesNotExist:

4
oj/settings.py
View File

@@ -20,8 +20,8 @@ if production_env:
else: else:
from .dev_settings import * from .dev_settings import *
with open(os.path.join(DATA_DIR, "config", "secret.key"), "r") as f: # with open(os.path.join(DATA_DIR, "config", "secret.key"), "r") as f:
SECRET_KEY = f.read() # SECRET_KEY = f.read()
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__))) BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))