Преобразование загруженного файла CSV с помощью Flask - PullRequest
Новая
       10

Преобразование загруженного файла CSV с помощью Flask

0 голосов
/ 15 ноября 2018

У меня есть простое приложение Flask, размещенное на Python Anywhere. Приложение предоставляет интерфейс, в котором пользователь может загрузить файл CSV, и приложение вставляет новый столбец в загруженный файл CSV, перезаписывая первый столбец (row[0]) и помещая исходный row[0] в конец файла и загружает преобразованный файл для пользователя. Итак, вход CSV выглядит следующим образом (первая строка является строкой заголовка, которая должна быть неизменена, за исключением добавления имени 4-го столбца): 

Pet1,Pet2,Pet3
Cats,Dogs,Mice

И вывод должен выглядеть так: 

Pet1,Pet2,Pet3,Pet4
Birds,Dogs,Mice,Cats

Вот мой код: 

from flask import Flask, make_response, request
import operator
import io
import csv

app = Flask(__name__)

def transform(text_file_contents):
  reader = csv.reader(text_file_contents)
  all = []
  row = next(reader)
  row.append('Pet4')
  all.append(row)
  for row in reader:
    pet4 = row[0]
    row[0] = "Birds"
    row.append(pet4)
    all.append(row)
  sortedlist = sorted(all, key=operator.itemgetter(0))
  return sortedlist


@app.route('/')
def form():
  return """
    <html>
        <body>
            <h1>Upload a CSV File</h1>

            <form action="/transform" method="post" enctype="multipart/form-data">
                <input type="file" name="data_file" />
                <input type="submit" />
            </form>
        </body>
    </html>
"""

@app.route('/transform', methods=["POST"])
def transform_view():
  request_file = request.files['data_file']
  if not request_file:
    return "No file"

file_contents = request_file.stream.read().decode("utf-8")

result = transform(file_contents)

response = make_response(result)
response.headers["Content-Disposition"] = "attachment; filename=result.csv"
return response

Это ошибка, которую я получаю: 

Traceback (most recent call last):
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/app.py", line 2292, in wsgi_app
response = self.full_dispatch_request()
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/app.py", line 1815, in full_dispatch_request
rv = self.handle_user_exception(e)
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/app.py", line 1718, in handle_user_exception
reraise(exc_type, exc_value, tb)
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/_compat.py", line 35, in reraise
raise value
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/app.py", line 1813, in full_dispatch_request
rv = self.dispatch_request()
File "/home/user/.virtualenvs/myvirtualenv/lib/python3.6/site-packages/flask/app.py", line 1799, in dispatch_request
return self.view_functions[rule.endpoint](**req.view_args)
File "/home/user/mysite/callnumbers.py", line 62, in transform_view
result = transform(file_contents)
File "/home/user/mysite/callnumbers.py", line 18, in transform
row[0] = row[1]
IndexError: list index out of range

Я могу сказать, что мой сценарий неправильно читает загруженный входной файл (поэтому я не могу найти данные в столбцах, на которые я пытаюсь ссылаться, и в IndexError), но мне не удалось найти любые примеры использования Flask библиотеки Python CSV для анализа загруженного файла и манипулирования столбцами так, как мне нужно. Как я могу решить это? Любые указатели в правильном направлении очень ценятся!

1 Ответ

0 голосов
/ 15 ноября 2018

Функция csv.reader() принимает текстовый объект вместо текста. Один из способов сделать это - сохранить файл и снова открыть его в функции преобразования. 

from flask import Flask, make_response, request
import operator
import io
import csv    

app = Flask(__name__)


def transform():
  with open('myfile.csv', 'rb') as f:
    reader = csv.reader(f)  
    all = []
    row = next(reader)
    row.append('Pet4')
    all.append(row)
    for row in reader:
      pet4 = row[0]
      row[0] = "Birds"
      row.append(pet4)
      all.append(row)
    sortedlist = sorted(all, key=operator.itemgetter(0))  
  return sortedlist

@app.route('/')
def form():
  return """
    <html>
        <body>
            <h1>Upload a CSV File</h1>

            <form action="/transform" method="post" enctype="multipart/form-data">
                <input type="file" name="data_file" />
                <input type="submit" />
            </form>
        </body>
    </html>
"""

@app.route('/transform', methods=["POST"])
def transform_view():
  request_file = request.files['data_file']
  request_file.save("myfile.csv")
  if not request_file:
    return "No file"
  #file_contents = io.StringIO(request_file.stream.read().decode("UTF8"), newline=None)
  #file_contents = request_file.stream.read().decode("utf-8")

  result = transform()
  print result

  response = make_response(result)
  response.headers["Content-Disposition"] = "attachment; filename=result.csv"
  return response
if __name__ == '__main__':
    app.run()
...