修复 #30 弹幕下载失败的问题

itKelis · May 9, 2024 · d9f3222 · d9f3222
1 parent caadab2
commit d9f3222
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/scripts/bilibiliAssert/Danmu2Ass.py b/scripts/bilibiliAssert/Danmu2Ass.py
@@ -432,9 +432,12 @@ def export(func):
 
 
 def getComments(cid,font_size = 25):
-    # url = 'https://comment.bilibili.com/{}.xml'.format(cid[0])
-    url = ''.join(['https://comment.bilibili.com/',cid[0],'.xml'])
-    response = request.urlopen(url, context=ssl.SSLContext(ssl.PROTOCOL_TLS))
+    response = request.urlopen(request.Request(
+        url = ''.join(['https://comment.bilibili.com/',cid[0],'.xml']),
+        headers = {
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36 Edg/124.0.0.0'
+        }
+    ))
     data = str(zlib.decompress(response.read(), -zlib.MAX_WBITS), "utf-8")
     response.close()
     comments = []