src/main.py

@@ -1,6 +1,5 @@
 import time
-from pastebin.client import PastebinClient
-from pastebin.util import search_by_language
+from pastebin.client import PastebinClient, search_by_language
 
 
 def main():

src/pastebin/client.py

@@ -182,3 +182,17 @@ class PastebinClient(PastebinAPI):
             self.pastes[i] for i, paste in enumerate(self.pastes)
             if paste.lang.lower() == lang.lower()
         ]
+
+
+def search_by_language(pastes, lang):
+    return [
+        pastes[i] for i, paste in enumerate(pastes)
+        if paste.lang.lower() == lang.lower()
+    ]
+
+
+def search_by_category(pastes, category):
+    return [
+        pastes[i] for i, paste in enumerate(pastes)
+        if paste.category == category
+    ]

src/pastebin/util.py

@@ -1,12 +0,0 @@
-def search_by_language(pastes, lang):
-    return [
-        pastes[i] for i, paste in enumerate(pastes)
-        if paste.lang.lower() == lang.lower()
-    ]
-
-
-def search_by_category(pastes, category):
-    return [
-        pastes[i] for i, paste in enumerate(pastes)
-        if paste.category == category
-    ]