format knext code with black

Author: xionghuaidong

knext/builder/client.py

@@ -31,7 +31,7 @@ def write_graph(self, sub_graph: dict, operation: str, lead_to_builder: bool):
             sub_graph=sub_graph,
             operation=operation,
             enable_lead_to=lead_to_builder,
-            token="openspg@8380255d4e49_"
+            token="openspg@8380255d4e49_",
         )
         self._rest_client.builder_job_writer_graph_post(writer_graph_request=request)
 

knext/builder/rest/models/writer_graph_request.py

@@ -48,25 +48,25 @@ class WriterGraphRequest(object):
         "operation": "str",
         "sub_graph": "object",
         "enable_lead_to": "bool",
-        "token": "str"
+        "token": "str",
     }
 
     attribute_map = {
         "project_id": "projectId",
         "operation": "operation",
         "sub_graph": "subGraph",
         "enable_lead_to": "enableLeadTo",
-        "token": "token"
+        "token": "token",
     }
 
     def __init__(
-            self,
-            project_id=None,
-            operation=None,
-            sub_graph=None,
-            enable_lead_to=None,
-            token=None,
-            local_vars_configuration=None,
+        self,
+        project_id=None,
+        operation=None,
+        sub_graph=None,
+        enable_lead_to=None,
+        token=None,
+        local_vars_configuration=None,
     ):  # noqa: E501
         """WriterGraphRequest - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:

knext/command/sub_command/project.py

@@ -39,7 +39,7 @@
     pass
 
 yaml = YAML()
-yaml.default_flow_style = False 
+yaml.default_flow_style = False
 yaml.indent(mapping=2, sequence=4, offset=2)
 
 logger = logging.getLogger(__name__)
@@ -104,13 +104,13 @@ def _recover_project(prj_path: str):
         sys.exit()
     client = ProjectClient(host_addr=env.host_addr, project_id=env.project_id)
     project = client.get(namespace=namespace) or client.create(
-        name=project_name, 
-        desc=desc, 
-        namespace=namespace, 
-        config=env._config, 
-        visibility=env.project_config.get("visibility", "PRIVATE"), 
-        tag=env.project_config.get("tag", "LOCAL"), 
-        userNo=env.project_config.get("userNo", "openspg")
+        name=project_name,
+        desc=desc,
+        namespace=namespace,
+        config=env._config,
+        visibility=env.project_config.get("visibility", "PRIVATE"),
+        tag=env.project_config.get("tag", "LOCAL"),
+        userNo=env.project_config.get("userNo", "openspg"),
     )
 
     env._config["project"]["id"] = project.id
@@ -177,12 +177,14 @@ def create_project(
 
     if host_addr:
         client = ProjectClient(host_addr=host_addr)
-        project = client.create(name=name, 
-                                namespace=namespace, 
-                                config=config, 
-                                visibility=env.project_config.get("visibility", "PRIVATE"), 
-                                tag=env.project_config.get("tag", "LOCAL"), 
-                                userNo=env.project_config.get("userNo", "openspg"))
+        project = client.create(
+            name=name,
+            namespace=namespace,
+            config=config,
+            visibility=env.project_config.get("visibility", "PRIVATE"),
+            tag=env.project_config.get("tag", "LOCAL"),
+            userNo=env.project_config.get("userNo", "openspg"),
+        )
 
         if project and project.id:
             project_id = project.id
@@ -230,12 +232,14 @@ def restore_project(host_addr, proj_path):
     if not project_wanted:
         if host_addr:
             client = ProjectClient(host_addr=host_addr)
-            project = client.create(name=env.name, 
-                                    namespace=env.namespace, 
-                                    config=env._config, 
-                                    visibility=env.project_config.get("visibility", "PRIVATE"), 
-                                    tag=env.project_config.get("tag", "LOCAL"), 
-                                    userNo=env.project_config.get("userNo", "openspg"))
+            project = client.create(
+                name=env.name,
+                namespace=env.namespace,
+                config=env._config,
+                visibility=env.project_config.get("visibility", "PRIVATE"),
+                tag=env.project_config.get("tag", "LOCAL"),
+                userNo=env.project_config.get("userNo", "openspg"),
+            )
             project_id = project.id
     else:
         project_id = project_wanted.id
@@ -267,17 +271,23 @@ def update_project(proj_path):
         sys.exit()
 
     logger.info(f"project id: {env.id}")
-    client.update(id=env.id, namespace=env.namespace, config=env._config, visibility=env.project_config.get("visibility", "PRIVATE"), tag=env.project_config.get("tag", "LOCAL"), userNo=env.project_config.get("userNo", "openspg"))
+    client.update(
+        id=env.id,
+        namespace=env.namespace,
+        config=env._config,
+        visibility=env.project_config.get("visibility", "PRIVATE"),
+        tag=env.project_config.get("tag", "LOCAL"),
+        userNo=env.project_config.get("userNo", "openspg"),
+    )
     click.secho(
         f"Project [{env.name}] with namespace [{env.namespace}] was successfully updated from [{proj_path}].",
         fg="bright_green",
     )
 
+
 @click.option("--host_addr", help="Address of spg server.", default=DEFAULT_HOST_ADDR)
 def list_project(host_addr):
-    client = ProjectClient(
-        host_addr=host_addr
-    )
+    client = ProjectClient(host_addr=host_addr)
     projects = client.get_all()
 
     headers = ["Project Name", "Project ID"]

knext/command/sub_command/schema.py

@@ -32,7 +32,9 @@ def commit_schema():
     )
 
     if Path(schema_file).exists():
-        ml = SPGSchemaMarkLang(schema_file, host_addr=env.host_addr, project_id=env.project_id)
+        ml = SPGSchemaMarkLang(
+            schema_file, host_addr=env.host_addr, project_id=env.project_id
+        )
     else:
         ml = None
     index_managers = KAGIndexManager.list_available()
@@ -49,7 +51,12 @@ def commit_schema():
         if not schema_str:
             continue
         schema_str = f"namespace {env.namespace}\n" + schema_str
-        cur_index_ml = SPGSchemaMarkLang(filename="", script_data_str=schema_str, host_addr=env.host_addr, project_id=env.project_id)
+        cur_index_ml = SPGSchemaMarkLang(
+            filename="",
+            script_data_str=schema_str,
+            host_addr=env.host_addr,
+            project_id=env.project_id,
+        )
         if index_ml is None:
             index_ml = cur_index_ml
         else:

knext/common/env.py

@@ -19,7 +19,7 @@
 from typing import Union, Optional
 
 yaml = YAML()
-yaml.default_flow_style = False 
+yaml.default_flow_style = False
 yaml.indent(mapping=2, sequence=4, offset=2)
 logger = logging.getLogger(__name__)
 
@@ -36,9 +36,11 @@ def __new__(cls):
             try:
                 log_config = cls._instance.config.get("log", {})
                 value = log_config.get("level", "INFO")
-                logging.basicConfig(level=logging.getLevelName(value),
-                                    format='%(asctime)s - %(levelname)s - %(name)s - %(message)s',
-                                    datefmt='%Y-%m-%d %H:%M:%S')
+                logging.basicConfig(
+                    level=logging.getLevelName(value),
+                    format="%(asctime)s - %(levelname)s - %(name)s - %(message)s",
+                    datefmt="%Y-%m-%d %H:%M:%S",
+                )
             except:
                 logger.info("logger info not set")
         return cls._instance
@@ -47,13 +49,13 @@ def __new__(cls):
     def config(self):
 
         closest_config = self._closest_config()
-        if not hasattr(self, '_config_path') or self._config_path != closest_config:
+        if not hasattr(self, "_config_path") or self._config_path != closest_config:
             self._config_path = closest_config
             self._config = self.get_config()
-            
+
         if self._config is None:
             self._config = self.get_config()
-                
+
         return self._config
 
     @property
@@ -63,7 +65,7 @@ def project_path(self):
 
     @property
     def config_path(self):
-        if not hasattr(self, '_config_path') or self._config_path is None:
+        if not hasattr(self, "_config_path") or self._config_path is None:
             self._config_path = self._closest_config()
         return self._config_path
 
@@ -77,7 +79,7 @@ def id(self):
         id = self.project_config.get("id", None)
         if id is None:
             logger.warning("can not find id in project config")
-            if os.getenv("KAG_PROJECT_ID",None):
+            if os.getenv("KAG_PROJECT_ID", None):
                 return os.getenv("KAG_PROJECT_ID")
             else:
                 raise Exception(
@@ -137,11 +139,16 @@ def _closest_config(
             return ""
         path = Path(path).resolve()
         cfg_files = list(path.glob("*.yaml"))
-        cfg_file = next((f for f in cfg_files if f.name == "kag_config.yaml"), cfg_files[0] if cfg_files else None)
+        cfg_file = next(
+            (f for f in cfg_files if f.name == "kag_config.yaml"),
+            cfg_files[0] if cfg_files else None,
+        )
         if cfg_file and cfg_file.exists():
             return str(cfg_file)
         if path.parent == path:
-            raise FileNotFoundError("No kag_config.yaml file found in current directory or any parent directories")
+            raise FileNotFoundError(
+                "No kag_config.yaml file found in current directory or any parent directories"
+            )
         return self._closest_config(path.parent, path)
 
     def dump(self, path=None, **kwargs):

knext/graph/client.py

@@ -50,7 +50,9 @@ def calculate_pagerank_scores(self, target_vertex_type, start_nodes: List[Dict])
         such as failures in running `run_pagerank_igraph_chunk`, are logged.
         """
         ppr_start_nodes = [
-            GetPageRankScoresRequestStartNodes(id=node.get("id",node["name"]), type=node["type"])
+            GetPageRankScoresRequestStartNodes(
+                id=node.get("id", node["name"]), type=node["type"]
+            )
             for node in start_nodes
         ]
         req = GetPageRankScoresRequest(
@@ -67,7 +69,7 @@ def write_graph(self, sub_graph: dict, operation: str, lead_to_builder: bool):
             sub_graph=sub_graph,
             operation=operation,
             enable_lead_to=lead_to_builder,
-            token="openspg@8380255d4e49_"
+            token="openspg@8380255d4e49_",
         )
         self._rest_client.graph_writer_graph_post(writer_graph_request=request)
 

