Browse Source

send-to-bot.py: use requests instead of urllib (#6184)

Stig124 3 years ago
parent
commit
f4390b1637
1 changed files with 9 additions and 14 deletions
  1. 9 14
      scripts/send-to-bot.py

+ 9 - 14
scripts/send-to-bot.py

@@ -3,8 +3,7 @@
 
 import os
 import sys
-import json
-import urllib.request
+import requests
 
 BOT_URL = 'https://tldr-bot.starbeamrainbowlabs.com'
 
@@ -34,22 +33,18 @@ def post_comment(pr_id, body, once):
   if once:
     endpoint += '/once'
 
-  headers = {'Content-Type': 'application/json'}
-  data = json.dumps({'pr_id': pr_id, 'body': body})
-  req = urllib.request.Request(endpoint, data.encode(), headers)
+  data = {'pr_id': pr_id, 'body': body}
 
   try:
-    resp = urllib.request.urlopen(req)
-    code = resp.getcode()
-  except Exception as e:
+    with requests.post(endpoint, json=data) as r:
+      if r.status_code != requests.codes.ok:
+        print('Error: tldr-bot responded with code', r.status_code, file=sys.stderr)
+        print(r.text, file=sys.stderr)
+        return False
+  except requests.exceptions.RequestException as e:
     print('Error sending data to tldr-bot:', str(e), file=sys.stderr)
     return False
-
-  if code != 200:
-    print('Error: tldr-bot responded with code', code, file=sys.stderr)
-    print(resp.read(), file=sys.stderr)
-    return False
-
+    
   return True
 
 def main(action):