Test Failed
Push — master ( 2e7a01...f7a23a )
by Humberto
02:19
created

build.models   F

Complexity

Total Complexity 160

Size/Duplication

Total Lines 895
Duplicated Lines 0 %

Test Coverage

Coverage 73.48%

Importance

Changes 0
Metric Value
wmc 160
eloc 505
dl 0
loc 895
ccs 338
cts 460
cp 0.7348
rs 2
c 0
b 0
f 0

58 Methods

Rating   Name   Duplication   Size   Complexity  
A Path.__eq__() 0 5 3
A Path.link_affected_by_interface() 0 8 4
A Path.make_vlans_available() 0 5 2
A Path.is_affected_by_link() 0 5 2
A Path.choose_vlans() 0 6 2
A EVCBase.sync() 0 4 1
B EVCDeploy._prepare_push_flow() 0 44 5
A EVCDeploy.remove_current_flows() 0 20 3
A DynamicPathManager.get_best_path() 0 7 2
A EVCDeploy._prepare_nni_flow() 0 11 1
A EVCDeploy.discover_new_paths() 0 3 1
A EVCBase.__eq__() 0 10 4
A EVCDeploy.change_path() 0 2 1
A LinkProtection.deploy_to() 0 10 3
A LinkProtection.is_using_dynamic_path() 0 8 5
A EVCBase.__init__() 0 86 3
B Path.status() 0 27 7
A LinkProtection.is_using_primary_path() 0 3 1
A EVCDeploy.remove() 0 5 1
A LinkProtection.is_affected_by_link() 0 3 1
A DynamicPathManager.get_best_paths() 0 5 2
A EVCDeploy.is_primary_path_affected_by_link() 0 3 1
A DynamicPathManager.get_paths() 0 14 2
A EVCDeploy._install_uni_flows() 0 44 4
A EVCBase.id() 0 4 1
A EVCDeploy.is_backup_path_affected_by_link() 0 3 1
A EVCDeploy.link_affected_by_interface() 0 3 1
A EVCDeploy.should_deploy() 0 15 4
A Path.as_dict() 0 3 1
A EVCDeploy.create() 0 2 1
A EVCDeploy.is_using_backup_path() 0 3 1
B EVCBase._validate() 0 24 6
A EVCDeploy.deploy() 0 14 3
A EVCDeploy.reprovision() 0 2 1
A DynamicPathManager.set_controller() 0 4 1
A EVCBase.archive() 0 3 1
B EVCDeploy._install_direct_uni_flows() 0 33 7
A EVCDeploy.deploy_to_backup_path() 0 25 5
A EVCBase.__repr__() 0 3 1
B EVCDeploy.deploy_to_path() 0 41 7
A EVCDeploy._prepare_flow_mod() 0 10 1
A EVCDeploy.get_path_status() 0 13 4
A EVCDeploy.is_using_dynamic_path() 0 8 5
C EVCBase.update() 0 37 9
A EVCDeploy.deploy_to_primary_path() 0 14 3
A DynamicPathManager.create_path() 0 17 5
A EVCDeploy._send_flow_mods() 0 14 1
C LinkProtection.handle_link_up() 0 36 9
A EVCDeploy.is_using_primary_path() 0 3 1
A EVCDeploy.get_cookie() 0 4 1
A EVCDeploy.is_affected_by_link() 0 3 1
B LinkProtection.handle_link_down() 0 25 6
A EVCDeploy._prepare_pop_flow() 0 7 1
A EVCDeploy.links_zipped() 0 6 2
A LinkProtection.is_using_backup_path() 0 3 1
A DynamicPathManager._clear_path() 0 4 1
A EVCDeploy._install_nni_flows() 0 17 2
A EVCBase.as_dict() 0 48 4

How to fix   Complexity   

Complexity

Complex classes like build.models often do a lot of different things. To break such a class down, we need to identify a cohesive component within that class. A common approach to find such a component is to look for fields/methods that share the same prefixes, or suffixes.

Once you have determined the fields that belong together, you can apply the Extract Class refactoring. If the component makes sense as a sub-class, Extract Subclass is also a candidate, and is often faster.

1
"""Classes used in the main application."""
2 1
from datetime import datetime
3 1
from uuid import uuid4
4
5 1
import requests
6
7 1
from kytos.core import log
8 1
from kytos.core.common import EntityStatus, GenericEntity
9 1
from kytos.core.exceptions import KytosNoTagAvailableError
10 1
from kytos.core.helpers import get_time, now
11 1
from kytos.core.interface import UNI
12 1
from kytos.core.link import Link
13 1
from napps.kytos.mef_eline import settings
14 1
from napps.kytos.mef_eline.storehouse import StoreHouse
15
16
17 1
class Path(list, GenericEntity):
18
    """Class to represent a Path."""
