Skip to content

Commit

Permalink
Remove TrustDomain class
Browse files Browse the repository at this point in the history
  • Loading branch information
BrunoCoimbra committed Mar 6, 2024
1 parent 435a3cc commit 0842b5b
Showing 1 changed file with 7 additions and 18 deletions.
25 changes: 7 additions & 18 deletions lib/credentials.py
Original file line number Diff line number Diff line change
Expand Up @@ -163,17 +163,6 @@ def __repr__(self) -> str:
return f"{self.__class__.__name__}.{self.name}"


# TODO: Define better trust domains
class TrustDomain(enum.Enum):
GRID = "grid"

def __str__(self) -> str:
return self.value

def __repr__(self) -> str:
return f"{self.__class__.__name__}.{self.name}"


class Credential(ABC, Generic[T]):
cred_type: Optional[CredentialType] = None
classad_attribute: Optional[str] = None
Expand All @@ -184,7 +173,7 @@ def __init__(
string: Optional[bytes] = None,
path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
) -> None:
self._string = None
Expand Down Expand Up @@ -339,14 +328,14 @@ def __init__(
private_string: Optional[bytes] = None,
private_path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
) -> None:
if len(self.__class__.__bases__) < 2 or not issubclass(self.__class__.__bases__[1], Credential):
raise CredentialError("CredentialPair requires a Credential subclass as second base class")

credential_class = self.__class__.__bases__[1]
super(credential_class, self).__init__(
super(credential_class, self).__init__( # pylint: disable=bad-super-call # type: ignore[call-arg]
string, path, purpose, trust_domain, security_class
) # pylint: disable=bad-super-call # type: ignore[call-arg]
self.private_credential = credential_class(private_string, private_path, purpose, trust_domain, security_class)
Expand Down Expand Up @@ -611,7 +600,7 @@ def __init__(
private_string: Optional[bytes] = None,
private_path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
) -> None:
super().__init__(string, path, private_string, private_path, purpose, trust_domain, security_class)
Expand All @@ -629,7 +618,7 @@ def __init__(
private_string: Optional[bytes] = None,
private_path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
) -> None:
super().__init__(string, path, private_string, private_path, purpose, trust_domain, security_class)
Expand Down Expand Up @@ -887,7 +876,7 @@ def create_credential(
string: Optional[bytes] = None,
path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
cred_type: Optional[CredentialType] = None,
) -> Credential:
Expand All @@ -910,7 +899,7 @@ def create_credential_pair(
private_string: Optional[bytes] = None,
private_path: Optional[str] = None,
purpose: Optional[CredentialPurpose] = None,
trust_domain: Optional[TrustDomain] = None,
trust_domain: Optional[str] = None,
security_class: Optional[str] = None,
cred_type: Optional[CredentialPairType] = None,
) -> CredentialPair:
Expand Down

0 comments on commit 0842b5b

Please sign in to comment.