1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
# flask_shop/role/view.py
from ast import Delete
from flask import request
from flask_shop.role import role,role_api
from flask_shop import models,db
from flask_restful import Resource
from flask_shop.utils.message import to_dict_msg

class Role(Resource):
    # 获取数据
    def get(self):
        role_list = []
        try:
            roles = models.Role.query.all()
            role_list = [r.to_dict() for r in roles]
            return to_dict_msg(200, role_list, '获取角色列表成功')
        except Exception as e:
            return to_dict_msg(20000)
    
    # 增加数据
    def post(self):
        name = request.form.get('name')
        desc = request.form.get('desc')
        try:
            if name:
                role = models.Role(name= name, desc= desc)
                db.session.add(role)
                db.session.commit()
                return to_dict_msg(200, msg='增加角色成功！！！！')
        except Exception as e:
            print(e)
            return to_dict_msg(20000)
    
    # 删除数据
    def delete(self):
        try:
            id = int(request.form.get('id'))
            r = models.Role.query.get(id)
            if r:
                db.session.delete(r)
                db.session.commit()
                return to_dict_msg(200,msg="删除角色成功！！！！！")
        except Exception:
            return to_dict_msg(20000)
    
    # 修改数据
    def put(self):
        try:
            id = int(request.form.get('id'))
            name = request.form.get('name').strip() if request.form.get('name') else ''
            desc = request.form.get('desc').strip() if request.form.get('desc') else ''
            if name:
                r = models.Role.query.get(id)
                if r:
                    r.name = name
                    r.desc = desc
                    db.session.commit()
                    return to_dict_msg(200, msg= '修改角色信息成功！！！')
                return to_dict_msg(10020)
        except Exception as e:
            return to_dict_msg(20000)


role_api.add_resource(Role, '/role')