|
13 | 13 |
|
14 | 14 | """This module contains model that stores Robot Framework variables.""" |
15 | 15 |
|
16 | | -from distutils.util import strtobool |
17 | 16 | from os import path |
18 | 17 | from typing import Optional, Union, Dict, Tuple, Any, List |
19 | 18 | from warnings import warn |
20 | 19 |
|
21 | 20 | from reportportal_client import OutputType, ClientType |
| 21 | +from reportportal_client.helpers import to_bool |
22 | 22 | from reportportal_client.logs import MAX_LOG_BATCH_PAYLOAD_SIZE |
23 | 23 | from robot.libraries.BuiltIn import BuiltIn, RobotNotRunningError |
24 | 24 |
|
@@ -76,26 +76,23 @@ def __init__(self) -> None: |
76 | 76 |
|
77 | 77 | self._pabot_pool_id = None |
78 | 78 | self._pabot_used = None |
79 | | - self.attach_log = bool(strtobool(get_variable( |
80 | | - 'RP_ATTACH_LOG', default='False'))) |
81 | | - self.attach_report = bool(strtobool(get_variable('RP_ATTACH_REPORT', default='False'))) |
82 | | - self.attach_xunit = bool(strtobool(get_variable('RP_ATTACH_XUNIT', default='False'))) |
| 79 | + self.attach_log = to_bool(get_variable('RP_ATTACH_LOG', default='False')) |
| 80 | + self.attach_report = to_bool(get_variable('RP_ATTACH_REPORT', default='False')) |
| 81 | + self.attach_xunit = to_bool(get_variable('RP_ATTACH_XUNIT', default='False')) |
83 | 82 | self.launch_attributes = get_variable('RP_LAUNCH_ATTRIBUTES', default='').split() |
84 | 83 | self.launch_id = get_variable('RP_LAUNCH_UUID') |
85 | 84 | self.launch_doc = get_variable('RP_LAUNCH_DOC') |
86 | 85 | self.log_batch_size = int(get_variable( |
87 | 86 | 'RP_LOG_BATCH_SIZE', default='20')) |
88 | 87 | self.mode = get_variable('RP_MODE') |
89 | 88 | self.pool_size = int(get_variable('RP_MAX_POOL_SIZE', default='50')) |
90 | | - self.rerun = bool(strtobool(get_variable( |
91 | | - 'RP_RERUN', default='False'))) |
| 89 | + self.rerun = to_bool(get_variable('RP_RERUN', default='False')) |
92 | 90 | self.rerun_of = get_variable('RP_RERUN_OF', default=None) |
93 | | - self.skipped_issue = bool(strtobool(get_variable( |
94 | | - 'RP_SKIPPED_ISSUE', default='True'))) |
| 91 | + self.skipped_issue = to_bool(get_variable('RP_SKIPPED_ISSUE', default='True')) |
95 | 92 | self.test_attributes = get_variable('RP_TEST_ATTRIBUTES', default='').split() |
96 | 93 | self.log_batch_payload_size = int(get_variable('RP_LOG_BATCH_PAYLOAD_SIZE', |
97 | 94 | default=str(MAX_LOG_BATCH_PAYLOAD_SIZE))) |
98 | | - self.launch_uuid_print = bool(strtobool(get_variable('RP_LAUNCH_UUID_PRINT', default='False'))) |
| 95 | + self.launch_uuid_print = to_bool(get_variable('RP_LAUNCH_UUID_PRINT', default='False')) |
99 | 96 | output_type = get_variable('RP_LAUNCH_UUID_PRINT_OUTPUT') |
100 | 97 | self.launch_uuid_print_output = OutputType[output_type.upper()] if output_type else None |
101 | 98 | client_type = get_variable('RP_CLIENT_TYPE') |
@@ -169,4 +166,4 @@ def verify_ssl(self) -> Union[bool, str]: |
169 | 166 | verify_ssl = get_variable('RP_VERIFY_SSL', default='True') |
170 | 167 | if path.exists(verify_ssl): |
171 | 168 | return verify_ssl |
172 | | - return bool(strtobool(verify_ssl)) |
| 169 | + return to_bool(verify_ssl) |
0 commit comments