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
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
|
import logging
import dateutil.parser
from newspipe.bootstrap import db
from datetime import datetime
from collections import defaultdict
from sqlalchemy import or_, func
from werkzeug.exceptions import Forbidden, NotFound
logger = logging.getLogger(__name__)
class AbstractController:
_db_cls = None # reference to the database class
_user_id_key = "user_id"
def __init__(self, user_id=None, ignore_context=False):
"""User id is a right management mechanism that should be used to
filter objects in database on their denormalized "user_id" field
(or "id" field for users).
Should no user_id be provided, the Controller won't apply any filter
allowing for a kind of "super user" mode.
"""
try:
self.user_id = int(user_id)
except TypeError:
self.user_id = user_id
def _to_filters(self, **filters):
"""
Will translate filters to sqlalchemy filter.
This method will also apply user_id restriction if available.
each parameters of the function is treated as an equality unless the
name of the parameter ends with either "__gt", "__lt", "__ge", "__le",
"__ne", "__in" ir "__like".
"""
db_filters = set()
for key, value in filters.items():
if key == "__or__":
db_filters.add(or_(*self._to_filters(**value)))
elif key.endswith("__gt"):
db_filters.add(getattr(self._db_cls, key[:-4]) > value)
elif key.endswith("__lt"):
db_filters.add(getattr(self._db_cls, key[:-4]) < value)
elif key.endswith("__ge"):
db_filters.add(getattr(self._db_cls, key[:-4]) >= value)
elif key.endswith("__le"):
db_filters.add(getattr(self._db_cls, key[:-4]) <= value)
elif key.endswith("__ne"):
db_filters.add(getattr(self._db_cls, key[:-4]) != value)
elif key.endswith("__in"):
db_filters.add(getattr(self._db_cls, key[:-4]).in_(value))
elif key.endswith("__contains"):
db_filters.add(getattr(self._db_cls, key[:-10]).contains(value))
elif key.endswith("__like"):
db_filters.add(getattr(self._db_cls, key[:-6]).like(value))
elif key.endswith("__ilike"):
db_filters.add(getattr(self._db_cls, key[:-7]).ilike(value))
else:
db_filters.add(getattr(self._db_cls, key) == value)
return db_filters
def _get(self, **filters):
""" Will add the current user id if that one is not none (in which case
the decision has been made in the code that the query shouldn't be user
dependent) and the user is not an admin and the filters doesn't already
contains a filter for that user.
"""
if (
self._user_id_key is not None
and self.user_id
and filters.get(self._user_id_key) != self.user_id
):
filters[self._user_id_key] = self.user_id
return self._db_cls.query.filter(*self._to_filters(**filters))
def get(self, **filters):
"""Will return one single objects corresponding to filters"""
obj = self._get(**filters).first()
if obj and not self._has_right_on(obj):
raise Forbidden(
{
"message": "No authorized to access %r (%r)"
% (self._db_cls.__class__.__name__, filters)
}
)
if not obj:
raise NotFound(
{"message": "No %r (%r)" % (self._db_cls.__class__.__name__, filters)}
)
return obj
def create(self, **attrs):
assert attrs, "attributes to update must not be empty"
if self._user_id_key is not None and self._user_id_key not in attrs:
attrs[self._user_id_key] = self.user_id
assert (
self._user_id_key is None
or self._user_id_key in attrs
or self.user_id is None
), "You must provide user_id one way or another"
obj = self._db_cls(**attrs)
db.session.add(obj)
db.session.flush()
db.session.commit()
return obj
def read(self, **filters):
return self._get(**filters)
def update(self, filters, attrs, return_objs=False, commit=True):
assert attrs, "attributes to update must not be empty"
result = self._get(**filters).update(attrs, synchronize_session=False)
if commit:
db.session.flush()
db.session.commit()
if return_objs:
return self._get(**filters)
return result
def delete(self, obj_id):
obj = self.get(id=obj_id)
db.session.delete(obj)
try:
db.session.commit()
except Exception as e:
db.session.rollback()
return obj
def _has_right_on(self, obj):
# user_id == None is like being admin
if self._user_id_key is None:
return True
return (
self.user_id is None
or getattr(obj, self._user_id_key, None) == self.user_id
)
def _count_by(self, elem_to_group_by, filters):
if self.user_id:
filters["user_id"] = self.user_id
return dict(
db.session.query(elem_to_group_by, func.count("id"))
.filter(*self._to_filters(**filters))
.group_by(elem_to_group_by)
.all()
)
@classmethod
def _get_attrs_desc(cls, role, right=None):
result = defaultdict(dict)
if role == "admin":
columns = cls._db_cls.__table__.columns.keys()
else:
assert role in {"base", "api"}, "unknown role %r" % role
assert right in {"read", "write"}, (
"right must be 'read' or 'write' with role %r" % role
)
columns = getattr(cls._db_cls, "fields_%s_%s" % (role, right))()
for column in columns:
result[column] = {}
db_col = getattr(cls._db_cls, column).property.columns[0]
try:
result[column]["type"] = db_col.type.python_type
except NotImplementedError:
if db_col.default:
result[column]["type"] = db_col.default.arg.__class__
if column not in result:
continue
if issubclass(result[column]["type"], datetime):
result[column]["default"] = datetime.utcnow()
result[column]["type"] = lambda x: dateutil.parser.parse(x)
elif db_col.default:
result[column]["default"] = db_col.default.arg
return result
|
