check for none

2021-05-12 01:08:20 +02:00 · 2021-05-12 01:08:20 +02:00 · d080d51287
commit d080d51287
parent 832cb49357
1 changed files with 4 additions and 1 deletions
--- a/stream.py
+++ b/stream.py
@ -87,7 +87,10 @@ def get_proxy_url(proxy, current, path):
    return presp.text

 def upstream_type(current, proxy):
-    resp = requests.head(current, proxies=proxy.req)
+    if proxy is None:
+        resp = requests.head(current)
+    else:
+        resp = requests.head(current, proxies=proxy.req)
    return resp.headers.get("Content-Type", "binary/octet-stream")
 
 def rewrite(current, provider, proxy, ctype):