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
|
"""Decorators for Kytos-utils."""
import logging
import os
import sys
from getpass import getpass
import requests
from kytos.utils.config import KytosConfig
LOG = logging.getLogger(__name__)
# This class is used as decorator, so this class name is lowercase and the
# invalid-name warning from pylint is disabled below.
class kytos_auth: # pylint: disable=invalid-name
"""Class to be used as decorator to require authentication."""
def __init__(self, func):
"""Init method.
Save the function on the func attribute and bootstrap a new config.
"""
self.func = func
self.config = KytosConfig().config
self.cls = None
self.obj = None
def __call__(self, *args, **kwargs):
"""Code run when func is called."""
if not (self.config.has_option('napps', 'api') and
self.config.has_option('napps', 'repo')):
uri = input("Enter the kytos napps server address: ")
self.config.set('napps', 'api', os.path.join(uri, 'api', ''))
self.config.set('napps', 'repo', os.path.join(uri, 'repo', ''))
if not self.config.has_option('auth', 'user'):
user = input("Enter the username: ")
self.config.set('auth', 'user', user)
else:
user = self.config.get('auth', 'user')
if not self.config.has_option('auth', 'token'):
token = self.authenticate()
else:
token = self.config.get('auth', 'token')
# Ignore private attribute warning. We don't wanna make it public only
# because of a decorator.
config = self.obj._config # pylint: disable=protected-access
config.set('auth', 'user', user)
config.set('auth', 'token', token)
self.func.__call__(self.obj, *args, **kwargs)
def __get__(self, instance, owner):
"""Deal with owner class."""
self.cls = owner
self.obj = instance
return self.__call__
def authenticate(self):
"""Check the user authentication."""
endpoint = os.path.join(self.config.get('napps', 'api'), 'auth', '')
username = self.config.get('auth', 'user')
password = getpass("Enter the password for {}: ".format(username))
response = requests.get(endpoint, auth=(username, password))
if response.status_code != 201:
LOG.error(response.content)
LOG.error('ERROR: %s: %s', response.status_code, response.reason)
sys.exit(1)
else:
data = response.json()
KytosConfig().save_token(username, data.get('hash'))
return data.get('hash')
|
