CSV.file to Python-2-

sihwan_e·2020년 6월 1일
0

Django

목록 보기
12/21
import os
import sys
import csv
import django

os.environ.setdefault("DJANGO_SETTING_MODULE", "aboutteatime.settings")
django.setup()

from store.models import Store, StoreType
CSV_PATH_STORES = './output.csv'

def insert_store_type():
    with open(CSV_PATH_STORES) as in_file:
    data_reader = csv.reader(in_file)
    next(data_reader, None)
    for row in data_reader:
    	store_type = row[1]
        
        if StoreType.objects.filter(name=store_type).exist():
            pass
        else:
            StoreType.objects.create(name=store_type)
            
def insert_store():
    with open(CSV_PATH_STORES) as in_file:
    data_reader = csv.reader(in_file)
    next(data_reader, None)
    for row in data_reader:
    	store_type = row[1]
        type_id = StoreType.objects.get(name=store_type).id
        store_name = row[0]
        store_address = row[2]
        store_contact = row[3]
        store_open = row[4]
        store_lng = row[5]
        store_lat = row[6]
        Store.objects.create(name=store_name, address=store_address, contact=store_contact, opening_hours=store_open, longitude=store_lng, latitude=store_lat, category_id=type_id)
        
insert_store()
with open(CSV_PATH_STORES) as in_file:
if StoreType.objects.filter(name=store_type).exist():
            pass
        else:
            StoreType.objects.create(name=store_type)
            

Storetype테이블에 인스턴스(name)가 존재한다면 pass하고
그렇지 않으면 추가한다.

그리고 보면 알겠지만 당연히 테이블마다 만들어줘야함.

profile
Sometimes you gotta run before you can walk.

0개의 댓글