From 318b74827decbef368da5727f3536ba4c8139689 Mon Sep 17 00:00:00 2001
From: Mort Yao <soi@mort.ninja>
Date: Wed, 16 Mar 2016 20:36:02 +0100
Subject: [PATCH] [imgur] fix for direct images

---
 src/you_get/extractor.py        |  2 +-
 src/you_get/extractors/imgur.py | 14 +++++++++-----
 2 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/you_get/extractor.py b/src/you_get/extractor.py
index 1b547905..3cc78289 100644
--- a/src/you_get/extractor.py
+++ b/src/you_get/extractor.py
@@ -33,7 +33,7 @@ class VideoExtractor():
 
     def download_by_url(self, url, **kwargs):
         self.url = url
-        self.vid= None
+        self.vid = None
 
         if 'extractor_proxy' in kwargs and kwargs['extractor_proxy']:
             set_proxy(parse_host(kwargs['extractor_proxy']))
diff --git a/src/you_get/extractors/imgur.py b/src/you_get/extractors/imgur.py
index 80d68ce8..20848a87 100644
--- a/src/you_get/extractors/imgur.py
+++ b/src/you_get/extractors/imgur.py
@@ -37,11 +37,15 @@ class Imgur(VideoExtractor):
 
         elif re.search(r'i\.imgur\.com/', self.url):
             # direct image
-            universal_download(self.url,
-                               output_dir=kwargs['output_dir'],
-                               merge=kwargs['merge'],
-                               info_only=kwargs['info_only'])
-            exit(0) # FIXME!
+            _, container, size = url_info(self.url)
+            self.streams = {
+                'original': {
+                    'src': [self.url],
+                    'size': size,
+                    'container': container
+                }
+            }
+            self.title = r1(r'i\.imgur\.com/([^./]*)', self.url)
 
         else:
             # gallery image