From 822b01de0879a6c19bc5d222f2b23ced72e69f17 Mon Sep 17 00:00:00 2001
From: XuanRan_Nyan <31968665+xxxxuanran@users.noreply.github.com>
Date: Thu, 27 Jul 2023 22:59:35 +0800
Subject: [PATCH] Fix Huya (#555)

---
 biliup/engine/download.py | 2 +-
 biliup/plugins/huya.py    | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/biliup/engine/download.py b/biliup/engine/download.py
index 025a0bb..45c609e 100644
--- a/biliup/engine/download.py
+++ b/biliup/engine/download.py
@@ -39,7 +39,7 @@ class DownloadBase:
             'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,*/*;q=0.8',
             'Accept-Encoding': 'gzip, deflate',
             'Accept-Language': 'zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3',
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.159 Safari/537.36'
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/92.0.4515.159 Safari/537.36',
         }
 
         self.default_output_args = [
diff --git a/biliup/plugins/huya.py b/biliup/plugins/huya.py
index aed9e2c..6a8ce12 100644
--- a/biliup/plugins/huya.py
+++ b/biliup/plugins/huya.py
@@ -16,9 +16,9 @@ class Huya(DownloadBase):
     def __init__(self, fname, url, suffix='flv'):
         super().__init__(fname, url, suffix)
         self.huya_danmaku = config.get('huya_danmaku', False)
+        self.fake_headers['Referer'] = 'https://www.huya.com/'
 
     def check_stream(self):
-        logger.debug(self.fname)
         try:
             res = requests.get(self.url, timeout=5, headers=self.fake_headers)
             res.close()