19
20 1
    def __eq__(self, other=None):
21
        """Compare paths."""
22 1
        if not other or not isinstance(other, Path):
23 1
            return False
24 1
        return super().__eq__(other)
25
26 1
    def is_affected_by_link(self, link=None):
27
        """Verify if the current path is affected by link."""
28 1
        if not link:
29
            return False
30 1
        return link in self
31
32 1
    def link_affected_by_interface(self, interface=None):
33
        """Return the link using this interface, if any, or None otherwise."""
34
        if not interface:
35
            return None
36
        for link in self:
37
            if interface in (link.endpoint_a, link.endpoint_b):
38
                return link
39
        return None
40
41 1
    def choose_vlans(self):
42
        """Choose the VLANs to be used for the circuit."""
43
        for link in self:
44
            tag = link.get_next_available_tag()
45
            link.use_tag(tag)
46
            link.add_metadata('s_vlan', tag)
47
48 1
    def make_vlans_available(self):
49
        """Make the VLANs used in a path available when undeployed."""
50 1
        for link in self:
51 1
            link.make_tag_available(link.get_metadata('s_vlan'))
52 1
            link.remove_metadata('s_vlan')
53
54 1
    @property
55
    def status(self):
56
        """Check for the  status of a path.
57
58
        If any link in this path is down, the path is considered down.
59
        """
60 1
        if not self:
61 1
            return EntityStatus.DISABLED
62
63 1
        endpoint = '%s/%s' % (settings.TOPOLOGY_URL, 'links')
64 1
        api_reply = requests.get(endpoint)
65 1
        if api_reply.status_code != getattr(requests.codes, 'ok'):
66
            log.error('Failed to get links at %s. Returned %s',
67
                      endpoint, api_reply.status_code)
68
            return None
69 1
        links = api_reply.json()['links']
70 1
        return_status = EntityStatus.UP
71 1
        for path_link in self:
72 1
            try:
73 1
                link = links[path_link.id]
74
            except KeyError:
75
                return EntityStatus.DISABLED
76 1
            if link['enabled'] is False:
77 1
                return EntityStatus.DISABLED
78 1
            if link['active'] is False:
79 1
                return_status = EntityStatus.DOWN
80 1
        return return_status
81
82 1
    def as_dict(self):
83
        """Return list comprehension of links as_dict."""
84 1
        return [link.as_dict() for link in self if link]
85
86
87 1
class DynamicPathManager:
88
    """Class to handle and create paths."""
89
90 1
    controller = None
91
92 1
    @classmethod
93 1
    def set_controller(cls, controller=None):
94
        """Set the controller to discovery news paths."""
95 1
        cls.controller = controller
96
97 1
    @staticmethod
98
    def get_paths(circuit):
99
        """Get a valid path for the circuit from the Pathfinder."""
100
        endpoint = settings.PATHFINDER_URL
101
        request_data = {"source": circuit.uni_a.interface.id,
102
                        "destination": circuit.uni_z.interface.id}
103
        api_reply = requests.post(endpoint, json=request_data)
104
105
        if api_reply.status_code != getattr(requests.codes, 'ok'):
106
            log.error("Failed to get paths at %s. Returned %s",
107
                      endpoint, api_reply.status_code)
108
            return None
109
        reply_data = api_reply.json()
110
        return reply_data.get('paths')
111
112 1
    @staticmethod
113
    def _clear_path(path):
114
        """Remove switches from a path, returning only interfaces."""
115
        return [endpoint for endpoint in path if len(endpoint) > 23]
116
117 1
    @classmethod
118
    def get_best_path(cls, circuit):
119
        """Return the best path available for a circuit, if exists."""
120
        paths = cls.get_paths(circuit)
121
        if paths:
122
            return cls.create_path(cls.get_paths(circuit)[0]['hops'])
123
        return None
124
125 1
    @classmethod
126
    def get_best_paths(cls, circuit):
127
        """Return the best paths available for a circuit, if they exist."""
128
        for path in cls.get_paths(circuit):
129
            yield cls.create_path(path['hops'])
130
131 1
    @classmethod
132
    def create_path(cls, path):
133
        """Return the path containing only the interfaces."""
134
        new_path = Path()
135
        clean_path = cls._clear_path(path)
136
137
        if len(clean_path) % 2:
138
            return None
139
140
        for link in zip(clean_path[1:-1:2], clean_path[2::2]):
141
            interface_a = cls.controller.get_interface_by_id(link[0])
142
            interface_b = cls.controller.get_interface_by_id(link[1])
143
            if interface_a is None or interface_b is None:
144
                return None
145
            new_path.append(Link(interface_a, interface_b))
146
147
        return new_path
148
149
150 1
class EVCBase(GenericEntity):
151
    """Class to represent a circuit."""
