from Tkinter import *
import cv2
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import os
import shutil
import csv

file_name_list = []
csv_list = []

now_id = 0
dic_name = ""


def is_normal():
    global now_id
    if now_id < 0:
        now_id = 0
    if now_id >= len(file_name_list):
        now_id = len(file_name_list)-1
def init():
    global dic_name
    dic_name = raw_input("dict name: ")
    plt.ion()

    for root, dirs, files in os.walk("./clip/"+dic_name+"/", topdown=False):
        for name in files:
            print (name)
            file_name_list.append(name)
    read_csv("./csv/box_key.csv")
def up_pic():
    global now_id
    is_normal()
    name = "clip/"+dic_name+"/"+file_name_list[now_id]
    print (name)
    I = mpimg.imread(name)
    plt.imshow(I)
    plt.show()
    now_id -= 1
def next_pic():
    global now_id
    is_normal()
    name = "clip/"+dic_name+"/"+file_name_list[now_id]
    print (name)
    I = mpimg.imread(name)
    plt.imshow(I)
    plt.show()
    now_id += 1

def move_error():
    shutil.move("./clip/"+dic_name+"/"+file_name_list[now_id],"./error/"+dic_name)
    file_name_list.pop(now_id)
    next_pic()

def del_csv(csv_reader,line):
    csv_reader.drop(line)
    csv_reader.close()

    return True

def find_error():
    init()

    root = Tk()

    Button(root, text='up', width=20,height=10,command=up_pic).pack()
    Button(root, text='next', width=20,height=10,command=next_pic).pack()
    Button(root, text='move', width=20,height=10,command=move_error).pack()

    root.mainloop()


read_csv("./csv/box_key.csv")
dir_name_erro = raw_input("dic_name")
for root, dirs, files in os.walk("./error/"+dic_name+"/", topdown=False):
        for name in files:
            print (name)