feat: add pre-commit-hook

httprunner/step.py

@@ -64,4 +64,4 @@ class Step(object):
         return self.__step.type()
 
     def run(self, runner: HttpRunner) -> StepResult:
-        return self.__step.run(runner)
+        return self.__step.run(runner)

httprunner/step_sql_request.py

@@ -11,7 +11,11 @@ from httprunner.models import IStep, StepResult, TStep
 from httprunner.models import SqlMethodEnum, TSqlRequest
 from httprunner.response import SqlResponseObject
 from httprunner.runner import HttpRunner
-from httprunner.step_request import (StepRequestExtraction, StepRequestValidation, call_hooks)
+from httprunner.step_request import (
+    StepRequestExtraction,
+    StepRequestValidation,
+    call_hooks,
+)
 
 try:
     import sqlalchemy
@@ -73,6 +77,7 @@ def run_step_sql_request(runner: HttpRunner, step: TStep) -> StepResult:
     if not runner.db_engine:
         ensure_sql_ready()
         from httprunner.database.engine import DBEngine
+
         runner.db_engine = DBEngine(
             f'mysql+pymysql://{parsed_request_dict["db_config"]["user"]}:'
             f'{parsed_request_dict["db_config"]["password"]}@{parsed_request_dict["db_config"]["ip"]}:'

httprunner/step_thrift_request.py

@@ -66,30 +66,30 @@ def run_step_thrift_request(runner: HttpRunner, step: TStep) -> StepResult:
     parsed_request_dict["psm"] = parsed_request_dict["psm"] or config.thrift.psm
     parsed_request_dict["env"] = parsed_request_dict["env"] or config.thrift.env
     parsed_request_dict["cluster"] = (
-            parsed_request_dict["cluster"] or config.thrift.cluster
+        parsed_request_dict["cluster"] or config.thrift.cluster
     )
     parsed_request_dict["idl_path"] = (
-            parsed_request_dict["idl_path"] or config.thrift.idl_path
+        parsed_request_dict["idl_path"] or config.thrift.idl_path
     )
     parsed_request_dict["include_dirs"] = (
-            parsed_request_dict["include_dirs"] or config.thrift.include_dirs
+        parsed_request_dict["include_dirs"] or config.thrift.include_dirs
     )
     parsed_request_dict["method"] = (
-            parsed_request_dict["method"] or config.thrift.method
+        parsed_request_dict["method"] or config.thrift.method
     )
     parsed_request_dict["service_name"] = (
-            parsed_request_dict["service_name"] or config.thrift.service_name
+        parsed_request_dict["service_name"] or config.thrift.service_name
     )
     parsed_request_dict["ip"] = parsed_request_dict["ip"] or config.thrift.ip
     parsed_request_dict["port"] = parsed_request_dict["port"] or config.thrift.port
     parsed_request_dict["proto_type"] = (
-            parsed_request_dict["proto_type"] or config.thrift.proto_type
+        parsed_request_dict["proto_type"] or config.thrift.proto_type
     )
     parsed_request_dict["trans_port"] = (
-            parsed_request_dict["trans_type"] or config.thrift.trans_type
+        parsed_request_dict["trans_type"] or config.thrift.trans_type
     )
     parsed_request_dict["timeout"] = (
-            parsed_request_dict["timeout"] or config.thrift.timeout
+        parsed_request_dict["timeout"] or config.thrift.timeout
     )
     parsed_request_dict["thrift_client"] = parsed_request_dict["thrift_client"]
 
@@ -105,6 +105,7 @@ def run_step_thrift_request(runner: HttpRunner, step: TStep) -> StepResult:
     if not runner.thrift_client:
         ensure_thrift_ready()
         from httprunner.thrift.thrift_client import ThriftClient
+
         runner.thrift_client = ThriftClient(
             thrift_file=parsed_request_dict["idl_path"],
             service_name=parsed_request_dict["service_name"],
@@ -214,7 +215,7 @@ class RunThriftRequest(IStep):
         return self
 
     def teardown_hook(
-            self, hook: Text, assign_var_name: Text = None
+        self, hook: Text, assign_var_name: Text = None
     ) -> "RunThriftRequest":
         if assign_var_name:
             self.__step.teardown_hooks.append({assign_var_name: hook})
@@ -224,7 +225,7 @@ class RunThriftRequest(IStep):
         return self
 
     def setup_hook(
-            self, hook: Text, assign_var_name: Text = None
+        self, hook: Text, assign_var_name: Text = None
     ) -> "RunThriftRequest":
         if assign_var_name:
             self.__step.setup_hooks.append({assign_var_name: hook})
@@ -247,7 +248,7 @@ class RunThriftRequest(IStep):
         return self
 
     def with_thrift_client(
-            self, thrift_client: Union["ThriftClient", str]
+        self, thrift_client: Union["ThriftClient", str]
     ) -> "RunThriftRequest":
         self.__step.thrift_request.thrift_client = thrift_client
         return self
@@ -287,7 +288,7 @@ class RunThriftRequest(IStep):
         return StepThriftRequestValidation(self.__step)
 
     def with_jmespath(
-            self, jmes_path: Text, var_name: Text
+        self, jmes_path: Text, var_name: Text
     ) -> "StepThriftRequestExtraction":
         self.__step.extract[var_name] = jmes_path
         return StepThriftRequestExtraction(self.__step)