152
153 1
    unique_attributes = ['name', 'uni_a', 'uni_z']
154
155 1
    def __init__(self, controller, **kwargs):
156
        """Create an EVC instance with the provided parameters.
157
158
        Args:
159
            id(str): EVC identifier. Whether it's None an ID will be genereted.
160
            name: represents an EVC name.(Required)
161
            uni_a (UNI): Endpoint A for User Network Interface.(Required)
162
            uni_z (UNI): Endpoint Z for User Network Interface.(Required)
163
            start_date(datetime|str): Date when the EVC was registred.
164
                                      Default is now().
165
            end_date(datetime|str): Final date that the EVC will be fineshed.
166
                                    Default is None.
167
            bandwidth(int): Bandwidth used by EVC instance. Default is 0.
168
            primary_links(list): Primary links used by evc. Default is []
169
            backup_links(list): Backups links used by evc. Default is []
170
            current_path(list): Circuit being used at the moment if this is an
171
                                active circuit. Default is [].
172
            primary_path(list): primary circuit offered to user IF one or more
173
                                links were provided. Default is [].
174
            backup_path(list): backup circuit offered to the user IF one or
175
                               more links were provided. Default is [].
176
            dynamic_backup_path(bool): Enable computer backup path dynamically.
177
                                       Dafault is False.
178
            creation_time(datetime|str): datetime when the circuit should be
179
                                         activated. default is now().
180
            enabled(Boolean): attribute to indicate the administrative state;
181
                              default is False.
182
            active(Boolean): attribute to indicate the operational state;
183
                             default is False.
184
            archived(Boolean): indicate the EVC has been deleted and is
185
                               archived; default is False.
186
            owner(str): The EVC owner. Default is None.
187
            priority(int): Service level provided in the request. Default is 0.
188
189
        Raises:
190
            ValueError: raised when object attributes are invalid.
191
192
        """
193 1
        self._validate(**kwargs)
194 1
        super().__init__()
195
196
        # required attributes
197 1
        self._id = kwargs.get('id', uuid4().hex)
198 1
        self.uni_a = kwargs.get('uni_a')
199 1
        self.uni_z = kwargs.get('uni_z')
200 1
        self.name = kwargs.get('name')
201
202
        # optional attributes
203 1
        self.start_date = get_time(kwargs.get('start_date')) or now()
204 1
        self.end_date = get_time(kwargs.get('end_date')) or None
205
206 1
        self.bandwidth = kwargs.get('bandwidth', 0)
207 1
        self.primary_links = Path(kwargs.get('primary_links', []))
208 1
        self.backup_links = Path(kwargs.get('backup_links', []))
209 1
        self.current_path = Path(kwargs.get('current_path', []))
210 1
        self.primary_path = Path(kwargs.get('primary_path', []))
211 1
        self.backup_path = Path(kwargs.get('backup_path', []))
212 1
        self.dynamic_backup_path = kwargs.get('dynamic_backup_path', False)
213 1
        self.creation_time = get_time(kwargs.get('creation_time')) or now()
214 1
        self.owner = kwargs.get('owner', None)
215 1
        self.priority = kwargs.get('priority', 0)
216 1
        self.circuit_scheduler = kwargs.get('circuit_scheduler', [])
217
218 1
        self.current_links_cache = set()
219 1
        self.primary_links_cache = set()
220 1
        self.backup_links_cache = set()
221
222 1
        self.archived = kwargs.get('archived', False)
223
224 1
        self._storehouse = StoreHouse(controller)
225
226 1
        if kwargs.get('active', False):
227 1
            self.activate()
228
        else:
229 1
            self.deactivate()
230
231 1
        if kwargs.get('enabled', False):
232 1
            self.enable()
233
        else:
234 1
            self.disable()
235
236
        # datetime of user request for a EVC (or datetime when object was
237
        # created)
238 1
        self.request_time = kwargs.get('request_time', now())
239
        # dict with the user original request (input)
240 1
        self._requested = kwargs
241
242 1
    def sync(self):
243
        """Sync this EVC in the storehouse."""
244 1
        self._storehouse.save_evc(self)
245 1
        log.info(f'EVC {self.id} was synced to the storehouse.')
246
247 1
    def update(self, **kwargs):
248
        """Update evc attributes.
249
250
        This method will raises an error trying to change the following
251
        attributes: [name, uni_a and uni_z]
252
253
        Returns:
254
            the values for enable and a path attribute, if exists and None
255
            otherwise
256
        Raises:
257
            ValueError: message with error detail.
258
259
        """
260 1
        enable, path = (None, None)
261 1
        for attribute, value in kwargs.items():
262 1
            if attribute in self.unique_attributes:
263 1
                raise ValueError(f'{attribute} can\'t be be updated.')
264 1
            if hasattr(self, attribute):
