들어가며
웹 사이트를 만들다 보면 테이블 등을 csv파일로 다운받을 수 있도록 만들어달라는 요청이 자주 있습니다. 이번 글에서는 Flask에서 특정 URL로 들어갈 때 CSV파일을 받을 수 있도록 만들고, 다운받은 CSV파일을 엑셀로 열 때 한글이 깨지지 않게 처리해 봅시다.
이번에는 Flask + SQLAlchemy + Pandas를 사용합니다.
Flask 코드짜기
우선 Flask 코드를 하나 봅시다. app.py라는 이름을 갖고 있다고 생각해 봅시다.
아래 코드는 Post라는 모델을 모두 가져와 df라는 DataFrame객체로 만든 뒤 .to_csv를 통해 csv 객체로 만들어 준 뒤 StringIO를 통해 실제 io가 가능한 바이너리형태로 만들어 줍니다.
또, output을 해주기 전 u'\ufeff'를 미리 넣어줘 이 파일이 ‘UTF-8 with BOM’이라는 방식으로 인코딩 되어있다는 것을 명시적으로 알려줍니다.
인코딩 명시를 빼면 엑셀에서 파일을 열 때 한글이 깨져서 나옵니다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
from io import StringIO
from flask import Flask, jsonify, request, Response
from flask_sqlalchemy import SQLAlchemy
app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = '데이터베이스 URI'
db = SQLAlchemy()
db.init_app(app)
@app.route('/api/post/csv/') # URL 설정하기
def post_list_csv(self):
queryset = Post.query.all()
df = pd.read_sql(queryset.statement, queryset.session.bind)
output = StringIO()
output.write(u'\ufeff')
df.to_csv(output)
response = Response(
output.getvalue(),
mimetype="text/csv",
content_type='application/octet-stream',
)
response.headers["Content-Disposition"] = "attachment; filename=post_export.csv"
return response
|
