agatha/pastebin-client
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Create util module #1

Merged
agatha merged 1 commits from dev-v2 into master 2023-09-12 21:15:48 +00:00
Conversation 0 Commits 1 Files Changed 3 +14 -15
3 changed files with 14 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/main.py
View File

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

14
src/pastebin/client.py
View File

@ -182,17 +182,3 @@ class PastebinClient(PastebinAPI):
self.pastes[i] for i, paste in enumerate(self.pastes) self.pastes[i] for i, paste in enumerate(self.pastes)
if paste.lang.lower() == lang.lower() 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
]

12
src/pastebin/util.py Normal file
View File

@ -0,0 +1,12 @@
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
]