From 68f5626189c50c2429ffbf2c40a7807f6697f914 Mon Sep 17 00:00:00 2001 From: jesopo Date: Wed, 10 Oct 2018 13:41:58 +0100 Subject: Change utils.http to use requests --- modules/eval_lua.py | 2 +- modules/eval_rust.py | 23 +++++++++++++---------- 2 files changed, 14 insertions(+), 11 deletions(-) (limited to 'modules') diff --git a/modules/eval_lua.py b/modules/eval_lua.py index bb5dd868..5208d00e 100644 --- a/modules/eval_lua.py +++ b/modules/eval_lua.py @@ -9,7 +9,7 @@ class Module(ModuleManager.BaseModule): def eval(self, event): try: page = utils.http.get_url(EVAL_URL, - post_params={"input": event["args"]}, + post_data={"input": event["args"]}, method="POST", soup=True) except socket.timeout: diff --git a/modules/eval_rust.py b/modules/eval_rust.py index d0399b95..0b9d03d9 100644 --- a/modules/eval_rust.py +++ b/modules/eval_rust.py @@ -9,6 +9,15 @@ fn main() { }); } """ +API_ARGS = { + "channel": "nightly", + "crateType": "bin", + "mode": "debug", + "tests": False, + "execute": True, + "target": "ast", + "backtrace": False +} class Module(ModuleManager.BaseModule): _name = "Rust" @@ -18,17 +27,11 @@ class Module(ModuleManager.BaseModule): :help: Evaluate a rust statement :usage: """ + args = API_ARGS.copy() + args["code"] = FN_TEMPLATE % event["args"] try: - page = utils.http.get_url(EVAL_URL, post_data=json.dumps({ - "code": FN_TEMPLATE % event["args"], - "channel": "nightly", - "crateType": "bin", - "mode": "debug", - "tests": False, - "execute": True, - "target": "ast", - "backtrace": False - }), method="POST", json=True) + page = utils.http.get_url(EVAL_URL, json_data=args, + method="POST", json=True) except socket.timeout: event["stderr"].write("%s: eval timed out" % event["user"].nickname) -- cgit v1.3.1-10-gc9f91