pybiolib 1.2.138.dev1__py3-none-any.whl → 1.2.141.dev1__py3-none-any.whl

biolib/__init__.py

@@ -49,8 +49,8 @@ def get_data_record(uri: str) -> _DataRecord:
     return _DataRecord.get_by_uri(uri)
 
 
-def fetch_jobs(count: int = 25) -> List[_Job]:
-    return _Job.fetch_jobs(count)
+def fetch_jobs(count: int = 25, status: Optional[str] = None) -> List[_Job]:
+    return _Job.fetch_jobs(count, status)
 
 
 def fetch_data_records(uri: Optional[str] = None, count: Optional[int] = None) -> List[_DataRecord]:

biolib/jobs/job.py

@@ -22,7 +22,7 @@ from biolib.compute_node.utils import SystemExceptionCodeMap, SystemExceptionCod
 from biolib.jobs.job_result import JobResult
 from biolib.jobs.types import CloudJobDict, CloudJobStartedDict, JobDict
 from biolib.tables import BioLibTable
-from biolib.typing_utils import Dict, List, Optional, cast
+from biolib.typing_utils import Dict, List, Optional, Union, cast
 from biolib.utils import IS_RUNNING_IN_NOTEBOOK
 from biolib.utils.app_uri import parse_app_uri
 
@@ -232,8 +232,8 @@ class Job:
         self._result = JobResult(job_uuid=self._uuid, job_auth_token=self._auth_token, module_output=module_output)
 
     @staticmethod
-    def fetch_jobs(count: int) -> List['Job']:
-        job_dicts = Job._get_job_dicts(count)
+    def fetch_jobs(count: int, status: Optional[str] = None) -> List['Job']:
+        job_dicts = Job._get_job_dicts(count, status)
         return [Job(job_dict) for job_dict in job_dicts]
 
     @staticmethod
@@ -242,9 +242,27 @@ class Job:
         BioLibTable(columns_to_row_map=Job.table_columns_to_row_map, rows=job_dicts, title='Jobs').print_table()
 
     @staticmethod
-    def _get_job_dicts(count: int) -> List['JobDict']:
-        job_dicts: List['JobDict'] = api.client.get(path='/jobs/', params={'page_size': str(count)}).json()['results']
-        return job_dicts
+    def _get_job_dicts(count: int, status: Optional[str] = None) -> List['JobDict']:
+        job_states = ['in_progress', 'completed', 'failed', 'cancelled']
+        if status is not None and status not in job_states:
+            raise Exception('Invalid status filter')
+
+        page_size = min(count, 1_000)
+        params: Dict[str, Union[str, int]] = dict(page_size=page_size)
+        if status:
+            params['state'] = status
+
+        api_path = '/jobs/'
+        response = api.client.get(api_path, params=params).json()
+        jobs = [job_dict for job_dict in response['results']]
+
+        for page_number in range(2, response['page_count'] + 1):
+            page_response = api.client.get(path=api_path, params=dict(**params, page=page_number)).json()
+            jobs.extend([job_dict for job_dict in page_response['results']])
+            if len(jobs) >= count:
+                break
+
+        return jobs
 
     @staticmethod
     def _get_job_dict(uuid: str, auth_token: Optional[str] = None) -> JobDict:

