#!/usr/bin/env python3

import glob
import sys

DIR='products_img'

def list_files_glob(pattern=f'{DIR}/**/*.jpg', recursive=True, fprint=False):
    files = glob.glob(pattern, recursive=recursive)
    if fprint:
        for file in files:
            print(file)
    return files


def make_map(files):
    d = {}
    hl_cnt, ost_cnt = 0, 0
    for f in files:
        fname = f[f.find('/')+1:-4]
        if '_' in fname:
            hlavni = fname[:fname.find('_')]
        else:
            hlavni = fname
        sku = hlavni[4:]
        if sku in d:
            pr = d[sku]
        else:
            pr = {'main': '', 'other': []}
            d[sku] = pr

        if fname == hlavni:
            pr['main'] = fname+'.jpg'
            hl_cnt += 1
        else:
            pr['other'].append(fname+'.jpg')
            ost_cnt += 1
    #print(hl_cnt, ost_cnt)
    return d

def map2csv(d):
    print('SKU;hlavni;ostatni')
    for sku,pr in d.items():
        print(f"{sku.lower()};{pr['main']};{' '.join(pr['other'])}")

files = list_files_glob()
#print(len(list(files)))
#sys.exit(-1)
d = make_map(files)
#print(d)
map2csv(d)