Skip to content

Commit e3f3cd4

Browse files
committed
feat: Update connector classes to use Config from adf_core_python.core.config.config
1 parent 9f0aa2d commit e3f3cd4

File tree

7 files changed

+46
-68
lines changed

7 files changed

+46
-68
lines changed

adf_core_python/core/launcher/connect/connector.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
from abc import ABC, abstractmethod
22

33
from rcrs_core.connection.componentLauncher import ComponentLauncher
4-
from rcrs_core.config.config import Config
54

65
from adf_core_python.core.component.abstract_loader import AbstractLoader
6+
from adf_core_python.core.config.config import Config
77

88

99
class Connector(ABC):

adf_core_python/core/launcher/connect/connector_ambulance_centre.py

Lines changed: 7 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.ambulanceCenterAgent import AmbulanceCenterAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.config.config import Config
1010
from adf_core_python.core.launcher.config_key import ConfigKey
1111
from adf_core_python.core.launcher.connect.connector import Connector
1212

@@ -22,31 +22,29 @@ def connect(
2222
config: Config,
2323
loader: AbstractLoader,
2424
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
25+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2826
if count == 0:
2927
return
3028

3129
for _ in range(count):
3230
# tactics_ambulance_centre: TacticsAmbulanceCentre
33-
if loader.get_tactics_ambulance_centre() is not None:
31+
if loader.get_tactics_ambulance_center() is not None:
3432
self.logger.error("Cannot load ambulance centre tactics")
3533
# tactics_ambulance_centre = loader.get_tactics_ambulance_centre()
3634
else:
3735
# tactics_ambulance_centre = DummyTacticsAmbulanceCentre()
3836
pass
3937

4038
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
39+
config.get_value(
4240
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4341
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4442
)
4543
)
4644

4745
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
46+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
47+
config.get_value(
5048
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5149
),
5250
)
@@ -55,9 +53,7 @@ def connect(
5553
component_launcher.connect(
5654
# TODO: AmbulanceCenterAgent is not implemented precompute method and other methods
5755
AmbulanceCenterAgent(
58-
config.get_boolean_value_or_default(
59-
ConfigKey.KEY_PRECOMPUTE, False
60-
),
56+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6157
), # type: ignore
6258
component_launcher.generate_request_ID(),
6359
)

adf_core_python/core/launcher/connect/connector_ambulance_team.py

Lines changed: 14 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,15 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.ambulanceTeamAgent import AmbulanceTeamAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.component.tactics.tactics_ambulance_team import (
10+
TacticsAmbulanceTeam,
11+
)
12+
from adf_core_python.core.config.config import Config
1013
from adf_core_python.core.launcher.config_key import ConfigKey
1114
from adf_core_python.core.launcher.connect.connector import Connector
1215

@@ -22,41 +25,36 @@ def connect(
2225
config: Config,
2326
loader: AbstractLoader,
2427
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
28+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2829
if count == 0:
2930
return
3031

3132
for _ in range(count):
32-
# tactics_ambulance_team: TacticsAmbulanceTeam
33-
if loader.get_tactics_ambulance_team() is not None:
33+
if loader.get_tactics_ambulance_team() is None:
3434
self.logger.error("Cannot load ambulance team tactics")
35-
# tactics_ambulance_team = loader.get_tactics_ambulance_team()
36-
else:
37-
# tactics_ambulance_team = DummyTacticsAmbulanceTeam()
38-
pass
35+
36+
tactics_ambulance_team: TacticsAmbulanceTeam = (
37+
loader.get_tactics_ambulance_team()
38+
)
3939

4040
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
41+
config.get_value(
4242
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4343
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4444
)
4545
)
4646

4747
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
48+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
49+
config.get_value(
5050
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5151
),
5252
)
5353

5454
# TODO: component_launcher.generate_request_ID can cause race condition
5555
component_launcher.connect(
5656
AmbulanceTeamAgent(
57-
config.get_boolean_value_or_default(
58-
ConfigKey.KEY_PRECOMPUTE, False
59-
),
57+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6058
),
6159
component_launcher.generate_request_ID(),
6260
)

adf_core_python/core/launcher/connect/connector_fire_brigade.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.fireBrigadeAgent import FireBrigadeAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.config.config import Config
1010
from adf_core_python.core.launcher.config_key import ConfigKey
1111
from adf_core_python.core.launcher.connect.connector import Connector
1212

@@ -22,9 +22,7 @@ def connect(
2222
config: Config,
2323
loader: AbstractLoader,
2424
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
25+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2826
if count == 0:
2927
return
3028

@@ -38,25 +36,23 @@ def connect(
3836
pass
3937

4038
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
39+
config.get_value(
4240
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4341
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4442
)
4543
)
4644

4745
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
46+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
47+
config.get_value(
5048
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5149
),
5250
)
5351

