biocypher 0.9.4__py3-none-any.whl → 0.9.6__py3-none-any.whl

biocypher/_get.py

@@ -420,7 +420,7 @@ class Downloader:
             with open(self.cache_file, "w") as f:
                 json.dump({}, f)
 
-        with open(self.cache_file, "r") as f:
+        with open(self.cache_file) as f:
             logger.info(f"Loading cache file {self.cache_file}.")
             return json.load(f)
 

biocypher/_metadata.py

@@ -10,7 +10,7 @@ import pathlib
 
 import toml
 
-_VERSION = "0.9.4"
+_VERSION = "0.9.6"
 
 
 def get_metadata():

biocypher/output/connect/_get_connector.py

@@ -16,14 +16,18 @@ __all__ = ["get_connector"]
 def get_connector(
     dbms: str,
     translator: Translator,
-):
-    """
-    Function to return the connector class.
+) -> _Neo4jDriver:
+    """Return the connector class.
 
-    Returns:
+    Returns
+    -------
         class: the connector class
-    """
 
+    Raises
+    ------
+        NotImplementedError: if the DBMS is not supported
+
+    """
     dbms_config = _config(dbms)
 
     if dbms == "neo4j":
@@ -36,5 +40,7 @@ def get_connector(
             multi_db=dbms_config["multi_db"],
             translator=translator,
         )
-    else:
-        raise NotImplementedError(f"Online mode is not supported for the DBMS {dbms}.")
+
+    msg = f"Online mode is not supported for the DBMS {dbms}."
+    logger.error(msg)
+    raise NotImplementedError(msg)

biocypher/output/in_memory/_get_in_memory_kg.py

@@ -1,12 +1,20 @@
+"""BioCypher 'in_memory' module.
+
+Handles the in-memory Knowledge Graph instance.
 """
-BioCypher 'in_memory' module. Handles the in-memory Knowledge Graph instance.
-"""
 
-from biocypher._deduplicate import Deduplicator
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
 from biocypher._logger import logger
 from biocypher.output.in_memory._networkx import NetworkxKG
 from biocypher.output.in_memory._pandas import PandasKG
 
+if TYPE_CHECKING:
+    from biocypher._deduplicate import Deduplicator
+    from biocypher.output.in_memory._in_memory_kg import _InMemoryKG
+
 logger.debug(f"Loading module {__name__}.")
 
 __all__ = ["get_in_memory_kg"]
@@ -17,18 +25,20 @@ IN_MEMORY_DBMS = ["csv", "pandas", "tabular", "networkx"]
 def get_in_memory_kg(
     dbms: str,
     deduplicator: Deduplicator,
-):
-    """
-    Function to return the in-memory KG class.
+) -> _InMemoryKG:
+    """Return the in-memory KG class.
+
+    Returns
+    -------
+        _InMemoryKG: the in-memory KG class
 
-    Returns:
-        class: the in-memory KG class
     """
     if dbms in ["csv", "pandas", "tabular"]:
         return PandasKG(deduplicator)
-    elif dbms == "networkx":
+
+    if dbms == "networkx":
         return NetworkxKG(deduplicator)
-    else:
-        raise NotImplementedError(
-            f"Getting the in memory BioCypher KG is not supported for the DBMS {dbms}. Supported: {IN_MEMORY_DBMS}."
-        )
+
+    msg = f"Getting the in memory BioCypher KG is not supported for the DBMS {dbms}. Supported: {IN_MEMORY_DBMS}."
+    logger.error(msg)
+    raise NotImplementedError(msg)

biocypher/output/write/graph/_neo4j.py

