【Python3】csvファイル 関数について

商品のマスターデータ
→/input/items.csv
商品の売上データ（日毎に別ファイル）
→/input/sales_raw_20161030.csv
 /input/sales_raw_20161101.csv
 ...
 /input/sales_raw_20161106.csv
書き出し後の売上データ
→/output/sales.csv

商品ID、商品名、商品価格

購入ID、ユーザーID、商品ID、個数、販売日時

購入ID、ユーザーID、商品ID、商品名、商品価格、個数、販売日時

import os
import re

ITEMS_DATA = 'input/items.csv'
ITEMS_COLUMNS = ['item_id', 'name', 'price']

SALES_RAW_REGEX = re.compile(r'^sales_raw_(\d{4})(\d{2})(\d{2}).csv$')
SALES_COLUMNS = ['purchase_id', 'user_id', 'item_id',
                 'item_name', 'item_price', 'amount', 'sold_at']
SALES_DATA = 'output/sales.csv'

ENCODING = 'utf-8'
CSV_INPUT = input('')

# CSVファイルを読み込んで辞書のリストで返す関数を実装する
def read_csv(f, columns):
    pass

# Step5では以下の状態で実装する
def write_csv(f, data, columns):
    pass

# Step5では以下の状態で実装する
def read_items():
    pass

# Step5では以下の状態で実装する
def read_sales_raw(target_year, target_month):
    pass

# Step5では以下の状態で実装する
def write_sales(sales):
    pass

def main():
    # 商品のマスターデータ読み込み
    items = {}
    with open(ITEMS_DATA, encoding=ENCODING) as f:
        for row in read_csv(f, ITEMS_COLUMNS):
        # CSVファイルを読み込んで辞書のリストで返す関数使う
        item_id, item_name, item_price = row.rstrip().split(',')
        items[item_id] = {
            'item_name': item_name,
            'item_price': item_price
        }

    # 売上の生データを読み込んで、まとめる
    sales = []
    for filename in os.listdir('input/'):
        # * 対象の月のデータのみ読み込み
        if re.match(CSV_INPUT, filename):
            with open(os.path.join('input/', filename), encoding=ENCODING) as f:
                for row in f:
                    data = row.rstrip().split(',')
                    # 商品の情報を追加する
                    item_id = data[2]
                    if item_id in items:
                        sales.append({
                                'purchase_id': data[0],
                                'user_id': data[1],
                                'item_id': data[2],
                                'item_name': items[item_id]['item_name'],
                                'item_price': items[item_id]['item_price'],
                                'amount': data[3],
                                'sold_at': data[4]
                            })

    # まとめた売上データを書き出し
    m = SALES_RAW_REGEX.search(filename)
    if len(sales) == 0:
        print('書き出しファイルがありません')
        return
    SALES_COLUMNS
    with open(SALES_DATA, mode='w', encoding=ENCODING) as f:
        for row in sales:
            row_str = ','.join(str(row[column]) for column in SALES_COLUMNS)
            f.write(row_str + '\n')

if __name__ == "__main__":
    main()