20d54bbb138eebdf62b344c7bec3cc4cf046c5aa
1 # -*- coding: utf-8 -*-
3 # Licensed under the Apache License, Version 2.0 (the "License");
4 # you may not use this file except in compliance with the License.
5 # You may obtain a copy of the License at
7 # http://www.apache.org/licenses/LICENSE-2.0
9 # Unless required by applicable law or agreed to in writing, software
10 # distributed under the License is distributed on an "AS IS" BASIS,
11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
13 # See the License for the specific language governing permissions and
14 # limitations under the License.
17 from uuid
import uuid4
18 from http
import HTTPStatus
20 from osm_common
.dbbase
import deep_update_rfc7396
21 from validation
import validate_input
, ValidationError
, is_valid_uuid
23 __author__
= "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>"
26 class EngineException(Exception):
28 def __init__(self
, message
, http_code
=HTTPStatus
.BAD_REQUEST
):
29 self
.http_code
= http_code
30 Exception.__init
__(self
, message
)
33 def get_iterable(input_var
):
35 Returns an iterable, in case input_var is None it just returns an empty tuple
36 :param input_var: can be a list, tuple or None
37 :return: input_var or () if it is None
45 """utility for compare dot separate versions. Fills with zeros to proper number comparison"""
47 for point
in v
.split("."):
48 filled
.append(point
.zfill(8))
53 # static variables for all instance classes
54 topic
= None # to_override
55 topic_msg
= None # to_override
56 schema_new
= None # to_override
57 schema_edit
= None # to_override
58 multiproject
= True # True if this Topic can be shared by several projects. Then it contains _admin.projects_read
60 # Alternative ID Fields for some Topics
65 "roles_operations": "name"
68 def __init__(self
, db
, fs
, msg
):
72 self
.logger
= logging
.getLogger("nbi.engine")
75 def id_field(topic
, value
):
76 """Returns ID Field for given topic and field value"""
77 if topic
in BaseTopic
.alt_id_field
.keys() and not is_valid_uuid(value
):
78 return BaseTopic
.alt_id_field
[topic
]
83 def _remove_envelop(indata
=None):
88 def _validate_input_new(self
, input, force
=False):
90 Validates input user content for a new entry. It uses jsonschema. Some overrides will use pyangbind
91 :param input: user input content for the new topic
92 :param force: may be used for being more tolerant
93 :return: The same input content, or a changed version of it.
96 validate_input(input, self
.schema_new
)
99 def _validate_input_edit(self
, input, force
=False):
101 Validates input user content for an edition. It uses jsonschema. Some overrides will use pyangbind
102 :param input: user input content for the new topic
103 :param force: may be used for being more tolerant
104 :return: The same input content, or a changed version of it.
107 validate_input(input, self
.schema_edit
)
111 def _get_project_filter(session
):
113 Generates a filter dictionary for querying database, so that only allowed items for this project can be
114 addressed. Only propietary or public can be used. Allowed projects are at _admin.project_read/write. If it is
115 not present or contains ANY mean public.
116 :param session: contains:
117 project_id: project list this session has rights to access. Can be empty, one or several
118 set_project: items created will contain this project list
120 public: True, False or None
121 method: "list", "show", "write", "delete"
123 :return: dictionary with project filter
126 project_filter_n
= []
127 project_filter
= list(session
["project_id"])
129 if session
["method"] not in ("list", "delete"):
131 project_filter
.append("ANY")
132 elif session
["public"] is not None:
133 if session
["public"]:
134 project_filter
.append("ANY")
136 project_filter_n
.append("ANY")
138 if session
.get("PROJECT.ne"):
139 project_filter_n
.append(session
["PROJECT.ne"])
142 if session
["method"] in ("list", "show", "delete") or session
.get("set_project"):
143 p_filter
["_admin.projects_read.cont"] = project_filter
145 p_filter
["_admin.projects_write.cont"] = project_filter
147 if session
["method"] in ("list", "show", "delete") or session
.get("set_project"):
148 p_filter
["_admin.projects_read.ncont"] = project_filter_n
150 p_filter
["_admin.projects_write.ncont"] = project_filter_n
154 def check_conflict_on_new(self
, session
, indata
):
156 Check that the data to be inserted is valid
157 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
158 :param indata: data to be inserted
159 :return: None or raises EngineException
163 def check_conflict_on_edit(self
, session
, final_content
, edit_content
, _id
):
165 Check that the data to be edited/uploaded is valid
166 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
167 :param final_content: data once modified. This method may change it.
168 :param edit_content: incremental data that contains the modifications to apply
169 :param _id: internal _id
170 :return: None or raises EngineException
172 if not self
.multiproject
:
174 # Change public status
175 if session
["public"] is not None:
176 if session
["public"] and "ANY" not in final_content
["_admin"]["projects_read"]:
177 final_content
["_admin"]["projects_read"].append("ANY")
178 final_content
["_admin"]["projects_write"].clear()
179 if not session
["public"] and "ANY" in final_content
["_admin"]["projects_read"]:
180 final_content
["_admin"]["projects_read"].remove("ANY")
182 # Change project status
183 if session
.get("set_project"):
184 for p
in session
["set_project"]:
185 if p
not in final_content
["_admin"]["projects_read"]:
186 final_content
["_admin"]["projects_read"].append(p
)
188 def check_unique_name(self
, session
, name
, _id
=None):
190 Check that the name is unique for this project
191 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
192 :param name: name to be checked
193 :param _id: If not None, ignore this entry that are going to change
194 :return: None or raises EngineException
196 if not self
.multiproject
:
199 _filter
= self
._get
_project
_filter
(session
)
200 _filter
["name"] = name
202 _filter
["_id.neq"] = _id
203 if self
.db
.get_one(self
.topic
, _filter
, fail_on_empty
=False, fail_on_more
=False):
204 raise EngineException("name '{}' already exists for {}".format(name
, self
.topic
), HTTPStatus
.CONFLICT
)
207 def format_on_new(content
, project_id
=None, make_public
=False):
209 Modifies content descriptor to include _admin
210 :param content: descriptor to be modified
211 :param project_id: if included, it add project read/write permissions. Can be None or a list
212 :param make_public: if included it is generated as public for reading.
213 :return: op_id: operation id on asynchronous operation, None otherwise. In addition content is modified
216 if "_admin" not in content
:
217 content
["_admin"] = {}
218 if not content
["_admin"].get("created"):
219 content
["_admin"]["created"] = now
220 content
["_admin"]["modified"] = now
221 if not content
.get("_id"):
222 content
["_id"] = str(uuid4())
223 if project_id
is not None:
224 if not content
["_admin"].get("projects_read"):
225 content
["_admin"]["projects_read"] = list(project_id
)
227 content
["_admin"]["projects_read"].append("ANY")
228 if not content
["_admin"].get("projects_write"):
229 content
["_admin"]["projects_write"] = list(project_id
)
233 def format_on_edit(final_content
, edit_content
):
235 Modifies final_content to admin information upon edition
236 :param final_content: final content to be stored at database
237 :param edit_content: user requested update content
238 :return: operation id, if this edit implies an asynchronous operation; None otherwise
240 if final_content
.get("_admin"):
242 final_content
["_admin"]["modified"] = now
245 def _send_msg(self
, action
, content
):
247 content
.pop("_admin", None)
248 self
.msg
.write(self
.topic_msg
, action
, content
)
250 def check_conflict_on_del(self
, session
, _id
, db_content
):
252 Check if deletion can be done because of dependencies if it is not force. To override
253 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
254 :param _id: internal _id
255 :param db_content: The database content of this item _id
256 :return: None if ok or raises EngineException with the conflict
261 def _update_input_with_kwargs(desc
, kwargs
):
263 Update descriptor with the kwargs. It contains dot separated keys
264 :param desc: dictionary to be updated
265 :param kwargs: plain dictionary to be used for updating.
266 :return: None, 'desc' is modified. It raises EngineException.
271 for k
, v
in kwargs
.items():
272 update_content
= desc
276 if kitem_old
is not None:
277 update_content
= update_content
[kitem_old
]
278 if isinstance(update_content
, dict):
280 elif isinstance(update_content
, list):
281 kitem_old
= int(kitem
)
283 raise EngineException(
284 "Invalid query string '{}'. Descriptor is not a list nor dict at '{}'".format(k
, kitem
))
285 update_content
[kitem_old
] = v
287 raise EngineException(
288 "Invalid query string '{}'. Descriptor does not contain '{}'".format(k
, kitem_old
))
290 raise EngineException("Invalid query string '{}'. Expected integer index list instead of '{}'".format(
293 raise EngineException(
294 "Invalid query string '{}'. Index '{}' out of range".format(k
, kitem_old
))
296 def show(self
, session
, _id
):
298 Get complete information on an topic
299 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
300 :param _id: server internal id
301 :return: dictionary, raise exception if not found.
303 if not self
.multiproject
:
306 filter_db
= self
._get
_project
_filter
(session
)
307 # To allow project&user addressing by name AS WELL AS _id
308 filter_db
[BaseTopic
.id_field(self
.topic
, _id
)] = _id
309 return self
.db
.get_one(self
.topic
, filter_db
)
310 # TODO transform data for SOL005 URL requests
311 # TODO remove _admin if not admin
313 def get_file(self
, session
, _id
, path
=None, accept_header
=None):
315 Only implemented for descriptor topics. Return the file content of a descriptor
316 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
317 :param _id: Identity of the item to get content
318 :param path: artifact path or "$DESCRIPTOR" or None
319 :param accept_header: Content of Accept header. Must contain applition/zip or/and text/plain
320 :return: opened file or raises an exception
322 raise EngineException("Method get_file not valid for this topic", HTTPStatus
.INTERNAL_SERVER_ERROR
)
324 def list(self
, session
, filter_q
=None):
326 Get a list of the topic that matches a filter
327 :param session: contains the used login username and working project
328 :param filter_q: filter of data to be applied
329 :return: The list, it can be empty if no one match the filter.
333 if self
.multiproject
:
334 filter_q
.update(self
._get
_project
_filter
(session
))
336 # TODO transform data for SOL005 URL requests. Transform filtering
337 # TODO implement "field-type" query string SOL005
338 return self
.db
.get_list(self
.topic
, filter_q
)
340 def new(self
, rollback
, session
, indata
=None, kwargs
=None, headers
=None):
342 Creates a new entry into database.
343 :param rollback: list to append created items at database in case a rollback may to be done
344 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
345 :param indata: data to be inserted
346 :param kwargs: used to override the indata descriptor
347 :param headers: http request headers
349 _id: identity of the inserted data.
350 op_id: operation id if this is asynchronous, None otherwise
353 content
= self
._remove
_envelop
(indata
)
355 # Override descriptor with query string kwargs
356 self
._update
_input
_with
_kwargs
(content
, kwargs
)
357 content
= self
._validate
_input
_new
(content
, force
=session
["force"])
358 self
.check_conflict_on_new(session
, content
)
359 op_id
= self
.format_on_new(content
, project_id
=session
["project_id"], make_public
=session
["public"])
360 _id
= self
.db
.create(self
.topic
, content
)
361 rollback
.append({"topic": self
.topic
, "_id": _id
})
363 content
["op_id"] = op_id
364 self
._send
_msg
("create", content
)
366 except ValidationError
as e
:
367 raise EngineException(e
, HTTPStatus
.UNPROCESSABLE_ENTITY
)
369 def upload_content(self
, session
, _id
, indata
, kwargs
, headers
):
371 Only implemented for descriptor topics. Used for receiving content by chunks (with a transaction_id header
372 and/or gzip file. It will store and extract)
373 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
374 :param _id : the database id of entry to be updated
375 :param indata: http body request
376 :param kwargs: user query string to override parameters. NOT USED
377 :param headers: http request headers
378 :return: True package has is completely uploaded or False if partial content has been uplodaed.
379 Raise exception on error
381 raise EngineException("Method upload_content not valid for this topic", HTTPStatus
.INTERNAL_SERVER_ERROR
)
383 def delete_list(self
, session
, filter_q
=None):
385 Delete a several entries of a topic. This is for internal usage and test only, not exposed to NBI API
386 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
387 :param filter_q: filter of data to be applied
388 :return: The deleted list, it can be empty if no one match the filter.
390 # TODO add admin to filter, validate rights
393 if self
.multiproject
:
394 filter_q
.update(self
._get
_project
_filter
(session
))
395 return self
.db
.del_list(self
.topic
, filter_q
)
397 def delete_extra(self
, session
, _id
, db_content
):
399 Delete other things apart from database entry of a item _id.
400 e.g.: other associated elements at database and other file system storage
401 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
402 :param _id: server internal id
403 :param db_content: The database content of the _id. It is already deleted when reached this method, but the
404 content is needed in same cases
405 :return: None if ok or raises EngineException with the problem
409 def delete(self
, session
, _id
, dry_run
=False):
411 Delete item by its internal _id
412 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
413 :param _id: server internal id
414 :param dry_run: make checking but do not delete
415 :return: operation id (None if there is not operation), raise exception if error or not found, conflict, ...
418 # To allow addressing projects and users by name AS WELL AS by _id
419 filter_q
= {BaseTopic
.id_field(self
.topic
, _id
): _id
}
420 item_content
= self
.db
.get_one(self
.topic
, filter_q
)
422 # TODO add admin to filter, validate rights
423 # data = self.get_item(topic, _id)
424 self
.check_conflict_on_del(session
, _id
, item_content
)
428 if self
.multiproject
:
429 filter_q
.update(self
._get
_project
_filter
(session
))
430 if self
.multiproject
and session
["project_id"]:
431 # remove reference from project_read. If not last delete
432 # if this topic is not part of session["project_id"] no midification at database is done and an exception
434 self
.db
.set_one(self
.topic
, filter_q
, update_dict
=None,
435 pull
={"_admin.projects_read": {"$in": session
["project_id"]}})
436 # try to delete if there is not any more reference from projects. Ignore if it is not deleted
437 filter_q
= {'_id': _id
, '_admin.projects_read': [[], ["ANY"]]}
438 v
= self
.db
.del_one(self
.topic
, filter_q
, fail_on_empty
=False)
439 if not v
or not v
["deleted"]:
442 self
.db
.del_one(self
.topic
, filter_q
)
443 self
.delete_extra(session
, _id
, item_content
)
444 self
._send
_msg
("deleted", {"_id": _id
})
447 def edit(self
, session
, _id
, indata
=None, kwargs
=None, content
=None):
449 Change the content of an item
450 :param session: contains "username", "admin", "force", "public", "project_id", "set_project"
451 :param _id: server internal id
452 :param indata: contains the changes to apply
453 :param kwargs: modifies indata
454 :param content: original content of the item
455 :return: op_id: operation id if this is processed asynchronously, None otherwise
457 indata
= self
._remove
_envelop
(indata
)
459 # Override descriptor with query string kwargs
461 self
._update
_input
_with
_kwargs
(indata
, kwargs
)
463 if indata
and session
.get("set_project"):
464 raise EngineException("Cannot edit content and set to project (query string SET_PROJECT) at same time",
465 HTTPStatus
.UNPROCESSABLE_ENTITY
)
466 indata
= self
._validate
_input
_edit
(indata
, force
=session
["force"])
468 # TODO self._check_edition(session, indata, _id, force)
470 content
= self
.show(session
, _id
)
471 deep_update_rfc7396(content
, indata
)
473 # To allow project addressing by name AS WELL AS _id. Get the _id, just in case the provided one is a name
474 _id
= content
.get("_id") or _id
476 self
.check_conflict_on_edit(session
, content
, indata
, _id
=_id
)
477 op_id
= self
.format_on_edit(content
, indata
)
479 self
.db
.replace(self
.topic
, _id
, content
)
481 indata
.pop("_admin", None)
483 indata
["op_id"] = op_id
485 self
._send
_msg
("edit", indata
)
487 except ValidationError
as e
:
488 raise EngineException(e
, HTTPStatus
.UNPROCESSABLE_ENTITY
)