Skip to content

Commit

Permalink
Fix handling of PYPRIMEMESH_LAUNCH_CONTAINER environment variable dur…
Browse files Browse the repository at this point in the history
…ing launch (#549)

Co-authored-by: Alex Fernandez <[email protected]>
  • Loading branch information
ninad-kamat and AlejandroFernandezLuces authored Jul 19, 2023
1 parent 07295a7 commit 53a704f
Show file tree
Hide file tree
Showing 2 changed files with 22 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/ansys/meshing/prime/internals/launcher.py
Original file line number Diff line number Diff line change
Expand Up @@ -226,8 +226,9 @@ def launch_prime(
if ip == defaults.ip():
port = utils.get_available_local_port(port)

if 'PYPRIMEMESH_LAUNCH_CONTAINER' in os.environ:
container_name = 'ansys-prime-server'
launch_container = bool(int(os.environ.get('PYPRIMEMESH_LAUNCH_CONTAINER', '0')))
if launch_container:
container_name = utils.make_unique_container_name('ansys-prime-server')
utils.launch_prime_github_container(port=port, name=container_name, version=version)
config.set_using_container(True)
client = Client(port=port, timeout=timeout)
Expand Down
19 changes: 19 additions & 0 deletions src/ansys/meshing/prime/internals/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,25 @@
import ansys.meshing.prime.internals.defaults as defaults

_LOCAL_PORTS = []
_PRIME_CONTAINER_COUNT = 0


def make_unique_container_name(name: str):
"""Make a unique container name.
Parameters
----------
name : str
Original name prefix that is provided.
Returns
-------
str
Unique name with a numeric integer added as suffix.
"""
global _PRIME_CONTAINER_COUNT
_PRIME_CONTAINER_COUNT = _PRIME_CONTAINER_COUNT + 1
return f'{name}-{_PRIME_CONTAINER_COUNT}'


def to_camel_case(snake_str):
Expand Down

0 comments on commit 53a704f

Please sign in to comment.