blob: a291258db2992f917c45b252e7c41f9dddaa0e3d (
plain) (
blame)
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
|
# -*- coding: utf-8 -*-
import abc
import six
from . import tracing
YDB_AUTH_TICKET_HEADER = "x-ydb-auth-ticket"
@six.add_metaclass(abc.ABCMeta)
class AbstractCredentials(object):
"""
An abstract class that provides auth metadata
"""
@six.add_metaclass(abc.ABCMeta)
class Credentials(object):
def __init__(self, tracer=None):
self.tracer = tracer if tracer is not None else tracing.Tracer(None)
@abc.abstractmethod
def auth_metadata(self):
"""
:return: An iterable with auth metadata
"""
pass
class AnonymousCredentials(Credentials):
@staticmethod
def auth_metadata():
return []
class AuthTokenCredentials(Credentials):
def __init__(self, token):
self._token = token
def auth_metadata(self):
return [(YDB_AUTH_TICKET_HEADER, self._token)]
class AccessTokenCredentials(Credentials):
def __init__(self, token):
self._token = token
def auth_metadata(self):
return [(YDB_AUTH_TICKET_HEADER, self._token)]
|