265 1
                if attribute in ('enable', 'enabled'):
266
                    if value:
267
                        self.enable()
268
                    else:
269
                        self.disable()
270
                    enable = value
271 1
                elif attribute in ('active', 'activate'):
272
                    if value:
273
                        self.activate()
274
                    else:
275
                        self.deactivate()
276
                else:
277 1
                    setattr(self, attribute, value)
278 1
                    if 'path' in attribute:
279 1
                        path = value
280
            else:
281
                raise ValueError(f'The attribute "{attribute}" is invalid.')
282 1
        self.sync()
283 1
        return enable, path
284
285 1
    def __repr__(self):
286
        """Repr method."""
287 1
        return f"EVC({self._id}, {self.name})"
288
289 1
    def _validate(self, **kwargs):
290
        """Do Basic validations.
291
292
        Verify required attributes: name, uni_a, uni_z
293
        Verify if the attributes uni_a and uni_z are valid.
294
295
        Raises:
296
            ValueError: message with error detail.
297
298
        """
299 1
        for attribute in self.unique_attributes:
300
301 1
            if attribute not in kwargs:
302 1
                raise ValueError(f'{attribute} is required.')
303
304 1
            if 'uni' in attribute:
305 1
                uni = kwargs.get(attribute)
306 1
                if not isinstance(uni, UNI):
307
                    raise ValueError(f'{attribute} is an invalid UNI.')
308
309 1
                if not uni.is_valid():
310 1
                    tag = uni.user_tag.value
311 1
                    message = f'VLAN tag {tag} is not available in {attribute}'
312 1
                    raise ValueError(message)
313
314 1
    def __eq__(self, other):
315
        """Override the default implementation."""
316 1
        if not isinstance(other, EVC):
317
            return False
318
319 1
        attrs_to_compare = ['name', 'uni_a', 'uni_z', 'owner', 'bandwidth']
320 1
        for attribute in attrs_to_compare:
321 1
            if getattr(other, attribute) != getattr(self, attribute):
322 1
                return False
323 1
        return True
324
325 1
    def as_dict(self):
326
        """Return a dictionary representing an EVC object."""
327 1
        evc_dict = {"id": self.id, "name": self.name,
328
                    "uni_a": self.uni_a.as_dict(),
329
                    "uni_z": self.uni_z.as_dict()}
330
331 1
        time_fmt = "%Y-%m-%dT%H:%M:%S"
332
333 1
        evc_dict["start_date"] = self.start_date
334 1
        if isinstance(self.start_date, datetime):
335 1
            evc_dict["start_date"] = self.start_date.strftime(time_fmt)
336
337 1
        evc_dict["end_date"] = self.end_date
338 1
        if isinstance(self.end_date, datetime):
339 1
            evc_dict["end_date"] = self.end_date.strftime(time_fmt)
340
341 1
        evc_dict['bandwidth'] = self.bandwidth
342 1
        evc_dict['primary_links'] = self.primary_links.as_dict()
343 1
        evc_dict['backup_links'] = self.backup_links.as_dict()
344 1
        evc_dict['current_path'] = self.current_path.as_dict()
345 1
        evc_dict['primary_path'] = self.primary_path.as_dict()
346 1
        evc_dict['backup_path'] = self.backup_path.as_dict()
347 1
        evc_dict['dynamic_backup_path'] = self.dynamic_backup_path
348
349
        # if self._requested:
350
        #     request_dict = self._requested.copy()
351
        #     request_dict['uni_a'] = request_dict['uni_a'].as_dict()
352
        #     request_dict['uni_z'] = request_dict['uni_z'].as_dict()
353
        #     request_dict['circuit_scheduler'] = self.circuit_scheduler
354
        #     evc_dict['_requested'] = request_dict
355
356 1
        evc_dict["request_time"] = self.request_time
357 1
        if isinstance(self.request_time, datetime):
358 1
            evc_dict["request_time"] = self.request_time.strftime(time_fmt)
359
360 1
        time = self.creation_time.strftime(time_fmt)
361 1
        evc_dict['creation_time'] = time
362
363 1
        evc_dict['owner'] = self.owner
364 1
        evc_dict['circuit_scheduler'] = [sc.as_dict()
365
                                         for sc in self.circuit_scheduler]
366
367 1
        evc_dict['active'] = self.is_active()
368 1
        evc_dict['enabled'] = self.is_enabled()
369 1
        evc_dict['archived'] = self.archived
370 1
        evc_dict['priority'] = self.priority
371
372 1
        return evc_dict
373
374 1
    @property
375
    def id(self):  # pylint: disable=invalid-name
376
        """Return this EVC's ID."""
377 1
        return self._id
378
379 1
    def archive(self):
380
        """Archive this EVC on deletion."""
381
        self.archived = True
