파이썬(Django) :: 데이터베이스에 csv파일 삽입하기

이번에는 python 코드를 이용하여 데이터베이스에 csv파일을 넣는 방법을 작성하려고한다.

 

이번 글에서는 생성한 앱에서 City 라는 클래스를 임포트 후 해당 테이블에 데이터를 넣는 작업을 위한 코드를 작성하였다.

 

아래 코드를 사용하면 데이터는 물론 foreign key 까지 모두 넣을 수 있다.

단, foreign key에 데이터를 넣기 위해서는 테이블 칼럼명에 "_id"를 붙여줘야 넣을 수 있다.

import csv
import os
import django

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "[ProjectName].settings")
django.setup()

from [AppName].models import City

CSV_PATH = '/Users/wave/Desktop/csv/cities.csv'

with open(CSV_PATH, newline='') as csvfile:
	data_reader = csv.DictReader(csvfile)
	for row in data_reader:
		print(row)
		City.objects.create(
                         name       = row['name'],
                         image      = row['image'],
                         country_id = row['country_id'],
                  )