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

support single ebook harvest

pull/94/head
eric 2020-07-29 13:34:11 -04:00
parent 73b863450e
commit 6566afd92f
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
core/management/commands/harvest_online_ebooks.py
View File

@ -9,11 +9,15 @@ class Command(BaseCommand):
def add_arguments(self, parser): def add_arguments(self, parser):
parser.add_argument('limit', nargs='?', type=int, default=0, help="max to harvest") parser.add_argument('limit', nargs='?', type=int, default=0, help="max to harvest")
parser.add_argument('--ebook', nargs='?', type=int, default=0, help="ebook to harvest")
def handle(self, limit=0, **options): def handle(self, limit=0, **options):
limit = int(limit) if limit else 0 limit = int(limit) if limit else 0
rl = RateLimiter() rl = RateLimiter()
onlines = Ebook.objects.filter(format='online') if options.get('ebook'):
onlines = Ebook.objects.filter(id=options.get('ebook'))
else:
onlines = Ebook.objects.filter(format='online')
done = 0 done = 0
for online in onlines: for online in onlines:
new_ebf, new = dl_online(online, limiter=rl.delay) new_ebf, new = dl_online(online, limiter=rl.delay)