import os
import shutil


def zip_directory(src='./', des='../zip/'):
    # 디렉토리 이름이 '/'로 끝나지 않으면 끝에 '/' 삽입
    if src[-1] != '/': src += '/'
    if des[-1] != '/': des += '/'
    
    # src와 des 경로 존재 확인
    # src 경로가 없거나 des 경로를 만들 수 없다면 종료
    # des 경로가 없다면 des 경로 생성
    if not os.path.exists(src):
        print ('Error: %s does not exist' %(src))
        return
    try:
        if not os.path.exists(des):
            os.makedirs(des)
    except OSError:
        print('Error: Creating directory. ' + des)
        return
        
    
    # README 작성
    f = open(des+"README.md", 'w')
    f.write("This directory is the compressed directory of the \"%s\" directory.\n" %(src))
    f.write("Files and directories in this directory follow the rules below.\n")
    f.write("\n")
    f.write("rule 1. All of the directories are copied.\n")
    f.write("rule 2. If there are more than 100 files in a directory, only 100 files of the same type are copied.\n")
    f.write("rule 3. Files over 100MB are copied only by the file name.\n")
    f.write("\t(The file will contain the phrase \"This file is more than 100MB.\").\n")
    f.close()
    
    # rule에 따라 src 파일들을 des로 복사
    global error_list
    error_list = ""
    copy_directory(src, des)
    
    # README 에 error list 추가
    f = open(des+"README.md", 'a')
    f.write("\n")
    f.write("*** Error ***\n")
    f.write(error_list)
    f.close()

    print("Success to zip the directory")
    
    
def copy_directory(src, des):
    global error_list
    
    # src 디렉토리에 있는 파일 목록 가져오기
    file_list = os.listdir(src)
    
    extension_count = {}
    for file in file_list:
        # file이 폴더인 경우
        if os.path.isdir(src + file):
            # des/file/ 디렉토리 생성
            try:
                if not os.path.exists(des + file):
                    os.makedirs(des + file)
            except OSError:
                print('Error: Creating directory. ' + des + file)
                error_list += 'Error: Creating directory. ' + des + file + '\n'
                continue
            
            # src/file/ 파일들을 des/file/ 로 복사
            copy_directory(src+file+'/', des+file+'/')
            
        # file이 폴더가 아닌 파일인 경우
        else:
            # 확장자 추출
            extension = os.path.splitext(file)[1]
            
            # 처음 나온 확장자면 사전에 추가
            if extension not in extension_count:
                extension_count[extension] = 0
            
            # 해당 확장자의 파일이 100개가 넘으면 패스
            if extension_count[extension] >= 100:
                continue
                
            # 파일이 100MB 이하면 파일 복사
            MB = 1024 * 1024
            if os.path.getsize(src+file) < 100 * MB:
                shutil.copy2(src+file, des+file)
            
            # 파일이 100MB 이상이면 파일 이름만 복사
            else:
                print("! %s is more than 100MB.\n" %(src+file))
                error_list += "! %s is more than 100MB.\n" %(src+file)
                f = open(des+file, 'w')
                f.write("This file is more than 100MB.")
                f.close()
            
            # 확장자 파일 카운트 증가
            extension_count[extension] += 1
            
            
def main():
    src = input("src: ")
    des = input("des: ")
    
    zip_directory(src, des)
    
    
if __name__ == "__main__":
    main()