Skip to content

Commit

Permalink
spalloc user and password from environment
Browse files Browse the repository at this point in the history
  • Loading branch information
Christian-B committed May 22, 2024
1 parent 8e626d4 commit 7ad2be0
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 6 deletions.
14 changes: 12 additions & 2 deletions spinnman/spalloc/spalloc_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@
"""
Implementation of the client for the Spalloc web service.
"""
import os
import time
from logging import getLogger

Expand Down Expand Up @@ -108,12 +109,21 @@ def __init__(
location; if so, the ``username`` and ``password`` arguments
*must* be ``None``. If ``username`` and ``password`` are not given,
not even within the URL, the ``bearer_token`` must be not ``None``.
:param str username: The user name to use
:param str password: The password to use
:param str username:
The user name to use. If not provided nor in service_url
environment variable SPALLOC_USER will be used.
:param str password:
The password to use. If not provided nor in service_url
environment variable SPALLOC_PASSWORD will be used.
:param str bearer_token: The bearer token to use
"""
if username is None and password is None:
service_url, username, password = parse_service_url(service_url)
if username is None:
username = os.environ["SPALLOC_USER"]
if password is None:
password = os.environ["SPALLOC_PASSWORD"]

self.__session: Optional[Session] = Session(
service_url, username, password, bearer_token)
obj = self.__session.renew()
Expand Down
5 changes: 1 addition & 4 deletions spinnman_integration_tests/test_job.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,12 +29,9 @@ def setUp(self):
set_config("Machine", "version", FIVE)
self.spalloc_url = "https://spinnaker.cs.man.ac.uk/spalloc"
self.spalloc_machine = "SpiNNaker1M"
self.spalloc_user = os.environ["SPALLOC_USER"]
self.spalloc_password = os.environ["SPALLOC_PASSWORD"]

def test_create_job(self):
client = SpallocClient(
self.spalloc_url, self.spalloc_user, self.spalloc_password)
client = SpallocClient(self.spalloc_url)
# job = client.create_job_rect_at_board(
# WIDTH, HEIGHT, triad=(x, y, b), machine_name=SPALLOC_MACHINE,
# max_dead_boards=1)
Expand Down

0 comments on commit 7ad2be0

Please sign in to comment.