{pybiolib-1.2.138.dev1.dist-info → pybiolib-1.2.141.dev1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pybiolib
-Version: 1.2.138.dev1
+Version: 1.2.141.dev1
 Summary: BioLib Python Client
 Home-page: https://github.com/biolib
 License: MIT

{pybiolib-1.2.138.dev1.dist-info → pybiolib-1.2.141.dev1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 LICENSE,sha256=F2h7gf8i0agDIeWoBPXDMYScvQOz02pAWkKhTGOHaaw,1067
 PYPI_README.md,sha256=_IH7pxFiqy2bIAmaVeA-iVTyUwWRjMIlfgtUbYTtmls,368
-biolib/__init__.py,sha256=q_YhAYw51Vq16IKtSk8_MJclDAa4CfCmPhvWDYmrSIg,4393
+biolib/__init__.py,sha256=4Rfa0AJKztqkCG5D67kFgSwdOUiBTa5HkAzIOqHNREU,4431
 biolib/_data_record/data_record.py,sha256=zVAhFU1RLI1-ptoQ_l639RNwrMANXV9j75yXHvB7dtA,10950
 biolib/_internal/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biolib/_internal/data_record/__init__.py,sha256=fGdME6JGRU_2VxpJbYpGXYndjN-feUkmKY4fuMyq3cg,76
@@ -102,7 +102,7 @@ biolib/compute_node/webserver/worker_thread.py,sha256=GRRBUqXdMKvbjyLQhYlqGIbFKe
 biolib/experiments/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biolib/experiments/experiment.py,sha256=jIRixmQm3Gq9YdJ3I0-rE1vFukXqq6U4zXehFOJ1yZk,7614
 biolib/jobs/__init__.py,sha256=aIb2H2DHjQbM2Bs-dysFijhwFcL58Blp0Co0gimED3w,32
-biolib/jobs/job.py,sha256=OfG8cLd3AjGjiMWRlJRZdVVbLsRWSX-OM5nxJhR6mPQ,19136
+biolib/jobs/job.py,sha256=ej-dxt1hMBQXzTfUZ2SLqIV8dtTbMSMn_q2rjIA7oN4,19903
 biolib/jobs/job_result.py,sha256=rALHiKYNaC9lHi_JJqBob1RubzNLwG9Z386kwRJjd2M,5885
 biolib/jobs/types.py,sha256=qhadtH2KDC2WUOOqPiwke0YgtQY4FtuB71Stekq1k48,970
 biolib/runtime/__init__.py,sha256=MlRepA11n2H-3plB5rzWyyHK2JmP6PiaP3i6x3vt0mg,506
@@ -119,8 +119,8 @@ biolib/utils/cache_state.py,sha256=u256F37QSRIVwqKlbnCyzAX4EMI-kl6Dwu6qwj-Qmag,3
 biolib/utils/multipart_uploader.py,sha256=XvGP1I8tQuKhAH-QugPRoEsCi9qvbRk-DVBs5PNwwJo,8452
 biolib/utils/seq_util.py,sha256=WJnU9vZdwY8RHXvzATyV80OXzyJ7w9EkG33Tna9Nr6A,5698
 biolib/utils/zip/remote_zip.py,sha256=0wErYlxir5921agfFeV1xVjf29l9VNgGQvNlWOlj2Yc,23232
-pybiolib-1.2.138.dev1.dist-info/LICENSE,sha256=F2h7gf8i0agDIeWoBPXDMYScvQOz02pAWkKhTGOHaaw,1067
-pybiolib-1.2.138.dev1.dist-info/METADATA,sha256=sRwnA9wyiSa5kH5A7aJpOs6rknXrr1SgKFVvbX5dLQ0,1512
-pybiolib-1.2.138.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-pybiolib-1.2.138.dev1.dist-info/entry_points.txt,sha256=p6DyaP_2kctxegTX23WBznnrDi4mz6gx04O5uKtRDXg,42
-pybiolib-1.2.138.dev1.dist-info/RECORD,,
+pybiolib-1.2.141.dev1.dist-info/LICENSE,sha256=F2h7gf8i0agDIeWoBPXDMYScvQOz02pAWkKhTGOHaaw,1067
+pybiolib-1.2.141.dev1.dist-info/METADATA,sha256=9NEHC1fbAZuf_mpTlZyUtdPhEGFvBkhcyKwWzxugvfI,1512
+pybiolib-1.2.141.dev1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+pybiolib-1.2.141.dev1.dist-info/entry_points.txt,sha256=p6DyaP_2kctxegTX23WBznnrDi4mz6gx04O5uKtRDXg,42
+pybiolib-1.2.141.dev1.dist-info/RECORD,,