yul/bookwyrm-mastodon
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Display results of import.

Browse Source
This commit is contained in:
Adam Kelly
2020-03-24 16:58:11 +00:00
parent 8fb082e2fc
commit 2188371f44
2 changed files with 59 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
fedireads/view_actions.py
View File

@ -297,20 +297,49 @@ def unquote_string(text):
else:
return text
def construct_search_term(title, author):
# Strip brackets (usually series title from search term)
title = re.sub(r'\s*\([^)]*\)\s*', '', title)
# Open library doesn't like including author initials in search term.
author = re.sub(r'(\w\.)+\s*', '', author)
return ' '.join([title, author])
import itertools
from io import TextIOWrapper
from requests import HTTPError
@login_required
def import_data(request):
form = forms.ImportForm(request.POST, request.FILES)
if form.is_valid():
reader = csv.DictReader(TextIOWrapper(request.FILES['csv_file'], encoding=request.encoding))
for line in itertools.islice(reader, 10):
results = []
failures = []
for line in itertools.islice(reader, 20):
isbn = unquote_string(line['ISBN13'])
print(line['Title'], isbn, line['Exclusive Shelf'])
search_results = books_manager.search(isbn)
if search_results:
book = books_manager.get_or_create_book(search_results[0].key)
return HttpResponse('thanks')
print(book)
results.append(book)
else:
try:
search_term = construct_search_term(line['Title'], line['Author'])
print("Search term: ", search_term)
search_results = books_manager.search(search_term)
if search_results:
book = books_manager.get_or_create_book(search_results[0].key)
print(book)
results.append(book)
else:
failures.append(line)
except HTTPError:
failures.append(line) #
return TemplateResponse(request, 'import_results.html', {
'results': results,
'failures': failures
})
else:
return HttpResponseBadRequest()