Bennett/Arcaea-server
1
0
Fork 0
mirror of https://github.com/Lost-MSth/Arcaea-server.git synced 2026-02-12 19:17:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Enhance] Add download error status code

Browse Source
This commit is contained in:
Lost-MSth
2022-10-04 00:28:19 +08:00
parent 8f1c28eb5c
commit 7a614eaa57
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
latest version/core/download.py
View File

@@ -93,7 +93,7 @@ class UserDownload:
x = self.c.fetchone() x = self.c.fetchone()
if not x: if not x:
raise NoAccess('The token `%s` is not valid.' % self.token) raise NoAccess('The token `%s` is not valid.' % self.token, status=403)
self.user = User() self.user = User()
self.user.user_id = x[0] self.user.user_id = x[0]
self.token_time = x[1] self.token_time = x[1]

6
latest version/main.py
View File

@@ -62,8 +62,10 @@ def download(file_path):
x.song_id, x.file_name = file_path.split('/', 1) x.song_id, x.file_name = file_path.split('/', 1)
x.select_for_check() x.select_for_check()
if x.is_limited: if x.is_limited:
raise ArcError('You have reached the download limit.', 903) raise ArcError(
if x.is_valid: 'You have reached the download limit.', 903, status=403)
if not x.is_valid:
raise ArcError('Expired token.', status=403)
x.insert_user_download() x.insert_user_download()
# response = make_response() # response = make_response()
# response.headers['Content-Type'] = 'application/octet-stream' # response.headers['Content-Type'] = 'application/octet-stream'