はじめに

Djangoを使ってお酒のECサイトを作っています。
商品のデータをCSVでインポートしようとした際にエラーと戦ったので、その時に調べた解決策を書いた記事になります。

Django2
PostgreSQL

やりたかったこと

インポートする前に2つのテーブルを作成した。
一つは商品の分類をするCategoryテーブル。もう一つは商品情報のProductテーブル。

models.py

# Metaタグなどは省略
class Category(models.Model):
    name = models.CharField(max_length=250, unique=True)
    slug = models.SlugField(max_length=250, unique=True)
    description = models.TextField(blank=True)
    image = models.ImageField(upload_to='category', blank=True)

class Product(models.Model):
    name = models.CharField(max_length=250, unique=True)
    slug = models.SlugField(max_length=250, unique=True)
    description = models.TextField(blank=True)
    category = models.ForeignKey(Category, on_delete=models.CASCADE)#★今回問題のやつ★
    price = models.IntegerField()
    image = models.ImageField(upload_to='product', blank=True)
    stock = models.IntegerField()
    available = models.BooleanField(default=True)
    created = models.DateTimeField(auto_now_add=True)
    updated = models.DateTimeField(auto_now=True, null=True)

インポートしたいCSVファイルは以下。

Category.csv

2. Product.csv

このCSVファイルをインポートしてテーブルに入れることが目標

出てきたエラー

まずはCategory.csvをインポートした。

practice_db=> \copy shop_product (name,slug,description,image) FROM 'path/to/Category.csv' (DELIMITER ',', FORMAT CSV, ENCODING 'UTF8', HEADER);
COPY 3 #正常にインポート

次にProduct.csvをインポートをしようとするとエラーが出た。

practice_db=> \copy shop_product (name,slug,description,category,price,image,stock,available,created) FROM 'path/to/Product.csv' (DELIMITER ',', FORMAT CSV, ENCODING 'UTF8', HEADER);
ERROR:リレーション"shop_product"の列"category"は存在しません

解決方法

エラーが出た時にインポートしようとしたProduct.csvのcategoryのデータにはCategory.csvのnameの値が入っている。

これをnameの値で指定するのではなく、Categoryのidで指定しないとエラーが出てしまう。
さらに、カラム名をcategoryではなくcategory_idと変更する。
変更後のProduct.csvファイルは以下。

どうやら、ForeiginKey型のデータにはidが入る決まりらしい。
よく考えたら、Categoryのnameではなく、絶対に重複しないidで指定するべきだなと思った。

参考サイト

DjangoでcsvファイルをアップロードしてDBにレコードを追加する方法