import pandas as pd
import requests
import os
from PIL import Image
from io import BytesIO

df = pd.read_table("../stock_each_shot_code.csv")

category_count = {}
for record in df.to_records():
    url = record[2]
    category = record[3]
    if category not in os.listdir():
        os.mkdir(category)
    if category not in category_count:
        category_count[category] = 1
    image_name = str(category_count[category]) + ".jpg"
    if image_name not in os.listdir(category):
        try:
            response = requests.get(url)
            img = Image.open(BytesIO(response.content)).convert('RGB')
            img.save(category + "/" + image_name)
            category_count[category] += 1
        except Exception:
            print("----------")
            print(category)
            print(url)