0.3.1 - Fixed #1

NyaaPy/nyaa.py

@@ -1,6 +1,7 @@
 import requests
 import xmltodict
-
+import json
+import collections
 
 class Nyaa():
     def search(keyword):
@@ -9,10 +10,16 @@ class Nyaa():
         request  = requests.get(nyaa_baseurl + keyword)
         response = xmltodict.parse(request.text)
 
+        results = []
+
         try:
-            results = response['rss']['channel']['item']
+            if type(response['rss']['channel']['item']) is collections.OrderedDict:
+                results.append(response['rss']['channel']['item'])
+            else:
+                results = response['rss']['channel']['item']
+
         except KeyError as ex:
-            results = {}
+            results = []
 
         return results
 
@@ -20,12 +27,18 @@ class NyaaPantsu():
     def search(keyword):
         nyaapantsu_baseurl = "https://nyaa.pantsu.cat/feed?c=_&s=0&max=99999&userID=0&q="
 
-        request  = requests.get(nyaapantsu_baseurl + keyword)
+        request  = requests.get(nyaa_baseurl + keyword)
         response = xmltodict.parse(request.text)
 
+        results = []
+
         try:
-            results = response['rss']['channel']['item']
+            if type(response['rss']['channel']['item']) is collections.OrderedDict:
+                results.append(response['rss']['channel']['item'])
+            else:
+                results = response['rss']['channel']['item']
+
         except KeyError as ex:
-            results = {}
+            results = []
 
         return results