import boto3
from botocore.credentials import Credentials


__all__ = ["AWSMixin"]


class AWSMixin:
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self._boto_session = None

    def _pre_run(self):
        super()._pre_run()
        self._boto_session = boto3.Session()

    @property
    def aws_credentials(self) -> Credentials:
        return self._boto_session.get_credentials()