@@ -87,6 +87,7 @@ def __init__(self, obj):
8787tests_name = 'rabbitmq-integration-tests'
8888nodeport_service_name = 'rabbitmq-nodeport'
8989cr_version = "v2"
90+ api_group = os .getenv ("API_GROUP" , "qubership.org" )
9091IN_PROGRESS = "In progress"
9192SUCCESSFUL = "Successful"
9293FAILED = "Failed"
@@ -1249,7 +1250,7 @@ def scale_up_backup_daemon(self,
12491250
12501251 def get_custom_resource (self ):
12511252 cr = self ._custom_objects_api .get_namespaced_custom_object (
1252- group = 'qubership.org' , version = cr_version ,
1253+ group = api_group , version = cr_version ,
12531254 namespace = self ._workspace ,
12541255 plural = 'rabbitmqservices' ,
12551256 name = 'rabbitmq-service'
@@ -1258,7 +1259,7 @@ def get_custom_resource(self):
12581259
12591260 def update_custom_resource (self , body ):
12601261 self ._custom_objects_api .patch_namespaced_custom_object (
1261- group = 'qubership.org' ,
1262+ group = api_group ,
12621263 version = cr_version ,
12631264 namespace = self ._workspace ,
12641265 plural = 'rabbitmqservices' ,
@@ -1268,7 +1269,7 @@ def update_custom_resource(self, body):
12681269
12691270 def get_custom_resource_status (self ):
12701271 return self ._custom_objects_api .get_namespaced_custom_object_status (
1271- group = 'qubership.org' ,
1272+ group = api_group ,
12721273 version = cr_version ,
12731274 namespace = self ._workspace ,
12741275 plural = 'rabbitmqservices' ,
@@ -1277,7 +1278,7 @@ def get_custom_resource_status(self):
12771278
12781279 def update_custom_resource_status (self , body ):
12791280 self ._custom_objects_api .patch_namespaced_custom_object_status (
1280- group = 'qubership.org' ,
1281+ group = api_group ,
12811282 version = cr_version ,
12821283 namespace = self ._workspace ,
12831284 plural = 'rabbitmqservices' ,
@@ -1413,7 +1414,7 @@ def configure(settings: kopf.OperatorSettings, **_):
14131414 settings .scanning .disabled = True
14141415
14151416
1416- @kopf .on .create ('qubership.org' , cr_version , 'rabbitmqservices' )
1417+ @kopf .on .create (api_group , cr_version , 'rabbitmqservices' )
14171418def on_create (body , meta , spec , status , ** kwargs ):
14181419 kub_helper = KubernetesHelper (spec )
14191420 logger .info ("New CRD is created" )
@@ -1501,7 +1502,7 @@ def validate_spec(spec):
15011502 pass
15021503
15031504
1504- # @kopf.on.field('qubership.org' , 'v1', 'rabbitmqservices', field='spec.rabbitmq.replicas')
1505+ # @kopf.on.field(api_group , 'v1', 'rabbitmqservices', field='spec.rabbitmq.replicas')
15051506# def update_lst(body, meta, spec, status, old, new, **kwargs):
15061507# print(f'Handling the FIELD = {old} -> {new}')
15071508
@@ -1558,7 +1559,7 @@ def on_update_configmap(diff, **kwargs):
15581559 custom_objects_api = client .CustomObjectsApi ()
15591560 namespace = KubernetesHelper .get_namespace ()
15601561 cr = custom_objects_api .get_namespaced_custom_object (
1561- group = 'qubership.org' ,
1562+ group = api_group ,
15621563 version = cr_version ,
15631564 namespace = namespace ,
15641565 plural = 'rabbitmqservices' ,
@@ -1588,7 +1589,7 @@ def on_update_secret(diff, **kwargs):
15881589 custom_objects_api = client .CustomObjectsApi ()
15891590 namespace = KubernetesHelper .get_namespace ()
15901591 cr = custom_objects_api .get_namespaced_custom_object (
1591- group = 'qubership.org' ,
1592+ group = api_group ,
15921593 version = cr_version ,
15931594 namespace = namespace ,
15941595 plural = 'rabbitmqservices' ,
@@ -1604,7 +1605,7 @@ def on_update_secret(diff, **kwargs):
16041605 wait_time = wait_time + 15
16051606 sleep (15 )
16061607 cr = custom_objects_api .get_namespaced_custom_object (
1607- group = 'qubership.org' ,
1608+ group = api_group ,
16081609 version = cr_version ,
16091610 namespace = namespace ,
16101611 plural = 'rabbitmqservices' ,
@@ -1642,7 +1643,7 @@ def on_update_secret(diff, **kwargs):
16421643 "All pods have been rebooted, changing credentials completed" )
16431644
16441645
1645- @kopf .on .update ('qubership.org' , cr_version , 'rabbitmqservices' , when = exclude_disaster_recovery_field )
1646+ @kopf .on .update (api_group , cr_version , 'rabbitmqservices' , when = exclude_disaster_recovery_field )
16461647def on_update (body , meta , spec , status , old , new , diff , ** kwargs ):
16471648 logger .info ("cr changes:" + str (diff ))
16481649 print ('Handling the diff' )
@@ -1763,7 +1764,7 @@ def perform_rabbit_pods_readiness_check(kub_helper: KubernetesHelper):
17631764 logger .info ("RabbitMQ pods are ready" )
17641765
17651766
1766- @kopf .on .delete ('qubership.org' , cr_version , 'rabbitmqservices' , optional = optional_delete )
1767+ @kopf .on .delete (api_group , cr_version , 'rabbitmqservices' , optional = optional_delete )
17671768def on_delete (spec , ** kwargs ):
17681769 kub_helper = KubernetesHelper (spec )
17691770 logger .info ("Deleting crd" )
@@ -1781,14 +1782,14 @@ def switchover_annotation_changed(diff, logger, **kwargs):
17811782 return False
17821783
17831784
1784- @kopf .on .update ('qubership.org' ,
1785+ @kopf .on .update (api_group ,
17851786 cr_version ,
17861787 'rabbitmqservices' ,
17871788 field = 'metadata.annotations.switchoverRetry' ,
17881789 old = kopf .ABSENT ,
17891790 new = kopf .PRESENT )
1790- @kopf .on .update ('qubership.org' , cr_version , 'rabbitmqservices' , when = switchover_annotation_changed )
1791- @kopf .on .field ('qubership.org' , cr_version , 'rabbitmqservices' , field = 'spec.disasterRecovery.mode' )
1791+ @kopf .on .update (api_group , cr_version , 'rabbitmqservices' , when = switchover_annotation_changed )
1792+ @kopf .on .field (api_group , cr_version , 'rabbitmqservices' , field = 'spec.disasterRecovery.mode' )
17921793def set_disaster_recovery_state (spec , status , namespace , diff , ** kwargs ):
17931794 mode = spec .get ('disasterRecovery' ).get ('mode' , None )
17941795 if mode is None :
0 commit comments