Fall in love with Python

Finished my first python program. A good start.

2019-04-09

# author: Weiliang Luo
import xlsxwriter
import os
import cv2
import time

def creat_excel():
    filename_excel = r'C:\Users\admin\PycharmProjects\img_1\test.xlsx'
    path = r'C:\Users\admin\PycharmProjects\img_1\picture'
    w = xlsxwriter.Workbook(filename_excel)
    ws = w.add_worksheet('mysheet')

    def visitDir(path):
        if not os.path.isdir(path):
            print('Error: "', path, '" is not a directory or does not exist.')
            return
        else:
            global x
            x = 1
            print(x)
            for lists in os.listdir(path):
                sub_path = os.path.join(path, lists)
                x += 1
                print(x,'    ',sub_path)
        return  x

    def read_directory(directory_name):
        array_of_img = []
        k = 0
        for j in range(0,x):
            ws.set_row(j, 70)
            ws.set_column(j, x, 30)
        for filename in os.listdir(path):

            sub_path = os.path.join(path, filename)
            img = cv2.imread(directory_name + "/" + filename)
            # print(image_file)
            array_of_img.append(img)
            time.sleep(0.01)
            ws.insert_image(0, k, sub_path, {'x_scale': 0.2, 'y_scale': 0.2})
            k += 1
            if os.path.isdir(sub_path):
                read_directory()
                print(sub_path)

            def img_show():
                cv2.namedWindow('image', cv2.WINDOW_NORMAL)
                # cv2.imshow('image',img)
                cv2.imshow('image',array_of_img[j])
                cv2.waitKey(0)
                cv2.destroyAllWindows()
            # img_show()

    visitDir(path)
    read_directory(path)
    w.close()


if __name__ == '__main__':
    creat_excel()