Source code for divera.api.v2.auth

import json

from divera.api import endpointwrappers as _epw


class Login(_epw.FunctionalEndPoint):
    method = 'POST'
    path = '/api/v2/auth/login'
    headers = {
        'accept': 'application/json',
        'Content-Type': 'application/json',
    }

    def __init__(
            self,
            username: str,
            password: str,
    ):
        self.username = username
        self.password = password

    def __call__(
            self,
            base_url: str,
            access_token: str = None,
            *args, **kwargs,
    ) -> [list, dict]:
        args_ = (
            base_url.rstrip('/') + self.path,
        )
        kwargs_ = {
            'headers': self.headers,
            'data': json.dumps({
                'Login': {
                    'username': self.username,
                    'password': self.password,
                    'jwt': False,
                }
            }),
        }
        return args_, kwargs_


class JWT(_epw.FunctionalEndPoint):
    method = 'GET'
    path = '/api/v2/auth/jwt'

    def __call__(
            self,
            base_url: str,
            access_token: str,
            *args, **kwargs
    ):
        args_ = (
            base_url.rstrip('/') + self.path + f'?accesskey={access_token}',
        )
        kwargs_ = {
        }
        return args_, kwargs_