382
383
384
# pylint: disable=fixme, too-many-public-methods
385 1
class EVCDeploy(EVCBase):
386
    """Class to handle the deploy procedures."""
387
388 1
    def create(self):
389
        """Create a EVC."""
390
391 1
    def discover_new_paths(self):
392
        """Discover new paths to satisfy this circuit and deploy it."""
393
        return DynamicPathManager.get_best_paths(self)
394
395 1
    def change_path(self):
396
        """Change EVC path."""
397
398 1
    def reprovision(self):
399
        """Force the EVC (re-)provisioning."""
400
401 1
    def is_affected_by_link(self, link):
402
        """Return True if this EVC has the given link on its current path."""
403
        return link in self.current_path
404
405 1
    def link_affected_by_interface(self, interface):
406
        """Return True if this EVC has the given link on its current path."""
407
        return self.current_path.link_affected_by_interface(interface)
408
409 1
    def is_backup_path_affected_by_link(self, link):
410
        """Return True if the backup path of this EVC uses the given link."""
411
        return link in self.backup_path
412
413
    # pylint: disable=invalid-name
414 1
    def is_primary_path_affected_by_link(self, link):
415
        """Return True if the primary path of this EVC uses the given link."""
416
        return link in self.primary_path
417
418 1
    def is_using_primary_path(self):
419
        """Verify if the current deployed path is self.primary_path."""
420
        return self.primary_path and (self.current_path == self.primary_path)
421
422 1
    def is_using_backup_path(self):
423
        """Verify if the current deployed path is self.backup_path."""
424
        return self.backup_path and (self.current_path == self.backup_path)
425
426 1
    def is_using_dynamic_path(self):
427
        """Verify if the current deployed path is a dynamic path."""
428
        if self.current_path and \
429
           not self.is_using_primary_path() and \
430
           not self.is_using_backup_path() and \
431
           self.current_path.status == EntityStatus.UP:
432
            return True
433
        return False
434
435 1
    def deploy_to_backup_path(self):
436
        """Deploy the backup path into the datapaths of this circuit.
437
438
        If the backup_path attribute is valid and up, this method will try to
439
        deploy this backup_path.
440
441
        If everything fails and dynamic_backup_path is True, then tries to
442
        deploy a dynamic path.
443
        """
444
        # TODO: Remove flows from current (cookies)
445 1
        if self.is_using_backup_path():
446
            # TODO: Log to say that cannot move backup to backup
447
            return True
448
449 1
        success = False
450 1
        if self.backup_path.status is EntityStatus.UP:
451 1
            success = self.deploy_to_path(self.backup_path)
452
453 1
        if success:
454 1
            return True
455
456 1
        if self.dynamic_backup_path:
457 1
            return self.deploy_to_path()
458
459 1
        return False
460
461 1
    def deploy_to_primary_path(self):
462
        """Deploy the primary path into the datapaths of this circuit.
463
464
        If the primary_path attribute is valid and up, this method will try to
465
        deploy this primary_path.
466
        """
467
        # TODO: Remove flows from current (cookies)
468 1
        if self.is_using_primary_path():
469
            # TODO: Log to say that cannot move primary to primary
470
            return True
471
472 1
        if self.primary_path.status is EntityStatus.UP:
473 1
            return self.deploy_to_path(self.primary_path)
474 1
        return False
475
476 1
    def deploy(self):
477
        """Deploy EVC to best path.
478
479
        Best path can be the primary path, if available. If not, the backup
480
        path, and, if it is also not available, a dynamic path.
481
        """
482 1
        if self.archived:
483
            return False
484 1
        self.enable()
485 1
        success = self.deploy_to_primary_path()
486 1
        if not success:
487 1
            success = self.deploy_to_backup_path()
488
489 1
        return success
490
491 1
    @staticmethod
492
    def get_path_status(path):
493
        """Check for the current status of a path.
494
495
        If any link in this path is down, the path is considered down.
496
        """
497
        if not path:
498
            return EntityStatus.DISABLED
499
500
        for link in path:
501
            if link.status is not EntityStatus.UP:
502
                return link.status
503
        return EntityStatus.UP
504
505
#    def discover_new_path(self):
506
#        # TODO: discover a new path to satisfy this circuit and deploy
507
508 1
    def remove(self):
509
        """Remove EVC path and disable it."""
510 1
        self.remove_current_flows()
511
        self.disable()
512
        self.sync()
513
514 1
    def remove_current_flows(self):
515
        """Remove all flows from current path."""
516 1
        switches = set()
517
518 1
        switches.add(self.uni_a.interface.switch)
519 1
        switches.add(self.uni_z.interface.switch)
520 1
        for link in self.current_path:
521 1
            switches.add(link.endpoint_a.switch)
522 1
            switches.add(link.endpoint_b.switch)
