Функция не будет петля

Question

Я сделал фрагмент кода, который отлично работает с печатью, но не работает, когда я сделал его функцию и попытался вернуть его. Вот исходный код:

import requests 
from bs4 import BeautifulSoup 
import wikipedia 

source_code = requests.get('http://en.wikipedia.org/wiki/IBM') 
plain_text = source_code.text 
plain_text = plain_text[:plain_text.find('id="toc"')] 
soup = BeautifulSoup(plain_text) 

for div in soup.findAll('a'): 
    if div.parent.name == 'p': 
     href = div.get('href') 
     href = href.replace(',', '') 
     href = href.replace('-', ' ') 
     href = href.replace('(', '') 
     href = href.replace(')', '') 
     href = href.replace('_', ' ') 

     print (href[6:]) 
     href = href.replace(' ', '_') 
     href = href.replace('^', '') 
     try: 
      print(wikipedia.summary(href[6:])) 
     except wikipedia.exceptions.DisambiguationError as e: 
      print (e.options) 

, который форматирует текст и дает мне название и резюме страницы википедии и все резюме ссылки в исходном резюме, что именно то, что я хочу. К сожалению, это должно быть частью более крупной программы, и поэтому я сделал функцию (возможно, я должен делать это по-другому?) это выглядит следующим образом:

import requests 
from bs4 import BeautifulSoup 
import wikipedia 

source_code = requests.get('http://en.wikipedia.org/wiki/IBM') 
plain_text = source_code.text 
plain_text = plain_text[:plain_text.find('id="toc"')] 
soup = BeautifulSoup(plain_text) 

def ELS(): 
    for div in soup.findAll('a'): 
     if div.parent.name == 'p': 
      href = div.get('href') 
      href = href.replace(',', '') 
      href = href.replace('-', ' ') 
      href = href.replace('(', '') 
      href = href.replace(')', '') 
      href = href.replace('_', ' ') 

      return href[6:] 
      href = href.replace(' ', '_') 
      href = href.replace('^', '') 
      try: 
       return wikipedia.summary(href[6:]) 
      except wikipedia.exceptions.DisambiguationError as e: 
       return e.options 

print (ELS()) 

, но по какой-то причине, он не цикл и просто печатает первый заголовок, а затем ломается, может быть, это легкая проблема, и только что-то я пропустил

Answer 1

Вы возвращаетесь из своей функции, поэтому ломаете петлю. Вам нужно добавить результаты поиска в список или в dict и вернуть его после вашего цикла.

Answer 2

return сразу же выходит из функции.

Сбор информации в список и вернуться, что:

def ELS(): 
    results = [] 
    for div in soup.findAll('a'): 
     if div.parent.name == 'p': 
      href = div.get('href') 
      href = href.replace(',', '') 
      href = href.replace('-', ' ') 
      href = href.replace('(', '') 
      href = href.replace(')', '') 
      href = href.replace('_', ' ') 

      href = href.replace(' ', '_') 
      href = href.replace('^', '') 
      try: 
       results.append((href[6:], wikipedia.summary(href[6:]))) 
      except wikipedia.exceptions.DisambiguationError as e: 
       results.append((href[6:], e.options)) 
    return results 

Вы можете затем цикл по результатам; каждая запись представляет собой кортеж с обработанным значением href и значением wikipedia.summary() или исключением e.options. Это позволяет вам повторно использовать эту информацию в другом коде.

Answer 3

Вы просто заменяете печать с возвратом, и ваше поведение функции теперь имеет проблему, так как функция завершает свое выполнение при вызове команды return.

Попробуйте что-то вроде этого:

def ELS(): 
    output = [] 
    for div in soup.findAll('a'): 
     if div.parent.name == 'p': 
      href = div.get('href') 
      href = href.replace(',', '') 
      href = href.replace('-', ' ') 
      href = href.replace('(', '') 
      href = href.replace(')', '') 
      href = href.replace('_', ' ') 

      output.append(href[6:]) 
      href = href.replace(' ', '_') 
      href = href.replace('^', '') 
      try: 
       output.append(wikipedia.summary(href[6:])) 
      except wikipedia.exceptions.DisambiguationError as e: 
       output.append(e.options) 

    return "\n".join(output)