FilesExpand file tree

 main

/

auth_command.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

78 lines (60 loc) · 2.75 KB

 main

/

auth_command.py

Copy path

Top

File metadata and controls

• Code
• Blame

78 lines (60 loc) · 2.75 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

import click

import traceback

from cycode.cli.models import CliResult, CliErrors, CliError

from cycode.cli.printers import ConsolePrinter

from cycode.cli.auth.auth_manager import AuthManager

from cycode.cli.user_settings.credentials_manager import CredentialsManager

from cycode.cli.exceptions.custom_exceptions import AuthProcessError, NetworkError, HttpUnauthorizedError

from cycode.cyclient import logger

from cycode.cyclient.cycode_token_based_client import CycodeTokenBasedClient

@click.group(invoke_without_command=True)

@click.pass_context

def authenticate(context: click.Context):

""" Authenticates your machine to associate CLI with your cycode account """

if context.invoked_subcommand is not None:

# if it is a subcommand do nothing

return

try:

logger.debug('Starting authentication process')

auth_manager = AuthManager()

auth_manager.authenticate()

result = CliResult(success=True, message='Successfully logged into cycode')

ConsolePrinter(context).print_result(result)

except Exception as e:

_handle_exception(context, e)

@authenticate.command(name='check')

@click.pass_context

def authorization_check(context: click.Context):

""" Check your machine associating CLI with your cycode account """

printer = ConsolePrinter(context)

passed_auth_check_res = CliResult(success=True, message='You are authorized')

failed_auth_check_res = CliResult(success=False, message='You are not authorized')

client_id, client_secret = CredentialsManager().get_credentials()

if not client_id or not client_secret:

return printer.print_result(failed_auth_check_res)

try:

if CycodeTokenBasedClient(client_id, client_secret).api_token:

return printer.print_result(passed_auth_check_res)

except (NetworkError, HttpUnauthorizedError):

if context.obj['verbose']:

click.secho(f'Error: {traceback.format_exc()}', fg='red', nl=False)

return printer.print_result(failed_auth_check_res)

def _handle_exception(context: click.Context, e: Exception):

if context.obj['verbose']:

click.secho(f'Error: {traceback.format_exc()}', fg='red', nl=False)

errors: CliErrors = {

AuthProcessError: CliError(

code='auth_error',

message='Authentication failed. Please try again later using the command `cycode auth`'

),

NetworkError: CliError(

code='cycode_error',

message='Authentication failed. Please try again later using the command `cycode auth`'

),

}

error = errors.get(type(e))

if error:

return ConsolePrinter(context).print_error(error)

if isinstance(e, click.ClickException):

raise e

raise click.ClickException(str(e))