523
524 1
        match = {'cookie': self.get_cookie(),
525
                 'cookie_mask': 18446744073709551615}
526
527 1
        for switch in switches:
528 1
            self._send_flow_mods(switch, [match], 'delete')
529
530 1
        self.current_path.make_vlans_available()
531 1
        self.current_path = Path([])
532 1
        self.deactivate()
533 1
        self.sync()
534
535 1
    @staticmethod
536 1
    def links_zipped(path=None):
537
        """Return an iterator which yields pairs of links in order."""
538 1
        if not path:
539
            return []
540 1
        return zip(path[:-1], path[1:])
541
542 1
    def should_deploy(self, path=None):
543
        """Verify if the circuit should be deployed."""
544 1
        if not path:
545 1
            log.debug("Path is empty.")
546 1
            return False
547
548 1
        if not self.is_enabled():
549 1
            log.debug(f'{self} is disabled.')
550 1
            return False
551
552 1
        if not self.is_active():
553 1
            log.debug(f'{self} will be deployed.')
554 1
            return True
555
556 1
        return False
557
558 1
    def deploy_to_path(self, path=None):
559
        """Install the flows for this circuit.
560
561
        Procedures to deploy:
562
563
        0. Remove current flows installed
564
        1. Decide if will deploy "path" or discover a new path
565
        2. Choose vlan
566
        3. Install NNI flows
567
        4. Install UNI flows
568
        5. Activate
569
        6. Update current_path
570
        7. Update links caches(primary, current, backup)
571
572
        """
573 1
        self.remove_current_flows()
574
        use_path = path
575
        if self.should_deploy(use_path):
576
            try:
577
                use_path.choose_vlans()
578
            except KytosNoTagAvailableError:
579
                use_path = None
580
        else:
581
            for use_path in self.discover_new_paths():
582
                try:
583
                    use_path.choose_vlans()
584
                    break
585
                except KytosNoTagAvailableError:
586
                    pass
587
            else:
588
                use_path = None
589
590
        if use_path:
591
            self._install_nni_flows(use_path)
592
            self._install_uni_flows(use_path)
593
            self.activate()
594
            self.current_path = use_path
595
            self.sync()
596
            log.info(f"{self} was deployed.")
597
            return True
598
        return False
599
600 1
    def _install_direct_uni_flows(self):
601
        """Install flows connecting two UNIs.
602
603
        This case happens when the circuit is between UNIs in the
604
        same switch.
605
        """
606
        vlan_a = self.uni_a.user_tag.value if self.uni_a.user_tag else None
607
        vlan_z = self.uni_z.user_tag.value if self.uni_z.user_tag else None
608
609
        flow_mod_az = self._prepare_flow_mod(self.uni_a.interface,
610
                                             self.uni_z.interface)
611
        flow_mod_za = self._prepare_flow_mod(self.uni_z.interface,
612
                                             self.uni_a.interface)
613
614
        if vlan_a and vlan_z:
615
            flow_mod_az['match']['dl_vlan'] = vlan_a
616
            flow_mod_za['match']['dl_vlan'] = vlan_z
617
            flow_mod_az['actions'].insert(0, {'action_type': 'set_vlan',
618
                                              'vlan_id': vlan_z})
619
            flow_mod_za['actions'].insert(0, {'action_type': 'set_vlan',
620
                                              'vlan_id': vlan_a})
621
        elif vlan_a:
622
            flow_mod_az['match']['dl_vlan'] = vlan_a
623
            flow_mod_az['actions'].insert(0, {'action_type': 'pop_vlan'})
624
            flow_mod_za['actions'].insert(0, {'action_type': 'set_vlan',
625
                                              'vlan_id': vlan_a})
626
        elif vlan_z:
627
            flow_mod_za['match']['dl_vlan'] = vlan_z
628
            flow_mod_za['actions'].insert(0, {'action_type': 'pop_vlan'})
629
            flow_mod_az['actions'].insert(0, {'action_type': 'set_vlan',
630
                                              'vlan_id': vlan_z})
631
        self._send_flow_mods(self.uni_a.interface.switch,
632
                             [flow_mod_az, flow_mod_za])
633
634 1
    def _install_nni_flows(self, path=None):
635
        """Install NNI flows."""
636 1
        for incoming, outcoming in self.links_zipped(path):
637 1
            in_vlan = incoming.get_metadata('s_vlan').value
638 1
            out_vlan = outcoming.get_metadata('s_vlan').value
639
640 1
            flows = []
641
            # Flow for one direction
642 1
            flows.append(self._prepare_nni_flow(incoming.endpoint_b,
643
                                                outcoming.endpoint_a,
644
                                                in_vlan, out_vlan))
645
646
            # Flow for the other direction
647 1
            flows.append(self._prepare_nni_flow(outcoming.endpoint_a,
648
                                                incoming.endpoint_b,
649
                                                out_vlan, in_vlan))
