| # -*- coding: utf-8 -*- |
| |
| import logging |
| from uuid import uuid4 |
| from http import HTTPStatus |
| from time import time |
| from osm_common.dbbase import deep_update_rfc7396 |
| from validation import validate_input, ValidationError |
| |
| __author__ = "Alfonso Tierno <alfonso.tiernosepulveda@telefonica.com>" |
| |
| |
| class EngineException(Exception): |
| |
| def __init__(self, message, http_code=HTTPStatus.BAD_REQUEST): |
| self.http_code = http_code |
| Exception.__init__(self, message) |
| |
| |
| def get_iterable(input_var): |
| """ |
| Returns an iterable, in case input_var is None it just returns an empty tuple |
| :param input_var: can be a list, tuple or None |
| :return: input_var or () if it is None |
| """ |
| if input_var is None: |
| return () |
| return input_var |
| |
| |
| def versiontuple(v): |
| """utility for compare dot separate versions. Fills with zeros to proper number comparison""" |
| filled = [] |
| for point in v.split("."): |
| filled.append(point.zfill(8)) |
| return tuple(filled) |
| |
| |
| class BaseTopic: |
| # static variables for all instance classes |
| topic = None # to_override |
| topic_msg = None # to_override |
| schema_new = None # to_override |
| schema_edit = None # to_override |
| |
| def __init__(self, db, fs, msg): |
| self.db = db |
| self.fs = fs |
| self.msg = msg |
| self.logger = logging.getLogger("nbi.engine") |
| |
| @staticmethod |
| def _remove_envelop(indata=None): |
| if not indata: |
| return {} |
| return indata |
| |
| def _validate_input_new(self, input, force=False): |
| """ |
| Validates input user content for a new entry. It uses jsonschema. Some overrides will use pyangbind |
| :param input: user input content for the new topic |
| :param force: may be used for being more tolerant |
| :return: The same input content, or a changed version of it. |
| """ |
| if self.schema_new: |
| validate_input(input, self.schema_new) |
| return input |
| |
| def _validate_input_edit(self, input, force=False): |
| """ |
| Validates input user content for an edition. It uses jsonschema. Some overrides will use pyangbind |
| :param input: user input content for the new topic |
| :param force: may be used for being more tolerant |
| :return: The same input content, or a changed version of it. |
| """ |
| if self.schema_edit: |
| validate_input(input, self.schema_edit) |
| return input |
| |
| @staticmethod |
| def _get_project_filter(session, write=False, show_all=True): |
| """ |
| Generates a filter dictionary for querying database, so that only allowed items for this project can be |
| addressed. Only propietary or public can be used. Allowed projects are at _admin.project_read/write. If it is |
| not present or contains ANY mean public. |
| :param session: contains "username", if user is "admin" and the working "project_id" |
| :param write: if operation is for reading (False) or writing (True) |
| :param show_all: if True it will show public or |
| :return: |
| """ |
| if write: |
| k = "_admin.projects_write.cont" |
| else: |
| k = "_admin.projects_read.cont" |
| if not show_all: |
| return {k: session["project_id"]} |
| elif session["admin"]: # and show_all: # allows all |
| return {} |
| else: |
| return {k: ["ANY", session["project_id"], None]} |
| |
| def check_conflict_on_new(self, session, indata, force=False): |
| """ |
| Check that the data to be inserted is valid |
| :param session: contains "username", if user is "admin" and the working "project_id" |
| :param indata: data to be inserted |
| :param force: boolean. With force it is more tolerant |
| :return: None or raises EngineException |
| """ |
| pass |
| |
| def check_conflict_on_edit(self, session, final_content, edit_content, _id, force=False): |
| """ |
| Check that the data to be edited/uploaded is valid |
| :param session: contains "username", if user is "admin" and the working "project_id" |
| :param final_content: data once modified |
| :param edit_content: incremental data that contains the modifications to apply |
| :param _id: internal _id |
| :param force: boolean. With force it is more tolerant |
| :return: None or raises EngineException |
| """ |
| pass |
| |
| def check_unique_name(self, session, name, _id=None): |
| """ |
| Check that the name is unique for this project |
| :param session: contains "username", if user is "admin" and the working "project_id" |
| :param name: name to be checked |
| :param _id: If not None, ignore this entry that are going to change |
| :return: None or raises EngineException |
| """ |
| _filter = self._get_project_filter(session, write=False, show_all=False) |
| _filter["name"] = name |
| if _id: |
| _filter["_id.neq"] = _id |
| if self.db.get_one(self.topic, _filter, fail_on_empty=False, fail_on_more=False): |
| raise EngineException("name '{}' already exists for {}".format(name, self.topic), HTTPStatus.CONFLICT) |
| |
| @staticmethod |
| def format_on_new(content, project_id=None, make_public=False): |
| """ |
| Modifies content descriptor to include _admin |
| :param content: descriptor to be modified |
| :param project_id: if included, it add project read/write permissions |
| :param make_public: if included it is generated as public for reading. |
| :return: None, but content is modified |
| """ |
| now = time() |
| if "_admin" not in content: |
| content["_admin"] = {} |
| if not content["_admin"].get("created"): |
| content["_admin"]["created"] = now |
| content["_admin"]["modified"] = now |
| if not content.get("_id"): |
| content["_id"] = str(uuid4()) |
| if project_id: |
| if not content["_admin"].get("projects_read"): |
| content["_admin"]["projects_read"] = [project_id] |
| if make_public: |
| content["_admin"]["projects_read"].append("ANY") |
| if not content["_admin"].get("projects_write"): |
| content["_admin"]["projects_write"] = [project_id] |
| |
| @staticmethod |
| def format_on_edit(final_content, edit_content): |
| if final_content.get("_admin"): |
| now = time() |
| final_content["_admin"]["modified"] = now |
| |
| def _send_msg(self, action, content): |
| if self.topic_msg: |
| content.pop("_admin", None) |
| self.msg.write(self.topic_msg, action, content) |
| |
| def check_conflict_on_del(self, session, _id, force=False): |
| """ |
| Check if deletion can be done because of dependencies if it is not force. To override |
| :param session: contains "username", if user is "admin" and the working "project_id" |
| :param _id: itnernal _id |
| :param force: Avoid this checking |
| :return: None if ok or raises EngineException with the conflict |
| """ |
| pass |
| |
| @staticmethod |
| def _update_input_with_kwargs(desc, kwargs): |
| """ |
| Update descriptor with the kwargs. It contains dot separated keys |
| :param desc: dictionary to be updated |
| :param kwargs: plain dictionary to be used for updating. |
| :return: None, 'desc' is modified. It raises EngineException. |
| """ |
| if not kwargs: |
| return |
| try: |
| for k, v in kwargs.items(): |
| update_content = desc |
| kitem_old = None |
| klist = k.split(".") |
| for kitem in klist: |
| if kitem_old is not None: |
| update_content = update_content[kitem_old] |
| if isinstance(update_content, dict): |
| kitem_old = kitem |
| elif isinstance(update_content, list): |
| kitem_old = int(kitem) |
| else: |
| raise EngineException( |
| "Invalid query string '{}'. Descriptor is not a list nor dict at '{}'".format(k, kitem)) |
| update_content[kitem_old] = v |
| except KeyError: |
| raise EngineException( |
| "Invalid query string '{}'. Descriptor does not contain '{}'".format(k, kitem_old)) |
| except ValueError: |
| raise EngineException("Invalid query string '{}'. Expected integer index list instead of '{}'".format( |
| k, kitem)) |
| except IndexError: |
| raise EngineException( |
| "Invalid query string '{}'. Index '{}' out of range".format(k, kitem_old)) |
| |
| def show(self, session, _id): |
| """ |
| Get complete information on an topic |
| :param session: contains the used login username and working project |
| :param _id: server internal id |
| :return: dictionary, raise exception if not found. |
| """ |
| filter_db = self._get_project_filter(session, write=False, show_all=True) |
| filter_db["_id"] = _id |
| return self.db.get_one(self.topic, filter_db) |
| # TODO transform data for SOL005 URL requests |
| # TODO remove _admin if not admin |
| |
| def get_file(self, session, _id, path=None, accept_header=None): |
| """ |
| Only implemented for descriptor topics. Return the file content of a descriptor |
| :param session: contains the used login username and working project |
| :param _id: Identity of the item to get content |
| :param path: artifact path or "$DESCRIPTOR" or None |
| :param accept_header: Content of Accept header. Must contain applition/zip or/and text/plain |
| :return: opened file or raises an exception |
| """ |
| raise EngineException("Method get_file not valid for this topic", HTTPStatus.INTERNAL_SERVER_ERROR) |
| |
| def list(self, session, filter_q=None): |
| """ |
| Get a list of the topic that matches a filter |
| :param session: contains the used login username and working project |
| :param filter_q: filter of data to be applied |
| :return: The list, it can be empty if no one match the filter. |
| """ |
| if not filter_q: |
| filter_q = {} |
| |
| filter_q.update(self._get_project_filter(session, write=False, show_all=True)) |
| |
| # TODO transform data for SOL005 URL requests. Transform filtering |
| # TODO implement "field-type" query string SOL005 |
| return self.db.get_list(self.topic, filter_q) |
| |
| def new(self, rollback, session, indata=None, kwargs=None, headers=None, force=False, make_public=False): |
| """ |
| Creates a new entry into database. |
| :param rollback: list to append created items at database in case a rollback may to be done |
| :param session: contains the used login username and working project |
| :param indata: data to be inserted |
| :param kwargs: used to override the indata descriptor |
| :param headers: http request headers |
| :param force: If True avoid some dependence checks |
| :param make_public: Make the created item public to all projects |
| :return: _id: identity of the inserted data. |
| """ |
| try: |
| content = self._remove_envelop(indata) |
| |
| # Override descriptor with query string kwargs |
| self._update_input_with_kwargs(content, kwargs) |
| content = self._validate_input_new(content, force=force) |
| self.check_conflict_on_new(session, content, force=force) |
| self.format_on_new(content, project_id=session["project_id"], make_public=make_public) |
| _id = self.db.create(self.topic, content) |
| rollback.append({"topic": self.topic, "_id": _id}) |
| self._send_msg("create", content) |
| return _id |
| except ValidationError as e: |
| raise EngineException(e, HTTPStatus.UNPROCESSABLE_ENTITY) |
| |
| def upload_content(self, session, _id, indata, kwargs, headers, force=False): |
| """ |
| Only implemented for descriptor topics. Used for receiving content by chunks (with a transaction_id header |
| and/or gzip file. It will store and extract) |
| :param session: session |
| :param _id : the database id of entry to be updated |
| :param indata: http body request |
| :param kwargs: user query string to override parameters. NOT USED |
| :param headers: http request headers |
| :param force: to be more tolerant with validation |
| :return: True package has is completely uploaded or False if partial content has been uplodaed. |
| Raise exception on error |
| """ |
| raise EngineException("Method upload_content not valid for this topic", HTTPStatus.INTERNAL_SERVER_ERROR) |
| |
| def delete_list(self, session, filter_q=None): |
| """ |
| Delete a several entries of a topic. This is for internal usage and test only, not exposed to NBI API |
| :param session: contains the used login username and working project |
| :param filter_q: filter of data to be applied |
| :return: The deleted list, it can be empty if no one match the filter. |
| """ |
| # TODO add admin to filter, validate rights |
| if not filter_q: |
| filter_q = {} |
| filter_q.update(self._get_project_filter(session, write=True, show_all=True)) |
| return self.db.del_list(self.topic, filter_q) |
| |
| def delete(self, session, _id, force=False, dry_run=False): |
| """ |
| Delete item by its internal _id |
| :param session: contains the used login username, working project, and admin rights |
| :param _id: server internal id |
| :param force: indicates if deletion must be forced in case of conflict |
| :param dry_run: make checking but do not delete |
| :return: dictionary with deleted item _id. It raises EngineException on error: not found, conflict, ... |
| """ |
| # TODO add admin to filter, validate rights |
| # data = self.get_item(topic, _id) |
| self.check_conflict_on_del(session, _id, force) |
| filter_q = self._get_project_filter(session, write=True, show_all=True) |
| filter_q["_id"] = _id |
| if not dry_run: |
| v = self.db.del_one(self.topic, filter_q) |
| self._send_msg("deleted", {"_id": _id}) |
| return v |
| return None |
| |
| def edit(self, session, _id, indata=None, kwargs=None, force=False, content=None): |
| indata = self._remove_envelop(indata) |
| |
| # Override descriptor with query string kwargs |
| if kwargs: |
| self._update_input_with_kwargs(indata, kwargs) |
| try: |
| indata = self._validate_input_edit(indata, force=force) |
| |
| # TODO self._check_edition(session, indata, _id, force) |
| if not content: |
| content = self.show(session, _id) |
| deep_update_rfc7396(content, indata) |
| self.check_conflict_on_edit(session, content, indata, _id=_id, force=force) |
| self.format_on_edit(content, indata) |
| self.db.replace(self.topic, _id, content) |
| |
| indata.pop("_admin", None) |
| indata["_id"] = _id |
| self._send_msg("edit", indata) |
| return id |
| except ValidationError as e: |
| raise EngineException(e, HTTPStatus.UNPROCESSABLE_ENTITY) |