Skip to content

Commit f3bef3b

Browse files
committed
Merge branch 'dev' into staging
2 parents a496688 + 33967cc commit f3bef3b

File tree

6 files changed

+101
-10
lines changed

6 files changed

+101
-10
lines changed

portal_client.thrift

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -548,7 +548,9 @@ service VirtualMachineService {
548548
12:optional list<string> additional_security_group_ids,
549549
13:optional string slurm_version,
550550
14:optional string metadata_token,
551-
15:optional string metadata_endpoint
551+
15:optional string metadata_endpoint,
552+
16:optional string additional_script
553+
552554

553555
)
554556

@@ -586,6 +588,8 @@ service VirtualMachineService {
586588
11:list <string> additional_user_keys,
587589
12:optional string metadata_token,
588590
13:optional string metadata_endpoint,
591+
14:optional string additional_script
592+
589593

590594

591595

requirements.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,14 +3,14 @@ thrift==0.22.0
33
openstacksdk==4.7.1
44
deprecated==1.2.18
55
Click==8.3.0
6-
ansible==11.9.0
6+
ansible==11.10.0
77
flake8==7.3.0
88
paramiko==4.0.0
99
ruamel.yaml==0.18.15
1010
pyvim==3.0.3
1111
redis==6.4.0
1212
requests==2.32.5
13-
pyyaml==6.0.2
13+
pyyaml==6.0.3
1414
pre-commit==4.3.0
1515
types-PyYAML==6.0.12.20250915
1616
sympy==1.14.0

simple_vm_client/VirtualMachineHandler.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -424,6 +424,7 @@ def start_server(
424424
slurm_version: str = None,
425425
metadata_token: str = None,
426426
metadata_endpoint: str = None,
427+
additional_script: str = "",
427428
) -> str:
428429
if research_environment:
429430
research_environment_metadata = (
@@ -448,6 +449,7 @@ def start_server(
448449
slurm_version=slurm_version,
449450
metadata_token=metadata_token,
450451
metadata_endpoint=metadata_endpoint,
452+
additional_script=additional_script,
451453
)
452454

453455
def start_server_with_custom_key(
@@ -464,6 +466,7 @@ def start_server_with_custom_key(
464466
additional_user_keys: list[str],
465467
metadata_token: str = None,
466468
metadata_endpoint: str = None,
469+
additional_script: str = "",
467470
) -> str:
468471
if research_environment:
469472
research_environment_metadata = (
@@ -486,6 +489,7 @@ def start_server_with_custom_key(
486489
additional_security_group_ids=additional_security_group_ids,
487490
metadata_token=metadata_token,
488491
metadata_endpoint=metadata_endpoint,
492+
additional_script=additional_script,
489493
)
490494
self.forc_connector.set_vm_wait_for_playbook(
491495
openstack_id=openstack_id, private_key=private_key, name=servername
@@ -552,7 +556,6 @@ def start_cluster(
552556
metadata: ClusterInstanceMetadata,
553557
shared_volume: ClusterVolume = None,
554558
) -> ClusterMessage:
555-
556559

557560
return self.bibigrid_connector.start_cluster(
558561
public_keys=public_keys,

simple_vm_client/VirtualMachineService-remote

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -61,12 +61,12 @@ if len(sys.argv) <= 1 or sys.argv[1] == "--help":
6161
)
6262
print(" void unrescue_server(string openstack_id)")
6363
print(
64-
" string start_server(string flavor_name, string image_name, string public_key, string servername, metadata, volume_ids_path_new, volume_ids_path_attach, additional_owner_keys, additional_user_keys, string research_environment, additional_security_group_ids, string slurm_version, string metadata_token, string metadata_endpoint)"
64+
" string start_server(string flavor_name, string image_name, string public_key, string servername, metadata, volume_ids_path_new, volume_ids_path_attach, additional_owner_keys, additional_user_keys, string research_environment, additional_security_group_ids, string slurm_version, string metadata_token, string metadata_endpoint, string additional_script)"
6565
)
6666
print(" bool is_bibigrid_available()")
6767
print(" void detach_ip_from_server(string server_id, string floating_ip)")
6868
print(
69-
" string start_server_with_custom_key(string flavor_name, string image_name, string servername, metadata, string research_environment, volume_ids_path_new, volume_ids_path_attach, additional_security_group_ids, additional_owner_keys, additional_user_keys, string metadata_token, string metadata_endpoint)"
69+
" string start_server_with_custom_key(string flavor_name, string image_name, string servername, metadata, string research_environment, volume_ids_path_new, volume_ids_path_attach, additional_security_group_ids, additional_owner_keys, additional_user_keys, string metadata_token, string metadata_endpoint, string additional_script)"
7070
)
7171
print(" bool exist_server(string name)")
7272
print(
@@ -452,8 +452,8 @@ elif cmd == "unrescue_server":
452452
)
453453

454454
elif cmd == "start_server":
455-
if len(args) != 14:
456-
print("start_server requires 14 args")
455+
if len(args) != 15:
456+
print("start_server requires 15 args")
457457
sys.exit(1)
458458
pp.pprint(
459459
client.start_server(
@@ -471,6 +471,7 @@ elif cmd == "start_server":
471471
args[11],
472472
args[12],
473473
args[13],
474+
args[14],
474475
)
475476
)
476477

@@ -492,8 +493,8 @@ elif cmd == "detach_ip_from_server":
492493
)
493494

494495
elif cmd == "start_server_with_custom_key":
495-
if len(args) != 12:
496-
print("start_server_with_custom_key requires 12 args")
496+
if len(args) != 13:
497+
print("start_server_with_custom_key requires 13 args")
497498
sys.exit(1)
498499
pp.pprint(
499500
client.start_server_with_custom_key(
@@ -509,6 +510,7 @@ elif cmd == "start_server_with_custom_key":
509510
eval(args[9]),
510511
args[10],
511512
args[11],
513+
args[12],
512514
)
513515
)
514516

simple_vm_client/VirtualMachineService.py

Lines changed: 70 additions & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)