650 1
            self._send_flow_mods(incoming.endpoint_b.switch, flows)
651
652 1
    def _install_uni_flows(self, path=None):
653
        """Install UNI flows."""
654 1
        if not path:
655
            log.info('install uni flows without path.')
656
            return
657
658
        # Determine VLANs
659 1
        in_vlan_a = self.uni_a.user_tag.value if self.uni_a.user_tag else None
660 1
        out_vlan_a = path[0].get_metadata('s_vlan').value
661
662 1
        in_vlan_z = self.uni_z.user_tag.value if self.uni_z.user_tag else None
663 1
        out_vlan_z = path[-1].get_metadata('s_vlan').value
664
665
        # Flows for the first UNI
666 1
        flows_a = []
667
668
        # Flow for one direction, pushing the service tag
669 1
        push_flow = self._prepare_push_flow(self.uni_a.interface,
670
                                            path[0].endpoint_a,
671
                                            in_vlan_a, out_vlan_a, in_vlan_z)
672 1
        flows_a.append(push_flow)
673
674
        # Flow for the other direction, popping the service tag
675 1
        pop_flow = self._prepare_pop_flow(path[0].endpoint_a,
676
                                          self.uni_a.interface, out_vlan_a)
677 1
        flows_a.append(pop_flow)
678
679 1
        self._send_flow_mods(self.uni_a.interface.switch, flows_a)
680
681
        # Flows for the second UNI
682 1
        flows_z = []
683
684
        # Flow for one direction, pushing the service tag
685 1
        push_flow = self._prepare_push_flow(self.uni_z.interface,
686
                                            path[-1].endpoint_b,
687
                                            in_vlan_z, out_vlan_z, in_vlan_a)
688 1
        flows_z.append(push_flow)
689
690
        # Flow for the other direction, popping the service tag
691 1
        pop_flow = self._prepare_pop_flow(path[-1].endpoint_b,
692
                                          self.uni_z.interface, out_vlan_z)
693 1
        flows_z.append(pop_flow)
694
695 1
        self._send_flow_mods(self.uni_z.interface.switch, flows_z)
696
697 1
    @staticmethod
698 1
    def _send_flow_mods(switch, flow_mods, command='flows'):
699
        """Send a flow_mod list to a specific switch.
700
701
        Args:
702
            switch(Switch): The target of flows.
703
            flow_mods(dict): Python dictionary with flow_mods.
704
            command(str): By default is 'flows'. To remove a flow is 'remove'.
705
706
        """
707 1
        endpoint = f'{settings.MANAGER_URL}/{command}/{switch.id}'
708
709 1
        data = {"flows": flow_mods}
710 1
        requests.post(endpoint, json=data)
711
712 1
    def get_cookie(self):
713
        """Return the cookie integer from evc id."""
