|
2 | 2 | from warnings import warn |
3 | 3 | import logging |
4 | 4 | import re |
5 | | -from typing import Dict, Any, List, Union |
6 | | - |
| 5 | +from typing import Dict, Any, List, Union, Tuple |
7 | 6 | import libsbml |
8 | 7 |
|
9 | 8 | logger = logging.getLogger(__name__) |
@@ -406,3 +405,62 @@ def write_sbml(sbml_doc: libsbml.SBMLDocument, filename: str) -> None: |
406 | 405 | if not ret: |
407 | 406 | raise RuntimeError(f"libSBML reported error {ret} when trying to " |
408 | 407 | f"create SBML file {filename}.") |
| 408 | + |
| 409 | + |
| 410 | +def get_sbml_model( |
| 411 | + filepath_or_buffer |
| 412 | +) -> Tuple[libsbml.SBMLReader, libsbml.SBMLDocument, libsbml.Model]: |
| 413 | + """Get an SBML model from file or URL or file handle |
| 414 | +
|
| 415 | + :param filepath_or_buffer: |
| 416 | + File or URL or file handle to read the model from |
| 417 | + :return: The SBML document, model and reader |
| 418 | + """ |
| 419 | + |
| 420 | + from pandas.io.common import get_filepath_or_buffer, is_url, is_file_like |
| 421 | + |
| 422 | + if is_file_like(filepath_or_buffer) or is_url(filepath_or_buffer): |
| 423 | + buffer = get_filepath_or_buffer(filepath_or_buffer, mode='r')[0] |
| 424 | + if is_url(filepath_or_buffer): |
| 425 | + buffer = ''.join(line.decode('utf-8') for line in buffer) |
| 426 | + else: |
| 427 | + buffer = ''.join(line for line in buffer) |
| 428 | + |
| 429 | + # URL or already opened file, we will load the model from a string |
| 430 | + |
| 431 | + return load_sbml_from_string(buffer) |
| 432 | + |
| 433 | + return load_sbml_from_file(filepath_or_buffer) |
| 434 | + |
| 435 | + |
| 436 | +def load_sbml_from_string( |
| 437 | + sbml_string: str |
| 438 | +) -> Tuple[libsbml.SBMLReader, libsbml.SBMLDocument, libsbml.Model]: |
| 439 | + """Load SBML model from string |
| 440 | +
|
| 441 | + :param sbml_string: Model as XML string |
| 442 | + :return: The SBML document, model and reader |
| 443 | + """ |
| 444 | + |
| 445 | + sbml_reader = libsbml.SBMLReader() |
| 446 | + sbml_document = \ |
| 447 | + sbml_reader.readSBMLFromString(sbml_string) |
| 448 | + sbml_model = sbml_document.getModel() |
| 449 | + |
| 450 | + return sbml_reader, sbml_document, sbml_model |
| 451 | + |
| 452 | + |
| 453 | +def load_sbml_from_file( |
| 454 | + sbml_file: str |
| 455 | +) -> Tuple[libsbml.SBMLReader, libsbml.SBMLDocument, libsbml.Model]: |
| 456 | + """Load SBML model from file |
| 457 | +
|
| 458 | + :param sbml_file: Filename of the SBML file |
| 459 | + :return: The SBML document, model and reader |
| 460 | + """ |
| 461 | + |
| 462 | + sbml_reader = libsbml.SBMLReader() |
| 463 | + sbml_document = sbml_reader.readSBML(sbml_file) |
| 464 | + sbml_model = sbml_document.getModel() |
| 465 | + |
| 466 | + return sbml_reader, sbml_document, sbml_model |
0 commit comments