Comment utiliser la chaîne comme entrée pour le lecteur csv sans le stocker dans le fichier

Question

J'essaie de faire défiler les lignes dans un fichier csv. Je reçois le fichier csv en tant que string à partir d'un site Web. Je sais comment créer csv.reader en utilisant with lorsque les données sont stockées dans un fichier. Qu'est-ce que je ne sais pas, comment obtenir des lignes en utilisant csv.reader sans stocker string dans un fichier. J'utilise Python 2.7.12.

J'ai essayé de créer StringIO objet comme celui-ci:

from StringIO import StringIO 

csv_data = "some_string\nfor_example" 
with StringIO(csv_data) as input_file: 
    csv_reader = reader(csv_data, delimiter=",", quotechar='"') 

Cependant, je reçois cette erreur:

Traceback (most recent call last): 
    File "scraper.py", line 228, in <module> 
    with StringIO(csv_data) as input_file: 
AttributeError: StringIO instance has no attribute '__exit__' 

Je comprends que StringIO classe n'a pas __exit__ méthode est appelé quand when finit de faire tout ce qu'il fait avec cet objet.

Ma réponse est comment faire ceci correctement? Je suppose que je peux modifier la classe StringIO en la sous-classant et en ajoutant la méthode __exit__, mais je soupçonne qu'il existe une solution plus facile.

Mise à jour:

Aussi, j'ai essayé différentes combinaisons qui sont venues à l'esprit:

with open(StringIO(csv_data)) as input_file: 

with csv_data as input_file: 

mais, bien sûr, aucun de ceux travaillé.

Answer 1

>>> import csv 
>>> csv_data = "some,string\nfor,example" 
>>> result = csv.reader(csv_data.splitlines()) 
>>> list(result) 
[['some', 'string'], ['for', 'example']] 

Answer 2

Si vous aimez les gestionnaires de contexte, vous pouvez utiliser tempfile à la place:

import tempfile 


with tempfile.NamedTemporaryFile(mode='w') as t: 
    t.write('csv_data') 
    t.seek(0) 
    csv_reader = reader(open(t.name), delimiter=",", quotechar='"') 

Comme un avantage de passer lignes de division de chaîne directement au lecteur csv, vous pouvez écrire dans le fichier de toute taille, puis lire en toute sécurité dans lecteur csv sans problèmes de mémoire.

Ce fichier sera fermé et supprimé automatiquement

Answer 3

Vous devez utiliser le module io au lieu du StringIO, parce que io.BytesIO pour chaîne d'octets ou io.StringIO pour les Unicode à la fois en charge l'interface du gestionnaire de contexte et peut être utilisé dans with instructions:

from io import BytesIO 
from csv import reader 
csv_data = "some_string\nfor_example" 
with BytesIO(csv_data) as input_file: 
    csv_reader = reader(input_file, delimiter=",", quotechar='"') 
    for row in csv_reader: 
     print row