pypush-plus-plus/ids/__init__.py

from base64 import b64encode
import apns
from . import profile, _helpers, identity

class IDSUser:
    # Sets self.user_id and self._auth_token
    def _authenticate_for_token(
        self, username: str, password: str, factor_callback: callable = None
    ):
        self.user_id, self._auth_token = profile._get_auth_token(
            username, password, factor_callback
        )

    # Sets self._auth_keypair using self.user_id and self._auth_token
    def _authenticate_for_cert(self):
        self._auth_keypair = profile._get_auth_cert(self.user_id, self._auth_token)

    # Factor callback will be called if a 2FA code is necessary
    def __init__(
        self,
        push_connection: apns.APNSConnection,
    ):
        self.push_connection = push_connection
        self._push_keypair = _helpers.KeyPair(self.push_connection.private_key, self.push_connection.cert)

    def __str__(self):
        return f"IDSUser(user_id={self.user_id}, handles={self.handles}, push_token={b64encode(self.push_connection.token).decode()})"
    
    # Authenticates with a username and password, to create a brand new authentication keypair
    def authenticate(self, username: str, password: str, factor_callback: callable = None):
        self._authenticate_for_token(username, password, factor_callback)
        self._authenticate_for_cert()
        self.handles = profile._get_handles(
            b64encode(self.push_connection.token),
            self.user_id,
            self._auth_keypair,
            self._push_keypair,
        )

    # Uses an existing authentication keypair
    def restore_authentication(self, auth_keypair: _helpers.KeyPair, user_id: str, handles: dict):
        self._auth_keypair = auth_keypair
        self.user_id = user_id
        self.handles = handles
        
    
    # This is a separate call so that the user can make sure the first part succeeds before asking for validation data
    def register(self, validation_data: str):
        resp = identity.register_request(
            b64encode(self.push_connection.token),
            self.handles,
            self.user_id,
            self._auth_keypair,
            self._push_keypair,
            validation_data
        )
        print(resp)