From ffb0da4a350278d9f4fac36b651b52d4dc64bd61 Mon Sep 17 00:00:00 2001
From: Yu Steven <78414278+yu-steven@users.noreply.github.com>
Date: Sun, 10 Jul 2022 22:50:31 +0800
Subject: [PATCH] Update main.py

---
 utils/pool/main.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/utils/pool/main.py b/utils/pool/main.py
index 6209f8d91..a156dff7f 100644
--- a/utils/pool/main.py
+++ b/utils/pool/main.py
@@ -7,10 +7,12 @@ import time
 import yaml
 import requests
 
+headers = {'User-Agent': 'clash', 'Accept': '*/*', 'Accept-Encoding': 'gzip, deflate, br', 'Connection': 'keep-alive'}
+
 def fetch(proxy_list, filename):
     current_date = time.strftime("%Y_%m_%d", time.localtime())
     baseurl = 'https://raw.githubusercontent.com/changfengoss/pub/main/data/'
-    working = yaml.safe_load(requests.get(url=baseurl + current_date + '/' + filename, ).text)
+    working = yaml.safe_load(requests.get(url=baseurl + current_date + '/' + filename, timeout=10).text)
     data_out = []
     for x in working['proxies']:
         data_out.append(x)
@@ -19,7 +21,7 @@ def fetch(proxy_list, filename):
 def url(proxy_list, link):
     try:
         data_out=[]
-        working = yaml.safe_load(requests.get(url=link).text)
+        working = yaml.safe_load(requests.get(url=link,timeout=10, headers=headers).text)
         for x in working['proxies']:
             data_out.append(x)
         proxy_list.append(data_out)