EbookFoundation
/
regluit
mirror of https://github.com/EbookFoundation/regluit.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set user on ebooks loaded from webpage

pull/43/head
eric 2017-08-08 12:38:54 -04:00
parent 0ebbb21d47
commit 8de43cfda8
2 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
core/bookloader.py
View File

@ -897,7 +897,7 @@ class BasePandataLoader(object):
edition.save()
return edition
def load_ebooks(self, metadata, edition, test_mode=False):
def load_ebooks(self, metadata, edition, test_mode=False, user=None):
default_edition = edition
for key in ['epub', 'pdf', 'mobi']:
url = metadata.metadata.get('download_url_{}'.format(key), None)
@ -917,7 +917,7 @@ class BasePandataLoader(object):
)
ebf.file.save(contentfile_name, contentfile)
ebf.file.close()
(ebook, created) = models.Ebook.objects.get_or_create(
ebook = models.Ebook.objects.create(
url=ebf.file.url,
provider='Unglue.it',
rights=license,
@ -925,6 +925,7 @@ class BasePandataLoader(object):
edition=edition,
filesize=contentfile.size,
active=False,
user=user,
)
ebf.ebook = ebook
ebf.save()
@ -1018,7 +1019,7 @@ def ebooks_in_github_release(repo_owner, repo_name, tag, token=None):
for asset in release.iter_assets()
if EBOOK_FORMATS.get(asset.content_type) is not None]
def add_by_webpage(url, work=None):
def add_by_webpage(url, work=None, user=None):
edition = None
scraper = BaseScraper(url)
loader = BasePandataLoader(url)
@ -1027,7 +1028,7 @@ def add_by_webpage(url, work=None):
for metadata in pandata.get_edition_list():
edition = loader.load_from_pandata(metadata, work)
work = edition.work
loader.load_ebooks(pandata, edition)
loader.load_ebooks(pandata, edition, user=user)
return edition if edition else None

6
frontend/views/bibedit.py
View File

@ -68,7 +68,7 @@ def get_edition(edition_id):
except models.Edition.DoesNotExist:
raise Http404 (duplicate-code)
def get_edition_for_id(id_type, id_value):
def get_edition_for_id(id_type, id_value, user=None):
''' the identifier is assumed to not be in database '''
identifiers = {id_type: id_value}
if id_type == 'http':
@ -107,7 +107,7 @@ def get_edition_for_id(id_type, id_value):
pass
if identifiers.has_key('http'):
edition = add_by_webpage(identifiers['http'])
edition = add_by_webpage(identifiers['http'], user=user)
return edition
@ -149,7 +149,7 @@ def new_edition(request, by=None):
work = ident.work
edition = ident.edition if ident.edition else work.preferred_edition
else:
edition = get_edition_for_id(id_type, id_value)
edition = get_edition_for_id(id_type, id_value, user=request.user)
return HttpResponseRedirect(
reverse('new_edition', kwargs={