Coverage for mindsdb / integrations / handlers / postgres_handler / connection_args.py: 100%

1from collections import OrderedDict 

2 

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

4 

5 

6connection_args = OrderedDict( 

7 user={ 

8 'type': ARG_TYPE.STR, 

9 'description': 'The user name used to authenticate with the PostgreSQL server.', 

10 'required': True, 

11 'label': 'User' 

12 }, 

13 password={ 

14 'type': ARG_TYPE.PWD, 

15 'description': 'The password to authenticate the user with the PostgreSQL server.', 

16 'required': True, 

17 'label': 'Password', 

18 'secret': True 

19 }, 

20 database={ 

21 'type': ARG_TYPE.STR, 

22 'description': 'The database name to use when connecting with the PostgreSQL server.', 

23 'required': True, 

24 'label': 'Database' 

25 }, 

26 host={ 

27 'type': ARG_TYPE.STR, 

28 'description': 'The host name or IP address of the PostgreSQL server. NOTE: use \'127.0.0.1\' instead of \'localhost\' to connect to local server.', 

29 'required': True, 

30 'label': 'Host' 

31 }, 

32 port={ 

33 'type': ARG_TYPE.INT, 

34 'description': 'The TCP/IP port of the PostgreSQL server. Must be an integer.', 

35 'required': True, 

36 'label': 'Port' 

37 }, 

38 schema={ 

39 'type': ARG_TYPE.STR, 

40 'description': 'The schema in which objects are searched first.', 

41 'required': False, 

42 'label': 'Schema' 

43 }, 

44 sslmode={ 

45 'type': ARG_TYPE.STR, 

46 'description': 'sslmode that will be used for connection.', 

47 'required': False, 

48 'label': 'sslmode' 

49 }, 

50 connection_parameters={ 

51 'type': ARG_TYPE.DICT, 

52 'description': 'Connection string parameters', 

53 'required': False, 

54 'label': 'connection_parameters' 

55 } 

56) 

57 

58connection_args_example = OrderedDict( 

59 host='127.0.0.1', 

60 port=5432, 

61 user='root', 

62 schema='public', 

63 password='password', 

64 database='database' 

65)