Coverage for mindsdb / integrations / handlers / athena_handler / connection_args.py: 0%

1from collections import OrderedDict 

2 

3from mindsdb.integrations.libs.const import HANDLER_CONNECTION_ARG_TYPE as ARG_TYPE 

4 

5 

6connection_args = OrderedDict( 

7 aws_access_key_id={ 

8 'type': ARG_TYPE.STR, 

9 'description': 'The access key for the AWS account.' 

10 }, 

11 aws_secret_access_key={ 

12 'type': ARG_TYPE.STR, 

13 'description': 'The secret key for the AWS account.', 

14 'secret': True 

15 }, 

16 region_name={ 

17 'type': ARG_TYPE.STR, 

18 'description': 'The AWS region where the Athena tables are created.' 

19 }, 

20 database={ 

21 'type': ARG_TYPE.STR, 

22 'description': 'The name of the Athena database.' 

23 }, 

24 workgroup={ 

25 'type': ARG_TYPE.STR, 

26 'description': 'The Athena Workgroup' 

27 }, 

28 catalog={ 

29 'type': ARG_TYPE.STR, 

30 'description': 'The AWS Data Catalog' 

31 }, 

32 results_output_location={ 

33 'type': ARG_TYPE.STR, 

34 'description': 'The Athena Query Results Output Location s3://bucket-path/athena-query-results' 

35 }, 

36 check_interval={ 

37 'type': ARG_TYPE.INT, 

38 'description': 'The interval in seconds to check Athena for query results. Default is 0 seconds.' 

39 } 

40) 

41 

42connection_args_example = OrderedDict( 

43 aws_access_key_id='<default>', 

44 aws_secret_access_key='<default>', 

45 region_name='us-east-1', 

46 catalog='AwsDataCatalog', 

47 database='default', 

48 workgroup='primary', 

49 results_output_location='s3://<bucket>/athena-query-results/', 

50 check_interval=0 

51)