@@ -278,9 +278,7 @@ class _Neo4jBatchWriter(_BatchWriter):
         """
         import_call_neo4j_v4 = self._get_import_call("import", "--database=", "--force=")
         import_call_neo4j_v5 = self._get_import_call("database import full", "", "--overwrite-destination=")
-        neo4j_version_check = (
-            f"version=$({self._get_default_import_call_bin_prefix()}neo4j-admin --version | cut -d '.' -f 1)"
-        )
+        neo4j_version_check = f"version=$({self.import_call_bin_prefix}neo4j-admin --version | cut -d '.' -f 1)"
 
         import_script = (
             f"#!/bin/bash\n{neo4j_version_check}\nif [[ $version -ge 5 ]]; "

biocypher/output/write/relational/_postgresql.py

@@ -6,7 +6,8 @@ from biocypher.output.write._batch_writer import _BatchWriter
 
 
 class _PostgreSQLBatchWriter(_BatchWriter):
-    """
+    """Write node and edge representations for PostgreSQL.
+
     Class for writing node and edge representations to disk using the
     format specified by PostgreSQL for the use of "COPY FROM...". Each batch
     writer instance has a fixed representation that needs to be passed
@@ -39,12 +40,13 @@ class _PostgreSQLBatchWriter(_BatchWriter):
         self._copy_from_csv_commands = set()
         super().__init__(*args, **kwargs)
 
-    def _get_default_import_call_bin_prefix(self):
-        """
-        Method to provide the default string for the import call bin prefix.
+    def _get_default_import_call_bin_prefix(self) -> str:
+        """Provide the default string for the import call bin prefix.
 
-        Returns:
+        Returns
+        -------
             str: The default location for the psql command
+
         """
         return ""
 
@@ -56,33 +58,36 @@ class _PostgreSQLBatchWriter(_BatchWriter):
             return "VARCHAR"
 
     def _quote_string(self, value: str) -> str:
-        """
-        Quote a string.
-        """
-
+        """Quote a string."""
         return f"{self.quote}{value}{self.quote}"
 
     def _write_array_string(self, string_list) -> str:
-        """
-        Abstract method to output.write the string representation of an array into a .csv file
-        as required by the postgresql COPY command, with '{','}' brackets and ',' separation.
+        """Write the string representation of an array into a .csv file.
+
+        Abstract method to output.write the string representation of an array
+        into a .csv file as required by the postgresql COPY command, with
+        '{','}' brackets and ',' separation.
 
         Args:
+        ----
             string_list (list): list of ontology strings
 
         Returns:
+        -------
             str: The string representation of an array for postgres COPY
+
         """
         string = ",".join(string_list)
         string = f'"{{{string}}}"'
         return string
 
     def _get_import_script_name(self) -> str:
-        """
-        Returns the name of the psql import script
+        """Return the name of the psql import script.
 
-        Returns:
+        Returns
+        -------
             str: The name of the import script (ending in .sh)
+
         """
         return f"{self.db_name}-import-call.sh"
 
@@ -91,14 +96,17 @@ class _PostgreSQLBatchWriter(_BatchWriter):
         string = string.lower()
         return string
 
-    def _write_node_headers(self):
-        """
+    def _write_node_headers(self) -> bool:
+        """Write node header files for PostgreSQL.
+
         Writes single CSV file for a graph entity that is represented
         as a node as per the definition in the `schema_config.yaml`,
         containing only the header for this type of node.
 
-        Returns:
+        Returns
+        -------
             bool: The return value. True for success, False otherwise.
+
         """
         # load headers from data parse
         if not self.node_property_dict:
@@ -158,7 +166,7 @@ class _PostgreSQLBatchWriter(_BatchWriter):
                         )
 
                     self._copy_from_csv_commands.add(
-                        f"\\copy {pascal_label} FROM '{parts_path}' DELIMITER E'{self.delim}' CSV;"
+                        f"\\copy {pascal_label} FROM '{parts_path}' DELIMITER E'{self.delim}' CSV;",
                     )
 
             # add file path to import statement
@@ -175,13 +183,14 @@ class _PostgreSQLBatchWriter(_BatchWriter):
         return True
 
     def _write_edge_headers(self):
-        """
-        Writes single CSV file for a graph entity that is represented
+        """Writes single CSV file for a graph entity that is represented
         as an edge as per the definition in the `schema_config.yaml`,
         containing only the header for this type of edge.
 
-        Returns:
+        Returns
+        -------
             bool: The return value. True for success, False otherwise.
+
         """
         # load headers from data parse
         if not self.edge_property_dict:
