Playing with adding z-Libary binding (download gets blocked by server)

zSearch.py

@@ -0,0 +1,155 @@
+import requests
+from bs4 import BeautifulSoup
+
+class Search:
+
+    def __init__(self):
+        self.searchType = "general"
+        self.searchOptions = {"e": "", "yearFrom": "", \
+                            "yearTo": "", "language": "", \
+                            "extension": "", "page":"1", \
+                            "order":""}
+        self.searchResults = "books"
+        self.input = ""
+
+    def nextPage(self):
+        '''Goes to next page'''
+        self.searchOptions["page"] = str(int(self.searchOptions["page"])+1)
+
+    def previousPage(self):
+        '''Goes to previous page'''
+        if self.searchOptions["page"] != "1":
+            self.searchOptions["page"] = str(int(self.searchOptions["page"])-1)
+        else:
+            return 1
+
+    def updateSearchOptions(self):
+        '''Depending on searchType and searchResults, updates appropriately
+        searchOptions
+        '''
+        if self.searchType == "general":
+            self.searchOptions.pop("matchPhrase", None)
+        else: 
+            self.searchOptions["matchPhrase"] = ""
+    
+        if self.searchResults == "books":
+            self.searchOptions["language"] = ""
+            self.searchOptions["extension"] = ""
+        else:
+            self.searchOptions.pop("language", None)
+            self.searchOptions.pop("extension", None)
+            
+    def executeSearch(self):
+        '''Executes get request and returns response'''
+        if self.searchResults == "books":
+            url = "http://b-ok.cc"
+        else:
+            url = "http://booksc.xyz"
+
+        r = requests.get(url+"/s/"+self.input, params=self.searchOptions)
+        return r
+
+
+    def getResults(self):
+        '''Fetches search results. Returns a list of books
+        '''
+        r = self.executeSearch()
+        soup = BeautifulSoup(r.text, "html.parser")
+        match = soup.find_all(class_="resItemBox")
+        counter = 1
+        results=[]
+        for e in match:
+            title = e.find(itemprop="name")
+            author = e.find(class_="authors")
+            year = e.find(class_="bookProperty property_year")
+            language = e.find(class_="bookProperty property_language")
+            fileInfo = e.find(class_="bookProperty property__file")
+            link = e.find("a", href=True)
+            link = link["href"]
+
+            if self.searchResults == "books":
+                fullLink = "https://b-ok.cc" + link
+            else:
+                fullLink = "booksc.xyz" + link
+            
+            title = isNone(title)
+            author = isNone(author)
+            year = isNone(year)
+            language = isNone(language)
+            fileInfo = isNone(fileInfo)
+            
+            book = Book(title,author,year,language,fileInfo,fullLink)
+            results += [book]
+        
+        return results
+
+    def reset(self):
+        '''Clears the search and resets to default options'''
+        self.searchType = "general"
+        self.searchOptions = {"e": "", "yearFrom": "", \
+                            "yearTo": "", "language": "", \
+                            "extension": "", "page":"1", \
+                            "order":""}
+        self.searchResults = "books"
+        self.input = ""
+        
+
+def isNone(e):
+    if e != None:
+        return "".join(e.text.splitlines())
+    else:
+        return ""
+
+class Book:
+    
+    def __init__(self,title,author,year,language,fileInfo,link):
+        self.title = title
+        self.author = author
+        self.year = year
+        self.language = language
+        self.fType, self.size = fileInfo.split(', ')
+        self.link = link
+        self.dlLink = None
+        self.page = None
+
+    def __repr__(self):
+        return '<zBook: '+str(self)+'>'
+
+    def __str__(self):
+        return " / ".join([self.title,self.author,self.year,\
+                self.language,self.fType,self.size])
+
+    def getDetails(self):
+        '''Returns more specific info about the book. The info is retrieved by the
+        link attribute
+        '''
+        if self.page == None:
+            self.page = requests.get(self.link).text
+        
+        soup = BeautifulSoup(self.page, "html.parser")
+        # for some reason, bookProperty also shows properties from other books
+        # the line below prevents this
+        soup = soup.find(class_="row cardBooks")
+        match = soup.find_all(class_="bookProperty")
+        
+        results = ""
+        for e in match:
+            results += "".join(e.text.splitlines())
+            results += "\n"
+
+        # this makes writing the category easier for some books
+        results = results.replace("\\\\", " \\ ")
+        return results
+    
+    def getDownloadURL(self):
+        if self.dlLink == None:
+            if self.page == None:
+                self.page = requests.get(self.link).text
+            soup = BeautifulSoup(self.page, "html.parser")
+            self.dlLink = soup.find('a', 'btn btn-primary dlButton addDownloadedBook')['href']
+        return 'https://b-ok.cc' + self.dlLink + '?dsource=recommend'
+
+    def saveBook(self, path):
+        r = requests.get(self.getDownloadURL(), allow_redirects=True)
+        with open(path, 'wb') as f:
+            f.write(r.content)