#!/usr/bin/env python

import os
import codecs
import csv

__author__ = 'Holger Knust'


def filter_inventory(in_file, out_file):
    inv_index = 5
    with codecs.open(in_file, "r", encoding='utf-8', errors='replace') as f, codecs.open(out_file, "w", encoding='utf-8') as of:
        reader = csv.reader(f, delimiter="," )
        writer = csv.writer(of, delimiter=",", quoting=csv.QUOTE_ALL, lineterminator="\n")

        for row in reader:
            row[inv_index] = row[inv_index].replace("+++", "100")
            inv_text = row[inv_index]

            try:
                in_stock = int(inv_text)
            except ValueError:
                in_stock = 0

            if in_stock > 0:
                writer.writerow(row)


def process_files(data_root):
    files = os.listdir(data_root)

    in_file = os.path.join(data_root, "hummel.txt")
    out_file = os.path.join(data_root, "hk_hummel.csv")

    if os.path.isfile(in_file):
        try:
            filter_inventory(in_file, out_file)
        except Exception as e:
            print(e)

def main():
    data_root = r"/ftproot/"
    # data_root = r"/home/holger/git/thelen/data/"

    inv_data_root = os.path.join(data_root, "Hummel")
    process_files(inv_data_root)


if __name__ == "__main__":
    main()