714 1
        value = self.id[len(self.id)//2:]
715 1
        return int(value, 16)
716
717 1
    def _prepare_flow_mod(self, in_interface, out_interface):
718
        """Prepare a common flow mod."""
719 1
        default_action = {"action_type": "output",
720
                          "port": out_interface.port_number}
721
722 1
        flow_mod = {"match": {"in_port": in_interface.port_number},
723
                    "cookie": self.get_cookie(),
724
                    "actions": [default_action]}
725
726 1
        return flow_mod
727
728 1
    def _prepare_nni_flow(self,
729
                          in_interface, out_interface, in_vlan, out_vlan):
730
        """Create NNI flows."""
731 1
        flow_mod = self._prepare_flow_mod(in_interface, out_interface)
732 1
        flow_mod['match']['dl_vlan'] = in_vlan
733
734 1
        new_action = {"action_type": "set_vlan",
735
                      "vlan_id": out_vlan}
736 1
        flow_mod["actions"].insert(0, new_action)
737
738 1
        return flow_mod
739
740 1
    def _prepare_push_flow(self, *args):
741
        """Prepare push flow.
742
743
        Arguments:
744
            in_interface(str): Interface input.
745
            out_interface(str): Interface output.
746
            in_vlan(str): Vlan input.
747
            out_vlan(str): Vlan output.
748
            new_in_vlan(str): Interface input.
749
750
        Return:
751
            dict: An python dictionary representing a FlowMod
752
753
        """
754
        # assign all arguments
755 1
        in_interface, out_interface, in_vlan, out_vlan, new_in_vlan = args
756
757 1
        flow_mod = self._prepare_flow_mod(in_interface, out_interface)
758
759
        # the service tag must be always pushed
760 1
        new_action = {"action_type": "set_vlan", "vlan_id": out_vlan}
761 1
        flow_mod["actions"].insert(0, new_action)
762
763 1
        new_action = {"action_type": "push_vlan", "tag_type": "s"}
764 1
        flow_mod["actions"].insert(0, new_action)
765
766 1
        if in_vlan:
767
            # if in_vlan is set, it must be included in the match
768 1
            flow_mod['match']['dl_vlan'] = in_vlan
769 1
        if new_in_vlan:
770
            # new_in_vlan is set, so an action to set it is necessary
771 1
            new_action = {"action_type": "set_vlan", "vlan_id": new_in_vlan}
772 1
            flow_mod["actions"].insert(0, new_action)
773 1
            if not in_vlan:
774
                # new_in_vlan is set, but in_vlan is not, so there was no
775
                # vlan set; then it is set now
776 1
                new_action = {"action_type": "push_vlan", "tag_type": "c"}
777 1
                flow_mod["actions"].insert(0, new_action)
778 1
        elif in_vlan:
779
            # in_vlan is set, but new_in_vlan is not, so the existing vlan
780
            # must be removed
781 1
            new_action = {"action_type": "pop_vlan"}
782 1
            flow_mod["actions"].insert(0, new_action)
783 1
        return flow_mod
784
785 1
    def _prepare_pop_flow(self, in_interface, out_interface, out_vlan):
786
        """Prepare pop flow."""
787 1
        flow_mod = self._prepare_flow_mod(in_interface, out_interface)
788 1
        flow_mod['match']['dl_vlan'] = out_vlan
789 1
        new_action = {"action_type": "pop_vlan"}
790 1
        flow_mod["actions"].insert(0, new_action)
791 1
        return flow_mod
792
793
794 1
class LinkProtection(EVCDeploy):
795
    """Class to handle link protection."""
796
797 1
    def is_affected_by_link(self, link=None):
798
        """Verify if the current path is affected by link down event."""
799
        return self.current_path.is_affected_by_link(link)
800
801 1
    def is_using_primary_path(self):
802
        """Verify if the current deployed path is self.primary_path."""
803 1
        return self.current_path == self.primary_path
804
805 1
    def is_using_backup_path(self):
806
        """Verify if the current deployed path is self.backup_path."""
807 1
        return self.current_path == self.backup_path
808
809 1
    def is_using_dynamic_path(self):
810
        """Verify if the current deployed path is dynamic."""
811 1
        if self.current_path and \
812
           not self.is_using_primary_path() and \
813
           not self.is_using_backup_path() and \
814
           self.current_path.status is EntityStatus.UP:
815
            return True
816 1
        return False
817
818 1
    def deploy_to(self, path_name=None, path=None):
819
        """Create a deploy to path."""
820 1
        if self.current_path == path:
821 1
            log.debug(f'{path_name} is equal to current_path.')
822 1
            return True
823
824 1
        if path.status is EntityStatus.UP:
825 1
            return self.deploy_to_path(path)
826
827 1
        return False
828
829 1
    def handle_link_up(self, link):
830
        """Handle circuit when link down.
831
832
        Args:
833
            link(Link): Link affected by link.down event.
834
835
        """
836 1
        if self.is_using_primary_path():
837 1
            return True
838
839 1
        success = False
840 1
        if self.primary_path.is_affected_by_link(link):
841 1
            success = self.deploy_to_primary_path()
842
843 1
        if success:
844 1
            return True
845
846
        # We tried to deploy(primary_path) without success.
847
        # And in this case is up by some how. Nothing to do.
848 1
        if self.is_using_backup_path() or self.is_using_dynamic_path():
849
            return True
850
851
        # In this case, probably the circuit is not being used and
852
        # we can move to backup
853 1
        if self.backup_path.is_affected_by_link(link):
854 1
            success = self.deploy_to_backup_path()
855
856 1
        if success:
857 1
            return True
858
859
        # In this case, the circuit is not being used and we should
860
        # try a dynamic path
861
        if self.dynamic_backup_path:
862
            return self.deploy_to_path()
863
864
        return True
865
866 1
    def handle_link_down(self):
867
        """Handle circuit when link down.
868
869
        Returns:
870
            bool: True if the re-deploy was successly otherwise False.
871
872
        """
873 1
        success = False
874 1
        if self.is_using_primary_path():
875 1
            success = self.deploy_to_backup_path()
876 1
        elif self.is_using_backup_path():
877 1
            success = self.deploy_to_primary_path()
878
879 1
        if not success and self.dynamic_backup_path:
880 1
            success = self.deploy_to_path()
881
882 1
        if success:
883 1
            log.debug(f"{self} deployed after link down.")
884
        else:
885 1
            self.deactivate()
886 1
            self.current_path = Path([])
887 1
            self.sync()
888 1
            log.debug(f'Failed to re-deploy {self} after link down.')
889
890 1
        return success
891
892
893 1
class EVC(LinkProtection):
894
    """Class that represents a E-Line Virtual Connection."""
895