@@ -221,7 +230,7 @@ class _PostgreSQLBatchWriter(_BatchWriter):
                     raise ValueError(
                         "Column name '_ID' is reserved for internal use, "
                         "denoting the relationship ID. Please choose a "
-                        "different name for your column."
+                        "different name for your column.",
                     )
 
                 columns.append(f"{col_name} {col_type}")
@@ -255,7 +264,7 @@ class _PostgreSQLBatchWriter(_BatchWriter):
                         )
 
                     self._copy_from_csv_commands.add(
-                        f"\\copy {pascal_label} FROM '{parts_path}' DELIMITER E'{self.delim}' CSV;"
+                        f"\\copy {pascal_label} FROM '{parts_path}' DELIMITER E'{self.delim}' CSV;",
                     )
 
             # add file path to import statement
@@ -272,14 +281,15 @@ class _PostgreSQLBatchWriter(_BatchWriter):
         return True
 
     def _construct_import_call(self) -> str:
-        """
-        Function to construct the import call detailing folder and
+        """Function to construct the import call detailing folder and
         individual node and edge headers and data files, as well as
         delimiters and database name. Built after all data has been
         processed to ensure that nodes are called before any edges.
 
-        Returns:
+        Returns
+        -------
             str: a bash command for postgresql import
+
         """
         import_call = ""
 

{biocypher-0.9.4.dist-info → biocypher-0.9.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: biocypher
-Version: 0.9.4
+Version: 0.9.6
 Summary: A unifying framework for biomedical research knowledge graphs
 Home-page: https://github.com/biocypher/biocypher
 License: MIT

{biocypher-0.9.4.dist-info → biocypher-0.9.6.dist-info}/RECORD

@@ -8,19 +8,19 @@ biocypher/_config/test_schema_config_extended.yaml,sha256=wn3A76142hhjnImhMF6ROD
 biocypher/_core.py,sha256=lP5CC35oXPtfiFY3C-UxZev8p1nMg9E2E4qN5BD9BTg,27437
 biocypher/_create.py,sha256=QsvXrwEQ8k0uNXvCG06UKejvw-QsJwzSaumrBjx9n1k,9884
 biocypher/_deduplicate.py,sha256=rtglcaLRaVzNjLtaPwTGP8VvCM4PHYQ5CZ-cm32CrKQ,4840
-biocypher/_get.py,sha256=BDNnvw6Xbz_Lq0ekU1Y60J4DI1zQKiZugJV6zGnWPsI,14873
+biocypher/_get.py,sha256=_wUjhRjH2J6Qhq0Ndy3kdfaWhHDTT-dxyCvtuH36My4,14868
 biocypher/_logger.py,sha256=y9dh3SPJOCWXnkFSYSK7aj_-pB7zlAkNCf43Dp1lt74,2941
 biocypher/_mapping.py,sha256=ntspG2C_NaQODhWTBFk0CDvolkOCjtqlQ9E-NkJAuTg,9030
-biocypher/_metadata.py,sha256=BUH21dOZRsE7nvKZhWyqRfIp1j1ZqzsfMf8EVi7P-ok,1415
+biocypher/_metadata.py,sha256=Jb4Uva2PzrPbxzio7DMQnX0WuIFrl_pgyDW89L1R1oQ,1415
 biocypher/_misc.py,sha256=YzlY7zwa0mim9QFg9HwXErkJFIH3cvLrbgjF8tKOIT8,6353
 biocypher/_ontology.py,sha256=lipZxU3aj6zrTbBrJZmCW6IRCuz-KQG3AfbYCVq6aFE,33133
 biocypher/_translate.py,sha256=9E19eLRL0VnxxDuiNhZ5vu54XyKXnfLuBhCgNcL9yAE,17000
 biocypher/output/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biocypher/output/connect/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-biocypher/output/connect/_get_connector.py,sha256=r-DeDnvGyFEWOd9bXaXQOgVws9rgWz5kwwdzbmt5eKk,1038
+biocypher/output/connect/_get_connector.py,sha256=Qimv3kTkXYkhJZRT6nq8mwIM2wORCnyqqHqF2IByuuc,1152
 biocypher/output/connect/_neo4j_driver.py,sha256=kXjOXW12wZFfEp7plAuo40bPSvOfd-i9m4YaXoMq-p0,12357
 biocypher/output/in_memory/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-biocypher/output/in_memory/_get_in_memory_kg.py,sha256=JuNwVx-qABeLikkI9VXQJNQH8xQWvYLEK_KKz3BrXPI,937
+biocypher/output/in_memory/_get_in_memory_kg.py,sha256=29DvmOdRUuzEblyMxy5001R4zjEXW3eHNv0htko7c4Y,1115
 biocypher/output/in_memory/_in_memory_kg.py,sha256=g1TPN8PkeAyXbrRuTAjshqC8voI6EmLqR8S_otmviwU,1423
 biocypher/output/in_memory/_networkx.py,sha256=cSOSAreP7S3oeGT6noZ1kAIvSnkVnU3NUp1OY4yqzn0,1515
 biocypher/output/in_memory/_pandas.py,sha256=Ot2jbK5t_YLHqw0BUv9Z_qWNy9r6IX1LYEyejOSJzos,3288
@@ -30,15 +30,15 @@ biocypher/output/write/_get_writer.py,sha256=JozRWCMhvh65aQAlcGiiD5x3Nl1HSW8mK1Z
 biocypher/output/write/_writer.py,sha256=y0dWI-RyQdrBLr9Fs91Y9KcCMjnlCaKJT0eWsIS2hG4,7158
 biocypher/output/write/graph/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biocypher/output/write/graph/_arangodb.py,sha256=xue3hm_DVB5pMR5qqfGXlXll3RpILA0tXos2J-as1-E,7906
-biocypher/output/write/graph/_neo4j.py,sha256=EK5gqQNyVMYfpH1DaDTtGfRKiiq4jx5DLtYCCjY-jbY,12081
+biocypher/output/write/graph/_neo4j.py,sha256=tBPhxn8JAmSS6KmiePofwr9LpGjHQH9BTnpHVK2ellM,12042
 biocypher/output/write/graph/_networkx.py,sha256=2WYkw5ZM3Bp236iwAxEAp3A1DxHKT4_hEPNMUKvPHp4,2320
 biocypher/output/write/graph/_owl.py,sha256=2DlxQuAXGnCW068N8bPgADjk_LFhU9D_LJV5lWQeu4A,21333
 biocypher/output/write/graph/_rdf.py,sha256=1TgECkoTHFX0eXtWc_-pr9G55AesReGlh7rg5zvUj5w,21925
 biocypher/output/write/relational/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 biocypher/output/write/relational/_csv.py,sha256=m0BSQXts88Qu5AEvoIgnwRz54ia38g4VN3PaA3LCYM8,2807
-biocypher/output/write/relational/_postgresql.py,sha256=RckQJBiuwvDmHAyXxS8zCavYqDecHHWW_piofurokfQ,11965
+biocypher/output/write/relational/_postgresql.py,sha256=75iJvv-0ewSsQXhVeoYoGnmYQnKY_B4iItZV7DpEBto,12190
 biocypher/output/write/relational/_sqlite.py,sha256=BuGWOeeNA83lbUvjpkzqcR9_baWLsbfmLXBKe4O1EPE,2105
-biocypher-0.9.4.dist-info/LICENSE,sha256=oejgxuxyjSnyPw3YPloz6-dCBB_nYizJ4jDQnr-xZUU,1082
-biocypher-0.9.4.dist-info/METADATA,sha256=0RHGsElP0N8_NFH_NtFR4aHhfT-WmPBhlrkDYo9DRfU,10600
-biocypher-0.9.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-biocypher-0.9.4.dist-info/RECORD,,
+biocypher-0.9.6.dist-info/LICENSE,sha256=oejgxuxyjSnyPw3YPloz6-dCBB_nYizJ4jDQnr-xZUU,1082
+biocypher-0.9.6.dist-info/METADATA,sha256=8zvMLLWli78WBCjcBMhkqdwQb2rEqIinHz4aePh8RZw,10600
+biocypher-0.9.6.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+biocypher-0.9.6.dist-info/RECORD,,