From cefb9adab4a68d5a04893cc1be75fb6cce9980d6 Mon Sep 17 00:00:00 2001 From: cesar <cesaryuan@qq.com> Date: Sun, 16 Jun 2024 23:19:50 +0800 Subject: [PATCH 1/2] Fix ModuleNotFoundError when running reqable script directly --- reqable/__init__.py | 2 +- reqable/addons.py | 6 +++--- reqable/addons_mini.py | 6 +++--- reqable/main.py | 4 ++-- 4 files changed, 9 insertions(+), 9 deletions(-) diff --git a/reqable/__init__.py b/reqable/__init__.py index f317161..d9242cf 100644 --- a/reqable/__init__.py +++ b/reqable/__init__.py @@ -1,4 +1,4 @@ -import main +from reqable import main if __name__== '__main__': main.main() \ No newline at end of file diff --git a/reqable/addons.py b/reqable/addons.py index 45d88aa..468e22f 100644 --- a/reqable/addons.py +++ b/reqable/addons.py @@ -1,8 +1,8 @@ # API Docs: https://reqable.com/docs/capture/addons -from reqable import * +from reqable.reqable import CaptureContext, CaptureHttpRequest, CaptureHttpResponse -def onRequest(context, request): +def onRequest(context: CaptureContext, request: CaptureHttpRequest) -> CaptureHttpRequest: # Print url to console # print('request url ' + context.url) @@ -26,7 +26,7 @@ def onRequest(context, request): # Done return request -def onResponse(context, response): +def onResponse(context: CaptureContext, response: CaptureHttpResponse) -> CaptureHttpResponse: # Update status code # response.code = 404 diff --git a/reqable/addons_mini.py b/reqable/addons_mini.py index e7a0446..eff3266 100644 --- a/reqable/addons_mini.py +++ b/reqable/addons_mini.py @@ -1,9 +1,9 @@ # API Docs: https://reqable.com/docs/capture/addons -from reqable import * +from reqable.reqable import CaptureContext, CaptureHttpRequest, CaptureHttpResponse -def onRequest(context, request): +def onRequest(context: CaptureContext, request: CaptureHttpRequest) -> CaptureHttpRequest: return request -def onResponse(context, response): +def onResponse(context: CaptureContext, response: CaptureHttpResponse) -> CaptureHttpResponse: return response \ No newline at end of file diff --git a/reqable/main.py b/reqable/main.py index babcdd3..fe93435 100644 --- a/reqable/main.py +++ b/reqable/main.py @@ -1,7 +1,7 @@ import sys import json -from reqable import CaptureContext, CaptureHttpRequest, CaptureHttpResponse -import addons +from reqable.reqable import CaptureContext, CaptureHttpRequest, CaptureHttpResponse +from reqable import addons def main(): argv = sys.argv[1:] From 74d9e06dd144400c951705f2002939ec0cab52fa Mon Sep 17 00:00:00 2001 From: cesar <cesaryuan@qq.com> Date: Mon, 17 Jun 2024 00:02:10 +0800 Subject: [PATCH 2/2] Fix ModuleNotFoundError when running reqable script directly --- reqable/__init__.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/reqable/__init__.py b/reqable/__init__.py index f317161..0dbeeda 100644 --- a/reqable/__init__.py +++ b/reqable/__init__.py @@ -1,4 +1 @@ -import main - -if __name__== '__main__': - main.main() \ No newline at end of file +from reqable.reqable import * \ No newline at end of file