From e614cda733f76ab80ed008d51a14b4520a38daf3 Mon Sep 17 00:00:00 2001 From: wwqgtxx Date: Tue, 25 Jul 2017 21:49:23 +0800 Subject: [PATCH] avoid AttributeError in json_output.py --- src/you_get/json_output.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/you_get/json_output.py b/src/you_get/json_output.py index 2c9950ae..d447ea2f 100644 --- a/src/you_get/json_output.py +++ b/src/you_get/json_output.py @@ -17,9 +17,9 @@ def output(video_extractor, pretty_print=True): except AttributeError: pass extra = {} - if ve.referer is not None: + if getattr(ve, 'referer', None) is not None: extra["referer"] = ve.referer - if ve.ua is not None: + if getattr(ve, 'ua', None) is not None: extra["ua"] = ve.ua if extra: out["extra"] = extra