def search_book(books=None, library=None):
"""
books is a list of new books to add, or None for backlog search
library is "eBook" or "AudioBook" or None to search all book types
"""
# noinspection PyBroadException
try:
threadname = threading.currentThread().name
if "Thread-" in threadname:
if books is None:
threading.currentThread().name = "SEARCHALLBOOKS"
else:
threading.currentThread().name = "SEARCHBOOKS"
myDB = database.DBConnection()
searchlist = []
searchbooks = []
if books is None:
# We are performing a backlog search
cmd = 'SELECT BookID, AuthorName, Bookname, BookSub, BookAdded, books.Status, AudioStatus '
cmd += 'from books,authors WHERE (books.Status="Wanted" OR AudioStatus="Wanted") '
cmd += 'and books.AuthorID = authors.AuthorID order by BookAdded desc'
results = myDB.select(cmd)
for terms in results:
searchbooks.append(terms)
else:
# The user has added a new book
for book in books:
if not book['bookid'] in ['booklang', 'library', 'ignored']:
cmd = 'SELECT BookID, AuthorName, BookName, BookSub, books.Status, AudioStatus '
cmd += 'from books,authors WHERE BookID=? AND books.AuthorID = authors.AuthorID'
results = myDB.select(cmd, (book['bookid'], ))
if results:
for terms in results:
searchbooks.append(terms)
else:
logger.debug(
"SearchBooks - BookID %s is not in the database" %
book['bookid'])
if len(searchbooks) == 0:
logger.debug("SearchBooks - No books to search for")
return
nprov = lazylibrarian.USE_NZB() + lazylibrarian.USE_TOR(
) + lazylibrarian.USE_RSS() + lazylibrarian.USE_DIRECT()
if nprov == 0:
logger.debug("SearchBooks - No providers to search")
return
modelist = []
if lazylibrarian.USE_NZB():
modelist.append('nzb')
if lazylibrarian.USE_TOR():
modelist.append('tor')
if lazylibrarian.USE_DIRECT():
modelist.append('direct')
if lazylibrarian.USE_RSS():
modelist.append('rss')
logger.info('Searching %s provider%s %s for %i book%s' %
(nprov, plural(nprov), str(modelist), len(searchbooks),
plural(len(searchbooks))))
for searchbook in searchbooks:
# searchterm is only used for display purposes
searchterm = searchbook['AuthorName'] + ' ' + searchbook['BookName']
if searchbook['BookSub']:
searchterm = searchterm + ': ' + searchbook['BookSub']
if library is None or library == 'eBook':
if searchbook[
'Status'] == "Wanted": # not just audiobook wanted
cmd = 'SELECT BookID from wanted WHERE BookID=? and AuxInfo="eBook" and Status="Snatched"'
snatched = myDB.match(cmd, (searchbook["BookID"], ))
if snatched:
logger.warn(
'eBook %s %s already marked snatched in wanted table'
%
(searchbook['AuthorName'], searchbook['BookName']))
else:
searchlist.append({
"bookid":
searchbook['BookID'],
"bookName":
searchbook['BookName'],
"bookSub":
searchbook['BookSub'],
"authorName":
searchbook['AuthorName'],
"library":
"eBook",
"searchterm":
searchterm
})
if library is None or library == 'AudioBook':
if searchbook[
'AudioStatus'] == "Wanted": # in case we just wanted eBook
cmd = 'SELECT BookID from wanted WHERE BookID=? and AuxInfo="AudioBook" and Status="Snatched"'
snatched = myDB.match(cmd, (searchbook["BookID"], ))
if snatched:
logger.warn(
'AudioBook %s %s already marked snatched in wanted table'
%
(searchbook['AuthorName'], searchbook['BookName']))
else:
searchlist.append({
"bookid":
searchbook['BookID'],
"bookName":
searchbook['BookName'],
"bookSub":
searchbook['BookSub'],
"authorName":
searchbook['AuthorName'],
"library":
"AudioBook",
"searchterm":
searchterm
})
# only get rss results once per run, as they are not search specific
rss_resultlist = None
if 'rss' in modelist:
rss_resultlist, nprov = IterateOverRSSSites()
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_RSS_MSG, 0) + 1200 < timenow:
logger.warn(
'No rss providers are available. Check config and blocklist'
)
lazylibrarian.NO_RSS_MSG = timenow
modelist.remove('rss')
book_count = 0
for book in searchlist:
matches = []
for mode in modelist:
# first attempt, try author/title in category "book"
if book['library'] == 'AudioBook':
searchtype = 'audio'
else:
searchtype = 'book'
resultlist = None
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG,
0) + 1200 < timenow:
logger.warn(
'No nzb providers are available. Check config and blocklist'
)
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
elif mode == 'tor' and 'tor' in modelist:
resultlist, nprov = IterateOverTorrentSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_TOR_MSG,
0) + 1200 < timenow:
logger.warn(
'No tor providers are available. Check config and blocklist'
)
lazylibrarian.NO_TOR_MSG = timenow
modelist.remove('tor')
elif mode == 'direct' and 'direct' in modelist:
resultlist, nprov = IterateOverDirectSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_DIRECT_MSG,
0) + 1200 < timenow:
logger.warn(
'No direct providers are available. Check config and blocklist'
)
lazylibrarian.NO_DIRECT_MSG = timenow
modelist.remove('direct')
elif mode == 'rss' and 'rss' in modelist:
if rss_resultlist:
resultlist = rss_resultlist
else:
logger.debug("No active rss providers found")
modelist.remove('rss')
if resultlist:
match = findBestResult(resultlist, book, searchtype, mode)
else:
match = None
# if you can't find the book, try author/title without any "(extended details, series etc)"
if not goodEnough(match) and '(' in book['bookName']:
searchtype = 'short' + searchtype
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG,
0) + 1200 < timenow:
logger.warn(
'No nzb providers are available. Check config and blocklist'
)
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
elif mode == 'tor' and 'tor' in modelist:
resultlist, nprov = IterateOverTorrentSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_TOR_MSG,
0) + 1200 < timenow:
logger.warn(
'No tor providers are available. Check config and blocklist'
)
lazylibrarian.NO_TOR_MSG = timenow
modelist.remove('tor')
elif mode == 'direct' and 'direct' in modelist:
resultlist, nprov = IterateOverDirectSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_DIRECT_MSG,
0) + 1200 < timenow:
logger.warn(
'No direct providers are available. Check config and blocklist'
)
lazylibrarian.NO_DIRECT_MSG = timenow
modelist.remove('direct')
elif mode == 'rss' and 'rss' in modelist:
resultlist = rss_resultlist
if resultlist:
match = findBestResult(resultlist, book, searchtype,
mode)
else:
match = None
# if you can't find the book under "books", you might find under general search
# general search is the same as booksearch for torrents and rss, no need to check again
if not goodEnough(match):
searchtype = 'general'
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG,
0) + 1200 < timenow:
logger.warn(
'No nzb providers are available. Check config and blocklist'
)
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
if resultlist:
match = findBestResult(resultlist, book,
searchtype, mode)
else:
match = None
# if still not found, try general search again without any "(extended details, series etc)"
if not goodEnough(match) and '(' in book['searchterm']:
searchtype = 'shortgeneral'
if mode == 'nzb' and 'nzb' in modelist:
resultlist, _ = IterateOverNewzNabSites(
book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG,
0) + 1200 < timenow:
logger.warn(
'No nzb providers are available. Check config and blocklist'
)
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
if resultlist:
match = findBestResult(resultlist, book,
searchtype, mode)
else:
match = None
if not goodEnough(match):
logger.info(
"%s Searches for %s %s returned no results." %
(mode.upper(), book['library'], book['searchterm']))
else:
logger.info("Found %s result: %s %s%%, %s priority %s" %
(mode.upper(), searchtype, match[0],
match[2]['NZBprov'], match[4]))
matches.append(match)
if matches:
highest = max(matches, key=lambda s:
(s[0], s[4])) # sort on percentage and priority
logger.info("Requesting %s download: %s%% %s: %s" %
(book['library'], highest[0],
highest[2]['NZBprov'], highest[1]))
if downloadResult(highest, book) > True:
book_count += 1 # we found it
logger.info("Search for Wanted items complete, found %s book%s" %
(book_count, plural(book_count)))
except Exception:
logger.error('Unhandled exception in search_book: %s' %
traceback.format_exc())
finally:
threading.currentThread().name = "WEBSERVER"
def search_book(books=None, library=None):
"""
books is a list of new books to add, or None for backlog search
library is "eBook" or "AudioBook" or None to search all book types
"""
# noinspection PyBroadException
try:
threadname = threading.currentThread().name
if 'SEARCHALL' in threadname or 'API-SEARCH' in threadname:
force = True
else:
force = False
if "Thread-" in threadname:
if not books:
threading.currentThread().name = "SEARCHALLBOOKS"
else:
threading.currentThread().name = "SEARCHBOOKS"
myDB = database.DBConnection()
searchlist = []
searchbooks = []
if not books:
# We are performing a backlog search
cmd = 'SELECT BookID, AuthorName, Bookname, BookSub, BookAdded, books.Status, AudioStatus '
cmd += 'from books,authors WHERE (books.Status="Wanted" OR AudioStatus="Wanted") '
cmd += 'and books.AuthorID = authors.AuthorID order by BookAdded desc'
results = myDB.select(cmd)
for terms in results:
searchbooks.append(terms)
else:
# The user has added a new book
for book in books:
if not book['bookid'] in ['booklang', 'library', 'ignored']:
cmd = 'SELECT BookID, AuthorName, BookName, BookSub, books.Status, AudioStatus '
cmd += 'from books,authors WHERE BookID=? AND books.AuthorID = authors.AuthorID'
results = myDB.select(cmd, (book['bookid'],))
if results:
for terms in results:
searchbooks.append(terms)
else:
logger.debug("SearchBooks - BookID %s is not in the database" % book['bookid'])
if len(searchbooks) == 0:
logger.debug("SearchBooks - No books to search for")
return
nprov = lazylibrarian.USE_NZB() + lazylibrarian.USE_TOR() + lazylibrarian.USE_RSS() + lazylibrarian.USE_DIRECT()
if nprov == 0:
logger.debug("SearchBooks - No providers to search")
return
modelist = []
if lazylibrarian.USE_NZB():
modelist.append('nzb')
if lazylibrarian.USE_TOR():
modelist.append('tor')
if lazylibrarian.USE_DIRECT():
modelist.append('direct')
if lazylibrarian.USE_RSS():
modelist.append('rss')
logger.info('Searching %s provider%s %s for %i book%s' %
(nprov, plural(nprov), str(modelist), len(searchbooks), plural(len(searchbooks))))
for searchbook in searchbooks:
# searchterm is only used for display purposes
searchterm = ''
if searchbook['AuthorName']:
searchterm = searchbook['AuthorName']
else:
logger.warn("No AuthorName for %s" % searchbook['BookID'])
if searchbook['BookName']:
if len(searchterm):
searchterm += ' '
searchterm += searchbook['BookName']
else:
logger.warn("No BookName for %s" % searchbook['BookID'])
if searchbook['BookSub']:
if len(searchterm):
searchterm += ': '
searchterm += searchbook['BookSub']
if library is None or library == 'eBook':
if searchbook['Status'] == "Wanted": # not just audiobook wanted
cmd = 'SELECT BookID from wanted WHERE BookID=? and AuxInfo="eBook" and Status="Snatched"'
snatched = myDB.match(cmd, (searchbook["BookID"],))
if snatched:
logger.warn('eBook %s %s already marked snatched in wanted table' %
(searchbook['AuthorName'], searchbook['BookName']))
else:
searchlist.append(
{"bookid": searchbook['BookID'],
"bookName": searchbook['BookName'],
"bookSub": searchbook['BookSub'],
"authorName": searchbook['AuthorName'],
"library": "eBook",
"searchterm": searchterm})
if library is None or library == 'AudioBook':
if searchbook['AudioStatus'] == "Wanted": # in case we just wanted eBook
cmd = 'SELECT BookID from wanted WHERE BookID=? and AuxInfo="AudioBook" and Status="Snatched"'
snatched = myDB.match(cmd, (searchbook["BookID"],))
if snatched:
logger.warn('AudioBook %s %s already marked snatched in wanted table' %
(searchbook['AuthorName'], searchbook['BookName']))
else:
searchlist.append(
{"bookid": searchbook['BookID'],
"bookName": searchbook['BookName'],
"bookSub": searchbook['BookSub'],
"authorName": searchbook['AuthorName'],
"library": "AudioBook",
"searchterm": searchterm})
# only get rss results once per run, as they are not search specific
rss_resultlist = None
if 'rss' in modelist:
rss_resultlist, nprov, dltypes = IterateOverRSSSites()
if not nprov or (library == 'Audiobook' and 'A' not in dltypes) or \
(library == 'eBook' and 'E' not in dltypes) or \
(library is None and dltypes == 'M'):
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_RSS_MSG, 0) + 1200 < timenow:
logger.warn('No rss providers are available. Check config and blocklist')
lazylibrarian.NO_RSS_MSG = timenow
modelist.remove('rss')
book_count = 0
for book in searchlist:
do_search = True
if lazylibrarian.CONFIG['DELAYSEARCH'] and not force:
res = myDB.match('SELECT * FROM failedsearch WHERE BookID=? AND Library=?',
(book['bookid'], book['library']))
if not res:
logger.debug("SearchDelay: %s %s has not failed before" % (book['library'], book['bookid']))
else:
skipped = check_int(res['Count'], 0)
interval = check_int(res['Interval'], 0)
if skipped < interval:
logger.debug("SearchDelay: %s %s not due (%d/%d)" %
(book['library'], book['bookid'], skipped, interval))
myDB.action("UPDATE failedsearch SET Count=? WHERE BookID=? AND Library=?",
(skipped + 1, book['bookid'], book['library']))
do_search = False
else:
logger.debug("SearchDelay: %s %s due this time (%d/%d)" %
(book['library'], book['bookid'], skipped, interval))
matches = []
if do_search:
for mode in modelist:
# first attempt, try author/title in category "book"
if book['library'] == 'AudioBook':
searchtype = 'audio'
else:
searchtype = 'book'
resultlist = None
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG, 0) + 1200 < timenow:
logger.warn('No nzb providers are available. Check config and blocklist')
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
elif mode == 'tor' and 'tor' in modelist:
resultlist, nprov = IterateOverTorrentSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_TOR_MSG, 0) + 1200 < timenow:
logger.warn('No tor providers are available. Check config and blocklist')
lazylibrarian.NO_TOR_MSG = timenow
modelist.remove('tor')
elif mode == 'direct' and 'direct' in modelist:
resultlist, nprov = IterateOverDirectSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_DIRECT_MSG, 0) + 1200 < timenow:
logger.warn('No direct providers are available. Check config and blocklist')
lazylibrarian.NO_DIRECT_MSG = timenow
modelist.remove('direct')
elif mode == 'rss' and 'rss' in modelist:
if rss_resultlist:
resultlist = rss_resultlist
else:
logger.debug("No active rss providers found")
modelist.remove('rss')
if resultlist:
match = findBestResult(resultlist, book, searchtype, mode)
else:
match = None
# if you can't find the book, try author/title without any "(extended details, series etc)"
if not goodEnough(match) and '(' in book['bookName']:
searchtype = 'short' + searchtype
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG, 0) + 1200 < timenow:
logger.warn('No nzb providers are available. Check config and blocklist')
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
elif mode == 'tor' and 'tor' in modelist:
resultlist, nprov = IterateOverTorrentSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_TOR_MSG, 0) + 1200 < timenow:
logger.warn('No tor providers are available. Check config and blocklist')
lazylibrarian.NO_TOR_MSG = timenow
modelist.remove('tor')
elif mode == 'direct' and 'direct' in modelist:
resultlist, nprov = IterateOverDirectSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_DIRECT_MSG, 0) + 1200 < timenow:
logger.warn('No direct providers are available. Check config and blocklist')
lazylibrarian.NO_DIRECT_MSG = timenow
modelist.remove('direct')
elif mode == 'rss' and 'rss' in modelist:
resultlist = rss_resultlist
if resultlist:
match = findBestResult(resultlist, book, searchtype, mode)
else:
match = None
# if you can't find the book under "books", you might find under general search
# general search is the same as booksearch for torrents and rss, no need to check again
if not goodEnough(match):
searchtype = 'general'
if mode == 'nzb' and 'nzb' in modelist:
resultlist, nprov = IterateOverNewzNabSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG, 0) + 1200 < timenow:
logger.warn('No nzb providers are available. Check config and blocklist')
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
if resultlist:
match = findBestResult(resultlist, book, searchtype, mode)
else:
match = None
# if still not found, try general search again without any "(extended details, series etc)"
if not goodEnough(match) and '(' in book['searchterm']:
searchtype = 'shortgeneral'
if mode == 'nzb' and 'nzb' in modelist:
resultlist, _ = IterateOverNewzNabSites(book, searchtype)
if not nprov:
# don't nag. Show warning message no more than every 20 mins
timenow = int(time.time())
if check_int(lazylibrarian.NO_NZB_MSG, 0) + 1200 < timenow:
logger.warn('No nzb providers are available. Check config and blocklist')
lazylibrarian.NO_NZB_MSG = timenow
modelist.remove('nzb')
if resultlist:
match = findBestResult(resultlist, book, searchtype, mode)
else:
match = None
if not goodEnough(match):
logger.info("%s Searches for %s %s returned no results." %
(mode.upper(), book['library'], book['searchterm']))
else:
logger.info("Found %s result: %s %s%%, %s priority %s" %
(mode.upper(), searchtype, match[0], match[1]['NZBprov'], match[3]))
matches.append(match)
if matches:
highest = max(matches, key=lambda s: (s[0], s[3])) # sort on percentage and priority
logger.info("Requesting %s download: %s%% %s: %s" %
(book['library'], highest[0], highest[1]['NZBprov'], highest[1]['NZBtitle']))
if downloadResult(highest, book) > 1:
book_count += 1 # we found it
myDB.action("DELETE from failedsearch WHERE BookID=? AND Library=?", (book['bookid'], book['library']))
elif lazylibrarian.CONFIG['DELAYSEARCH'] and not force and do_search and len(modelist):
res = myDB.match('SELECT * FROM failedsearch WHERE BookID=? AND Library=?',
(book['bookid'], book['library']))
if res:
interval = check_int(res['Interval'], 0)
else:
interval = 0
myDB.upsert("failedsearch",
{'Count': 0, 'Interval': interval + 1, 'Time': time.time()},
{'BookID': book['bookid'], 'Library': book['library']})
logger.info("Search for Wanted items complete, found %s book%s" % (book_count, plural(book_count)))
except Exception:
logger.error('Unhandled exception in search_book: %s' % traceback.format_exc())
finally:
threading.currentThread().name = "WEBSERVER"
