|
| 1 | +import click |
| 2 | +from meta.log import logger |
| 3 | +from meta.utils.store_control import StoreControlClient |
| 4 | +from meta.utils.verify import Verify |
| 5 | + |
| 6 | +@click.command() |
| 7 | +@click.option('--username', '-u', default=None, help='Your AWS access key') |
| 8 | +@click.option('--password', '-p', default=None, help='Your AWS secret key') |
| 9 | +@click.option('--bucket', '-b', default=None, help='Desired bucket name') |
| 10 | +def configure(username, password, bucket): |
| 11 | + """ Logs in to Meta""" |
| 12 | + logger.info("Please log in using your AWS credentials.") |
| 13 | + if not username: |
| 14 | + logger.debug("Prompting for Access Key") |
| 15 | + username = click.prompt('AWS Access Key ID', type=str, hide_input=False) |
| 16 | + access_key = username.strip() |
| 17 | + |
| 18 | + if not password: |
| 19 | + logger.debug("Prompting for Secret Key") |
| 20 | + password = click.prompt('AWS Secret Access Key', type=str, hide_input=False) |
| 21 | + secret_key = password.strip() |
| 22 | + |
| 23 | + if not bucket: |
| 24 | + logger.debug("Prompting for bucket name") |
| 25 | + bucket = click.prompt('Bucket Name (e.g. company-name)', type=str, hide_input=False) |
| 26 | + bucket = bucket.strip() |
| 27 | + |
| 28 | + success, creds = Verify(access_key, secret_key).verify_aws(bucket) |
| 29 | + if success: |
| 30 | + StoreControlClient().save_config(creds) |
| 31 | + logger.info("Login Successful.") |
| 32 | + else: |
| 33 | + logger.error("Login error, please try again") |
| 34 | + |
| 35 | +def logout(): |
| 36 | + """ Logs out of Snark AI""" |
| 37 | + # TODO remove scripts |
| 38 | + TokenManager.purge_token() |
0 commit comments