mirror of
https://gitcode.com/gh_mirrors/eas/EasyFace.git
synced 2026-04-08 21:30:17 +00:00
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
# Copyright (c) Alibaba, Inc. and its affiliates.
|
|
|
|
from http.cookiejar import CookieJar
|
|
from typing import Tuple
|
|
|
|
|
|
class BaseAuthConfig(object):
|
|
"""Base authorization config class."""
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
self.cookies = cookies
|
|
self.git_token = git_token
|
|
self.user_info = user_info
|
|
|
|
|
|
class OssAuthConfig(BaseAuthConfig):
|
|
"""The authorization config for oss dataset."""
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
super().__init__(cookies=cookies,
|
|
git_token=git_token,
|
|
user_info=user_info)
|
|
|
|
|
|
class MaxComputeAuthConfig(BaseAuthConfig):
|
|
# TODO: MaxCompute dataset to be supported.
|
|
def __init__(self, cookies: CookieJar, git_token: str,
|
|
user_info: Tuple[str, str]):
|
|
super().__init__(cookies=cookies,
|
|
git_token=git_token,
|
|
user_info=user_info)
|
|
|
|
self.max_compute_grant_cmd = None
|