Python/Django アプリでよく使うコード【都道府県】

この記事の背景

例えば店舗情報を登録するようなアプリの場合、その店舗の都道府県を設定することがあると思います。

この機能をいつでも使えるように、備忘録的にこの記事に残しました。

コード

models.py

class Pref(models.Model):
    code = models.CharField(max_length=2)
    name = models.CharField(max_length=4)

    def __str__(self):
        return str(self.name)

views.py

def upload(request):
    if 'csv' in request.FILES:
        form_data = TextIOWrapper(request.FILES['csv'].file, encoding='shift_jis')
        csv_file = csv.reader(form_data)
        for line in csv_file:
            pref, created = Pref.objects.get_or_create(code=line[0])
            pref.code = line[0]
            pref.name = line[1]
            pref.save()

        return render(request, 'pref/upload.html')

    else:
        return render(request, 'pref/upload.html')

importするCSVファイル

以上です。