added support for gif files when renaming downloaded image (#38) closes

#37 previously gif files would be downloaded with the default .jpg extension. now the mime type is detected for gifs and the files are renamed appropriately.
tasdikrahman · May 27, 2017 · 6f79005 · 6f79005
1 parent a4e7f23
commit 6f79005
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/xkcd_dl/cli.py b/xkcd_dl/cli.py
@@ -219,6 +219,10 @@ def download_one(xkcd_dict, xkcd_num):
                         os.rename(file_name, "{description}.jpeg".format(
                             description=new_description)
                         )
+                    elif 'gif' in magic_response:
+                        os.rename(file_name, "{description}.gif".format(
+                            description=new_description)
+                        )
 
     else: 
         print("{} does not exist! Please try with a different option".format(xkcd_number))