NTV.json_ntv.ntv_comment

  1# -*- coding: utf-8 -*-
  2"""
  3Created on Sun Oct  2 22:24:59 2022
  4
  5@author: philippe@loco-labs.io
  6
  7The `NTV.json_ntv.ntv_comment` module contains the `NtvComment` class.
  8"""
  9import json
 10from json_ntv.ntv_patch import NtvPatch
 11
 12
 13class NtvComment:
 14    '''This class includes comments and change management methods for NTV entities :
 15
 16    *Attributes :*
 17
 18    - **_ntv** : Ntv entity being commented on
 19    - **_comments**:  list of NtvPatch to apply to the NTV entity
 20
 21    *dynamic values (@property)*
 22    - `comments`
 23    - `ntv`
 24
 25    *instance method*
 26    - `add`
 27    - `accept`
 28    - `reject`
 29    - `json`
 30    '''
 31
 32    def __init__(self, ntv, comments=None):
 33        ''' constructor
 34
 35        *Parameters*
 36
 37        - **ntv**: Ntv - NTV entity being commented on
 38        - **comments**: list of NtvPatch (default None) - comments applied to the entity
 39        '''
 40        self._ntv = ntv
 41        self._comments = []
 42        if not comments:
 43            return
 44        if comments.__class__.__name__ in ('NtvPatch', 'NtvOp'):
 45            self._comments = [NtvPatch(comments)]
 46        elif isinstance(comments, list):
 47            self._comments = [NtvPatch(comment) for comment in comments]
 48
 49    def __repr__(self):
 50        ''' string representation'''
 51        if not self._comments:
 52            return 'no comments'
 53        return json.dumps(self.json())
 54
 55    def __eq__(self, other):
 56        ''' equal if _comments and _ntv are equal'''
 57        return self.__class__.__name__ == other.__class__.__name__ and\
 58            self._ntv == other._ntv and self._comments == other._comments
 59
 60    @property
 61    def comments(self):
 62        '''getters _comments'''
 63        return self._comments
 64
 65    @property
 66    def ntv(self):
 67        '''getters _ntv'''
 68        return self._ntv
 69
 70    def add(self, comment=None):
 71        '''add comment in the list of comments
 72
 73        *Parameters*
 74
 75        - **comment**: NtvPatch (default None) - new comment to apply
 76        '''
 77        #comment = comment if isinstance(comment, (NtvPatch, NtvOp)) else NtvPatch(comment)
 78        self._comments.append(NtvPatch(comment))
 79        return len(self._comments)-1
 80
 81    def reject(self, all_comment=False):
 82        '''remove the last or all comments
 83
 84        *Parameters*
 85
 86        - **all_comment**: Boolean (default False) - if False, remove the last comment
 87
 88        *return*: resulting NtvComment
 89        '''
 90        if all_comment or len(self._comments) <= 1:
 91            return NtvComment(self._ntv, [])
 92        return NtvComment(self._ntv, self._comments[:-1])
 93
 94    def accept(self, all_comment=True):
 95        '''apply the first or all comments
 96
 97        *Parameters*
 98
 99        - **all_comment**: Boolean (default True) - if False, apply the first comment
100
101        *return*: resulting NtvComment
102        '''
103        if not self._comments:
104            return NtvComment(self._ntv, self._comments)
105        apply = self._comments if all_comment else [self._comments[0]]
106        ntv = self._ntv
107        for comment in apply:
108            ntv = comment.exe(ntv)
109        return NtvComment(ntv, [] if all_comment else self._comments[1:])
110
111    def json(self, ntv=False, comments=True):
112        ''' return json-value.
113
114        *Parameters*
115
116        - **ntv**: Boolean (default False) - if True, include ntv
117        - **comments**: Boolean (default True) - if True, include comments
118
119        *return*: resulting json-value
120        '''
121        jsn = {}
122        if self._comments and comments:
123            jsn['comments'] = [comment.json for comment in self._comments]
124        if ntv:
125            jsn['ntv'] = self._ntv.to_obj()
126        return jsn