knext/graph/rest/models/writer_graph_request.py

@@ -48,25 +48,25 @@ class WriterGraphRequest(object):
         "operation": "str",
         "sub_graph": "object",
         "enable_lead_to": "bool",
-        "token": "str"
+        "token": "str",
     }
 
     attribute_map = {
         "project_id": "projectId",
         "operation": "operation",
         "sub_graph": "subGraph",
         "enable_lead_to": "enableLeadTo",
-        "token": "token"
+        "token": "token",
     }
 
     def __init__(
-            self,
-            project_id=None,
-            operation=None,
-            sub_graph=None,
-            enable_lead_to=None,
-            token=None,
-            local_vars_configuration=None,
+        self,
+        project_id=None,
+        operation=None,
+        sub_graph=None,
+        enable_lead_to=None,
+        token=None,
+        local_vars_configuration=None,
     ):  # noqa: E501
         """WriterGraphRequest - a model defined in OpenAPI"""  # noqa: E501
         if local_vars_configuration is None:

knext/project/client.py

@@ -76,8 +76,14 @@ def create(self, name: str, namespace: str, config: str, **kwargs):
         desc = kwargs.get("desc", None)
         userNo = kwargs.get("userNo", "openspg")
         project_create_request = rest.ProjectCreateRequest(
-            name=name, desc=desc, namespace=namespace, config=config, auto_schema=auto_schema, visibility=visibility,
-            tag=tag, userNo=userNo
+            name=name,
+            desc=desc,
+            namespace=namespace,
+            config=config,
+            auto_schema=auto_schema,
+            visibility=visibility,
+            tag=tag,
+            userNo=userNo,
         )
 
         project = self._rest_client.project_create_post(
@@ -88,8 +94,14 @@ def create(self, name: str, namespace: str, config: str, **kwargs):
     def update(self, id, namespace, config, visibility=None, tag=None, userNo=None):
         if not self._rest_client:
             raise Exception("Please set KAG_PROJECT_HOST_ADDR and KAG_PROJECT_ID")
-        project_create_request = rest.ProjectCreateRequest(id=id, name=namespace, config=config, visibility=visibility,
-                                                           tag=tag, userNo=userNo)
+        project_create_request = rest.ProjectCreateRequest(
+            id=id,
+            name=namespace,
+            config=config,
+            visibility=visibility,
+            tag=tag,
+            userNo=userNo,
+        )
         project = self._rest_client.project_update_post(
             project_create_request=project_create_request
         )