[Enhance] Add API about songs

- Add some API endpoints, including creating, changing, deleting song info.

latest version/api/api_auth.py

@@ -48,13 +48,14 @@ def role_required(request, powers=[]):
     return decorator
 
 
-def request_json_handle(request, required_keys=[], optional_keys=[]):
+def request_json_handle(request, required_keys: list = [], optional_keys: list = [], must_change: bool = False):
     '''
         提取post参数，返回dict，写成了修饰器\ 
         parameters: \ 
         `request`: `Request` - 当前请求\ 
         `required_keys`: `list` - 必须的参数\ 
-        `optional_keys`: `list` - 可选的参数
+        `optional_keys`: `list` - 可选的参数\ 
+        `must_change`: `bool` - 当全都是可选参数时，是否必须有至少一项修改
     '''
 
     def decorator(view):
@@ -67,8 +68,11 @@ def request_json_handle(request, required_keys=[], optional_keys=[]):
             else:
                 if request.method == 'GET' and 'query' in request.args:
                     # 处理axios没法GET传data的问题
-                    json_data = loads(
-                        b64decode(request.args['query']).decode())
+                    try:
+                        json_data = loads(
+                            b64decode(request.args['query']).decode())
+                    except:
+                        raise PostError(api_error_code=-105)
                 else:
                     json_data = {}
 
@@ -81,6 +85,9 @@ def request_json_handle(request, required_keys=[], optional_keys=[]):
                 if key in json_data:
                     data[key] = json_data[key]
 
+            if must_change and not data:
+                return error_return(PostError('No change', api_error_code=-100))
+
             return view(data, *args, **kwargs)
 
         return wrapped_view

latest version/api/api_code.py

@@ -9,6 +9,7 @@ CODE_MSG = {
     -1: 'See status code',  # 基础错误
     -2: 'No data',
     -3: 'No data or user',  # 不确定是无数据还是无用户
+    -4: 'Data exist',
     -100: 'Invalid post data',  # 1xx数据错误
     -101: 'Invalid data type',
     -102: 'Invalid query parameter',

latest version/api/songs.py

@@ -1,4 +1,4 @@
-from core.error import NoData, InputError
+from core.error import DataExist, NoData, InputError
 from core.rank import RankList
 from core.song import Song
 from core.sql import Connect, Query, Sql
@@ -21,6 +21,39 @@ def songs_song_get(user, song_id):
         return success_return(s.to_dict())
 
 
+@bp.route('/<string:song_id>', methods=['PUT'])
+@role_required(request, ['change'])
+@request_json_handle(request, optional_keys=['name', 'charts'], must_change=True)
+@api_try
+def songs_song_put(data, user, song_id):
+    '''修改歌曲信息'''
+    with Connect() as c:
+        s = Song(c, song_id).select()
+        if 'name' in data:
+            s.name = str(data['name'])
+        if 'charts' in data:
+            for i in data['charts']:
+                if 'difficulty' in i and 'chart_const' in i:
+                    s.charts[i['difficulty']].defnum = round(
+                        i['chart_const'] * 10)
+
+        s.update()
+        return success_return(s.to_dict())
+
+
+@bp.route('/<string:song_id>', methods=['DELETE'])
+@role_required(request, ['change'])
+@api_try
+def songs_song_delete(user, song_id):
+    '''删除歌曲信息'''
+    with Connect() as c:
+        s = Song(c, song_id)
+        if not s.select_exists():
+            raise NoData(f'No such song: `{song_id}`')
+        s.delete()
+        return success_return()
+
+
 @bp.route('', methods=['GET'])
 @role_required(request, ['select', 'select_song_info'])
 @request_json_handle(request, optional_keys=Constant.QUERY_KEYS)
@@ -43,6 +76,20 @@ def songs_get(data, user):
         return success_return([x.to_dict() for x in r])
 
 
+@bp.route('', methods=['POST'])
+@role_required(request, ['change'])
+@request_json_handle(request, ['song_id', 'charts'], ['name'])
+@api_try
+def songs_post(data, user):
+    '''添加歌曲信息'''
+    with Connect() as c:
+        s = Song(c).from_dict(data)
+        if s.select_exists():
+            raise DataExist(f'Song `{s.song_id}` already exists')
+        s.insert()
+        return success_return(s.to_dict())
+
+
 @bp.route('/<string:song_id>/<int:difficulty>/rank', methods=['GET'])
 @role_required(request, ['select', 'select_song_rank', 'select_song_rank_top'])
 @request_json_handle(request, optional_keys=['limit'])

latest version/api/users.py

@@ -78,7 +78,7 @@ def users_user_get(user, user_id):
 
 @bp.route('/<int:user_id>', methods=['PUT'])
 @role_required(request, ['change'])
-@request_json_handle(request, optional_keys=['name', 'password', 'user_code', 'ticket', 'email'])
+@request_json_handle(request, optional_keys=['name', 'password', 'user_code', 'ticket', 'email'], must_change=True)
 @api_try
 def users_user_put(data, user, user_id):
     '''修改一个用户'''
@@ -103,8 +103,7 @@ def users_user_put(data, user, user_id):
                 raise InputError('Ticket must be int')
             u.ticket = data['ticket']
             r['ticket'] = u.ticket
-        if r:
-            u.update_columns(d=r)
+        u.update_columns(d=r)
         return success_return(r)