5452
# TODO: component_launcher.generate_request_ID can cause race condition
5553
component_launcher.connect(
5654
FireBrigadeAgent(
57-
config.get_boolean_value_or_default(
58-
ConfigKey.KEY_PRECOMPUTE, False
59-
),
55+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6056
),
6157
component_launcher.generate_request_ID(),
6258
)

adf_core_python/core/launcher/connect/connector_fire_station.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.fireStationAgent import FireStationAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.config.config import Config
1010
from adf_core_python.core.launcher.config_key import ConfigKey
1111
from adf_core_python.core.launcher.connect.connector import Connector
1212

@@ -22,9 +22,7 @@ def connect(
2222
config: Config,
2323
loader: AbstractLoader,
2424
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
25+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2826
if count == 0:
2927
return
3028

@@ -38,25 +36,23 @@ def connect(
3836
pass
3937

4038
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
39+
config.get_value(
4240
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4341
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4442
)
4543
)
4644

4745
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
46+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
47+
config.get_value(
5048
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5149
),
5250
)
5351

5452
# TODO: component_launcher.generate_request_ID can cause race condition
5553
component_launcher.connect(
5654
FireStationAgent(
57-
config.get_boolean_value_or_default(
58-
ConfigKey.KEY_PRECOMPUTE, False
59-
),
55+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6056
), # type: ignore
6157
component_launcher.generate_request_ID(),
6258
)

adf_core_python/core/launcher/connect/connector_police_force.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.policeForceAgent import PoliceForceAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.config.config import Config
1010
from adf_core_python.core.launcher.config_key import ConfigKey
1111
from adf_core_python.core.launcher.connect.connector import Connector
1212

@@ -22,9 +22,7 @@ def connect(
2222
config: Config,
2323
loader: AbstractLoader,
2424
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
25+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2826
if count == 0:
2927
return
3028

@@ -38,25 +36,23 @@ def connect(
3836
pass
3937

4038
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
39+
config.get_value(
4240
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4341
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4442
)
4543
)
4644

4745
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
46+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
47+
config.get_value(
5048
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5149
),
5250
)
5351

5452
# TODO: component_launcher.generate_request_ID can cause race condition
5553
component_launcher.connect(
5654
PoliceForceAgent(
57-
config.get_boolean_value_or_default(
58-
ConfigKey.KEY_PRECOMPUTE, False
59-
),
55+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6056
),
6157
component_launcher.generate_request_ID(),
6258
)

adf_core_python/core/launcher/connect/connector_police_office.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
from logging import Logger, getLogger
22

33
from rcrs_core.agents.policeOfficeAgent import PoliceOfficeAgent
4-
from rcrs_core.config.config import Config
54
from rcrs_core.connection.componentLauncher import ComponentLauncher
65

76
from adf_core_python.core.agent.config.module_config import ModuleConfig
87
from adf_core_python.core.agent.develop.develop_data import DevelopData
98
from adf_core_python.core.component.abstract_loader import AbstractLoader
9+
from adf_core_python.core.config.config import Config
1010
from adf_core_python.core.launcher.config_key import ConfigKey
1111
from adf_core_python.core.launcher.connect.connector import Connector
1212

@@ -22,9 +22,7 @@ def connect(
2222
config: Config,
2323
loader: AbstractLoader,
2424
) -> None:
25-
count: int = config.get_int_value_or_default(
26-
ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0
27-
)
25+
count: int = config.get_value(ConfigKey.KEY_AMBULANCE_CENTRE_COUNT, 0)
2826
if count == 0:
2927
return
3028

@@ -38,25 +36,23 @@ def connect(
3836
pass
3937

4038
module_config: ModuleConfig = ModuleConfig( # noqa: F841
41-
config.get_value_or_default(
39+
config.get_value(
4240
ConfigKey.KEY_MODULE_CONFIG_FILE_NAME,
4341
ModuleConfig.DEFAULT_CONFIG_FILE_NAME,
4442
)
4543
)
4644

4745
develop_data: DevelopData = DevelopData( # noqa: F841
48-
config.get_boolean_value_or_default(ConfigKey.KEY_DEBUG_FLAG, False),
49-
config.get_value_or_default(
46+
config.get_value(ConfigKey.KEY_DEBUG_FLAG, False),
47+
config.get_value(
5048
ConfigKey.KEY_DEVELOP_DATA_FILE_NAME, DevelopData.DEFAULT_FILE_NAME
5149
),
5250
)
5351

5452
# TODO: component_launcher.generate_request_ID can cause race condition
5553
component_launcher.connect(
5654
PoliceOfficeAgent(
57-
config.get_boolean_value_or_default(
58-
ConfigKey.KEY_PRECOMPUTE, False
59-
),
55+
config.get_value(ConfigKey.KEY_PRECOMPUTE, False),
6056
),
6157
component_launcher.generate_request_ID(),
6258
)

0 commit comments

Comments
 (0)