# 실습에 쓸 데이터를 인터넷에서 받아온다.
# urllib 은 파이썬에 기본으로 들어 있는 인터넷 도구다. 설치할 것이 없다
import urllib.request

# Path 는 폴더와 파일 경로를 다루는 도구다. 이것도 기본으로 들어 있다
from pathlib import Path

# raw.githubusercontent.com 은 깃허브에 올라간 파일의 "날것"을 주는 주소다.
# 브라우저용 꾸밈 없이 파일 내용만 그대로 온다
base = "https://raw.githubusercontent.com/tubi55/dcodelab-class-data/main/"

# data 라는 폴더를 만든다. exist_ok=True 는 "이미 있으면 그냥 넘어가라"는 뜻이다.
# 이게 없으면 두 번째 실행할 때 에러가 난다
Path("data").mkdir(exist_ok=True)

for name in ["customers.csv", "products.csv", "purchases.csv"]:
    # urlretrieve 는 주소의 파일을 그대로 받아 저장한다.
    # 앞이 받아올 주소, 뒤가 저장할 자리다
    urllib.request.urlretrieve(base + name, f"data/{name}")
    print("받음:", name)

# csv 는 파이썬에 기본으로 들어 있는 도구다. 따로 설치할 게 없다.
# 쉼표로 나뉜 파일을 알아서 칸별로 잘라준다
import csv

for name in ["customers.csv", "products.csv", "purchases.csv"]:
    # open 은 파일을 연다. encoding="utf-8" 은 "한글이 든 파일이다" 라고 알려주는 것이다.
    # 이걸 안 적으면 윈도우에서 한글이 깨지거나 에러가 난다.
    # with 로 열면 블록이 끝날 때 파이썬이 알아서 파일을 닫아준다
    with open(f"data/{name}", encoding="utf-8") as f:
        # DictReader 는 첫 줄을 "칸 이름"으로 보고, 나머지 줄을 딕셔너리로 만들어준다.
        #   {'customer_id': 'C001', 'name': '박은수', ...}
        # 그냥 두면 한 줄씩 흘려보내는 물건이라, list() 로 감싸 전부 꺼내 담는다
        rows = list(csv.DictReader(f))

    print("==", name, len(rows), "줄")      # len 은 개수를 센다
    print("칸 이름:", list(rows[0]))        # 표[0] = 첫 줄. list() 로 감싸면 칸 이름만 나온다
    print("첫 줄  :", rows[0])
    print()                              # 빈 줄 하나

import csv
import sqlite3          # 데이터베이스 도구. 이것도 파이썬에 기본으로 들어 있다

# connect 는 데이터베이스 파일에 연결한다.
# 그 이름의 파일이 없으면 새로 만들고, 있으면 그냥 연다.
# con(connection) = "데이터베이스와 연결된 통로" 라고 생각하면 된다
con = sqlite3.connect("shop.db")

# cur(cursor) = "명령을 대신 실행해주고 결과를 받아오는 손" 이다.
# 통로(con)로는 명령을 못 보내고, 반드시 이 손(cur)을 통해 보낸다.
# 왜 나뉘어 있냐면 — 통로 하나로 여러 손이 각자 다른 조회를 할 수 있어서다
cur = con.cursor()

# execute 는 SQL 문장 하나만 실행한다. 여기서는 표를 셋 만들어야 하므로
# 여러 문장을 한 번에 실행하는 executescript 를 쓴다.
# 큰따옴표 세 개(""")로 감싸면 여러 줄짜리 글자를 그대로 적을 수 있다.
# 아래 -- 로 시작하는 줄은 SQL 의 주석이다 (파이썬의 # 과 같은 역할)
cur.executescript("""
CREATE TABLE customers (
    customer_id TEXT PRIMARY KEY,   -- TEXT=글자 · PRIMARY KEY=이 칸이 이 표의 이름표다.
                                    -- 같은 값이 두 번 들어오면 데이터베이스가 막아준다
    name        TEXT,
    gender      TEXT,
    age         INTEGER,            -- 숫자로 넣는다. 나중에 나이로 거르려면 필요하다
    skin_type   TEXT,
    phone       TEXT,
    email       TEXT,
    city        TEXT,
    joined_at   TEXT
);

CREATE TABLE products (
    product_id  TEXT PRIMARY KEY,
    name        TEXT,
    brand       TEXT,
    category    TEXT,
    price       INTEGER,            -- 가격도 숫자다. 3만원 이하를 세려면 반드시
    volume      TEXT,
    skin_type   TEXT,
    ingredient  TEXT,
    concern     TEXT,
    tags        TEXT,
    description TEXT                -- 다음 편에서 이 글을 숫자로 바꾼다
);

CREATE TABLE purchases (
    purchase_id  TEXT PRIMARY KEY,
    customer_id  TEXT,              -- 고객표의 번호를 가리킨다
    product_id   TEXT,              -- 상품표의 번호를 가리킨다
    purchased_at TEXT,
    quantity     INTEGER,
    rating       INTEGER,
    review       TEXT,
    is_holdout   INTEGER            -- 오늘의 함정. 뒤에서 본다
);
""")

def load(table, name):
    """CSV 파일 하나를 읽어서 표 하나에 통째로 넣는다.

    table 은 넣을 표 이름, name 은 읽을 파일 이름이다.
    """
    # 앞에서 열어볼 때와 똑같다. encoding 을 적어야 한글이 안 깨지고,
    # with 로 열면 블록이 끝날 때 파이썬이 알아서 닫아준다
    with open(f"data/{name}", encoding="utf-8") as f:
        rows = list(csv.DictReader(f))

    # 칸 이름 목록을 뽑는다. 첫 줄의 열쇠들이 곧 칸 이름이다
    cols = list(rows[0])                       # ['customer_id', 'name', 'gender', ...]

    # 파이썬에서는 글자에 곱하기를 하면 그만큼 반복된다.  "?" * 3  ->  "???"
    # join 은 목록을 정해진 글자로 이어붙인다.  ",".join("???")  ->  "?,?,?"
    # 칸이 9개면 "?,?,?,?,?,?,?,?,?" 가 만들어진다 — 값이 들어갈 빈자리다
    marks = ",".join("?" * len(cols))

    # 넣을 값을 줄마다 하나씩 준비한다. 아래 한 줄을 풀어 쓰면 이렇다 ─
    #   values = []
    #   for row in rows:
    #       values.append(tuple(row[c] for c in cols))
    # tuple 은 "순서가 정해진 값 묶음"이다. 물음표 순서와 짝이 맞아야 해서 목록이 아니라
    # 칸 순서대로 꺼낸 묶음으로 만든다
    values = [tuple(row[c] for c in cols) for row in rows]

    # execute 는 한 줄씩, executemany 는 여러 줄을 한 번에 넣는다.
    # 1,500번을 하나씩 보내는 것보다 훨씬 빠르다
    cur.executemany(
        f"INSERT INTO {table} VALUES ({marks})",
        values,
    )

    # {table:10s} 는 "글자를 10칸 폭으로 왼쪽 정렬", {len(rows):5d} 는 "숫자를 5칸 폭으로"
    print(f"{table:10s} {len(rows):5d}줄")


load("customers", "customers.csv")
load("products", "products.csv")
load("purchases", "purchases.csv")

# 지금까지 넣은 것을 파일에 확정해서 새긴다. 이걸 빼먹으면 아무것도 안 남는다
con.commit()

# 다 썼으면 연결을 닫는다
con.close()