diff --git a/gallery_dl/extractor/erome.py b/gallery_dl/extractor/erome.py index 2aed678f..6a0e069a 100644 --- a/gallery_dl/extractor/erome.py +++ b/gallery_dl/extractor/erome.py @@ -44,11 +44,15 @@ class EromeExtractor(Extractor): pos = page.index('
', '', pos) + data = { "album_id" : album_id, "title" : text.unescape(title), "user" : text.unquote(user), "_http_headers": {"Referer": url}, + "count" : text.parse_int(count), } yield Message.Directory, data diff --git a/test/results/erome.py b/test/results/erome.py index f3269c04..40f34aad 100644 --- a/test/results/erome.py +++ b/test/results/erome.py @@ -16,6 +16,7 @@ __tests__ = ( "#count" : 1, "album_id": "NQgdlWvk", + "count" : 1, "num" : 1, "title" : "porn", "user" : "yYgWBZw8o8qsMzM", @@ -29,6 +30,7 @@ __tests__ = ( "#count" : 6, "album_id": "TdbZ4ogi", + "count" : 6, "num" : int, "title" : "82e78cfbb461ad87198f927fcb1fda9a1efac9ff.", "user" : "yYgWBZw8o8qsMzM",