From 31ca595e875dd4e2cb07b2d3610b0d4b4f590abe Mon Sep 17 00:00:00 2001 From: Thibaut Horel Date: Thu, 27 Feb 2014 11:29:28 -0500 Subject: PEP8 --- wikisource.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) (limited to 'wikisource.py') diff --git a/wikisource.py b/wikisource.py index 070c84a..1459468 100644 --- a/wikisource.py +++ b/wikisource.py @@ -1,14 +1,14 @@ # -*- coding: utf-8 -*- import requests -import lxml import sys from bs4 import BeautifulSoup from itertools import takewhile, count URL = "http://fr.wikisource.org/w/index.php" + def get_page(title, page): - params = { "action": "render", "title": "Page:" + title + "/" + str(page) } + params = {"action": "render", "title": "Page:" + title + "/" + str(page)} r = requests.get(URL, params=params) if r.status_code == requests.codes.ok: soup = BeautifulSoup(r.text, "lxml") @@ -16,9 +16,10 @@ def get_page(title, page): else: return None + def get_pages(title, begin=1, end=None): if end: - return (get_page(title, i) for i in xrange(begin, end+1)) + return (get_page(title, i) for i in xrange(begin, end + 1)) else: return takewhile(lambda x: x is not None, (get_page(title, i) for i in count(begin))) -- cgit v1.2.3-70-g09d2