summaryrefslogtreecommitdiff
path: root/cli/app/controllers/crud_controller.py
blob: c1bde8d509acff46546b740a72b24384911392fd (plain)
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
from flask import request, jsonify
from flask_classful import FlaskView, route
from werkzeug.datastructures import MultiDict

from app.sql.common import db, Session
from app.server.helpers import parse_search_args, parse_sort_args

class CrudView(FlaskView):
  # to subclass CrudView, specify the model + form:
  # model = Collection
  # form = CollectionForm
  index_all = True
  excluded_methods = ['on_index', 'on_show', 'on_create', 'on_update', 'on_destroy']

  # implement these methods:
  def where(self, query, args):
    return query
  def on_index(self, session, data):
    return data
  def on_show(self, session, data):
    return data
  def on_create(self, session, form, item):
    pass
  def on_update(self, session, form, item):
    pass
  def on_destroy(self, session, item):
    pass

  def index(self):
    """
    List all {model}s
    """
    session = Session()
    if self.index_all:
      items = session.query(self.model).all()
    else:
      offset, limit = parse_search_args(request.args)
      sort, order, order_by, order_by_id = parse_sort_args(request.args, self.model)
      query = session.query(self.model)
      query = self.where(query, request.args)
      if order_by_id is not None:
        query = query.order_by(order_by, order_by_id)
      else:
        query = query.order_by(order_by)
      items = query.offset(offset).limit(limit).all()

    res = self.on_index(session, {
      'status': 'ok',
      'res': [ item.toJSON() for item in items ],
    })
    session.close()
    return jsonify(res)

  def get(self, id: int):
    """
    Fetch a single {model}.
    """
    session = Session()
    item = session.query(self.model).get(id)
    if not item:
      session.close()
      return jsonify({
        'status': 'error',
        'error': 'item not found'
      })
    result = self.on_show(session, {
      'status': 'ok',
      'res': item.toFullJSON() if hasattr(item, 'toFullJSON') else item.toJSON(),
    })
    session.close()
    return jsonify(result)

  def post(self):
    """
    Create a new {model}.

    * JSON params: {jsonparams}
    """
    session = Session()
    raw_form = MultiDict(request.json) if request.json is not None else request.form
    form = self.form(raw_form)
    if form.validate():
      item = self.model()
      form.populate_obj(item)
      self.on_create(session, raw_form, item)
      session.add(item)
      session.commit()
      res = {
        'status': 'ok',
        'res': item.toJSON(),
      }
    else:
      res = {
        'error': 'error',
        'errors': form.errors,
      }
    session.close()
    return jsonify(res)

  def put(self, id: int):
    """
    Update a {model}.

    * JSON params: {jsonparams}
    """
    session = Session()
    item = session.query(self.model).get(id)
    if item:
      raw_form = MultiDict(request.json) if request.json is not None else request.form
      form = self.form(raw_form, obj=item)
      # print(item.toJSON())
      if form.validate():
        form.populate_obj(item)
        self.on_update(session, raw_form, item)
        session.add(item)
        session.commit()
        res = {
          'status': 'ok',
          'res': item.toJSON(),
        }
      else:
        res = {
          'status': 'error',
          'error': form.errors,
        }
    else:
      res = {
        'status': 'error',
        'error': 'not found',
      }
    session.close()
    return jsonify(res)

  def delete(self, id: int):
    """
    Delete a {model}.
    """
    session = Session()
    item = session.query(self.model).get(id)
    if item:
      self.on_destroy(session, item)
      session.delete(item)
      session.commit()
      res = {
        'status': 'ok',
      }
    else:
      res = {
        'status': 'error',
        'error': 'not found',
      }
    session.close()
    return jsonify(res)