# 현재 폴더 위치 확인
import os
os.getcwd()

# 정리 대상 폴더 경로를 설정
target_path = './고라니'

# 압축 파일 확인
import glob
zipfile_path = []
for filename in glob.glob(os.path.join(target_path, '**/*.zip'), recursive=True):
    zipfile_path.append(filename)
    print(zipfile_path)

# 압축파일 해제
import zipfile

for filename in zipfile_path:
    with zipfile.ZipFile(filename) as myzip:
        zipinfo = myzip.infolist() # zip 파일 정보
        for info in zipinfo:
            decode_name = info.filename.encode('cp437').decode('euc-kr') # 한글 깨짐 방지
            info.filename = os.path.join(target_path, decode_name)
            myzip.extract(info)

# 주피터, 코랩 노트북에서 리눅스 환경 명령어로 모듈 설치
!pip install openpyxl

import openpyxl as opx

# 정리 대상 폴더 경로를 설정
target_path = './고라니'

# 폴더별 파일명을 입력 받아 엑셀 파일에 저장하는 함수

def getFileName(target_path):
    dir_list = os.listdir(target_path) # 고라니 폴더 안의 파일 확인
#     print(dir_list)

    wb = opx.Workbook() # 엑셀을 생성하는 객체
    ws = wb.active      # 새로 생성한 workbook의 활성화 시트를 ws로 정의
    
            #행    #열
    ws.cell(row=1, column=1).value = '파일경로' # 첫번째 로우 1번 컬럼에 파일경로로 입력
    ws.cell(row=1, column=2).value = '파일명(변경전)'
    ws.cell(row=1, column=3).value = '파일명(변경후)'
    

    
    i = 2 # 1 행에는 입력이 다 되어서 2행부터 입력할거라 2로 줌
    
    current_dir = target_path # 고라니폴더 경로 변수로 지정
    filelist = os.listdir(current_dir) # 고라니 폴더 안의 파일이름들 변수로 지정
    
    for filename in filelist:
        ws.cell(row = i, column = 1).value = current_dir +"/" 
        # ws엑셀을 .cell(i행 1열부터 값을 고라니폴더경로에서 하위 경로를 표현해주기위해 / 더한것 )
        
        ws.cell(row = i, column = 2).value = filename
        # ws엑셀을 i행 2열에 파일이름을 i증가하는값에다가 하나씩 넣기
        i += 1
    
    wb.save(os.path.join(target_path, 'filelist.xlsx'))

# 선생님 코드

# 폴더별 파일명을 입력 받아 엑셀파일에 저장하는 함수
def getFileName(target_path):

    wb = opx.Workbook()
    ws = wb.active # 새로 생성한 WorkBook의 활성화 시트를 ws로 정의
    ws.cell(row=1, column=1).value = '파일경로'
    ws.cell(row=1, column=2).value = '파일명(변경전)'
    ws.cell(row=1, column=3).value = '파일명(변경후)'
    i = 2
    current_dir = target_path
    filelist = os.listdir(current_dir)
    for filename in filelist:
        ws.cell(row=i, column=1).value = current_dir + "/"
        ws.cell(row=i, column=2).value = filename
        i = i+1
    wb.save(os.path.join(target_path, 'filelist.xlsx'))

getFileName(target_path)

def excelRead(file_path : str) -> tuple:
    
    wb = opx.load_workbook(file_path)
    ws = wb.active
    
    dirpath = [r[0].value for r in ws]
    file_before = [r[1].value for r in ws]
    file_after = [r[2].value for r in ws]
    
    len_num = len(dirpath) # for문 몇바퀴 돌까
    datalist = []
    
    for i in range(1, len_num):
        temp_tuple = (dirpath[i], file_before[i], file_after[i])
        datalist.append(temp_tuple)

    return datalist

rename_list = excelRead(os.path.join(target_path, 'filelist.xlsx'))
print(rename_list)

# 파일 이름 변경하기
import shutil

def fileRename(datalist : list):
    try:
        for data in datalist:
            print(data[1] + '의 파일명을 -> ' + data[2] + '로 변경합니다')
            shutil.move(data[0] + data[1], data[0] + data[2])
    except FileNotFoundError as e:
        print('이미 변경되었습니다.')

fileRename(rename_list)

import fnmatch

# 카테고리 리스트 만들기

def categoryList(target_path : str) -> list:
    file_list = []
    for filename in os.listdir(target_path): # 파일 패스를 받아서 
        if fnmatch.fnmatch(filename, '*_[0-9][0-9][0-9].*'):
            file_list.append(filename)
            
    category = []
    for file in file_list:
        
        # A_2022_06_30_생산로그_생산_002 이라면
        temp_list = file.split('_') # ['b', '2022', '02', .... 이런식으로 저장됌]
        category.append(temp_list[-2]) # 생산 부분에 있는글자를 리스트에 추가한다.
        
    temp_set = set(category)
    result = list(temp_set)
    return result
        

categoryList(target_path)

categorylist = categoryList(target_path) + ['기타']
print(categorylist)

# 폴더 생성하기
import pathlib

new_path = './new_dir'

def makeDir(new_path:str, categorylist:list):
    for category in categorylist:
        new_dir = pathlib.Path(os.path.join(new_path, category))
        new_dir.mkdir(parents=True, exist_ok=True) # 부모폴더가 없으면 만들고 # 이미 존재하면 덮어쓰기

makeDir(new_path, categorylist)

import shutil

new_path = './new_dir'
target_path = './고라니'
categorylist = categoryList(target_path) + ['기타']

import fileinput
import os # 현재 os의 파일들의 경로를 확인하는데 도움을 주는 라이브러리
import glob

# 현재 경로 확인
os.getcwd()

# 디렉토리 내 파일 확인
os.listdir(os.getcwd())

# 경로 설정
path = 'sample/'

# glob(): 해당 경로의 파일 이름을 리스트로 반환
glob.glob(os.path.join(path, '*txt'))

with fileinput.input(glob.glob(os.path.join(path, '*txt'))) as f:
    for line in f:
        print(line)

txt_files = glob.glob(os.path.join(path, '*txt'))

txt_files

# 각 파일의 첫번째 라인을 찾아 변경하기 # 위에서 모든 텍스트 파일로 지정해서 모든것이 바뀜
with fileinput.input(txt_files, inplace=True) as f: # inplace: 덮어쓰다
    for line in f:
        if f.isfirstline(): # 이게 첫번째라인인지가 T,F 로 반환되는 메소드
            print('첫번째 라인 입니다', end='\n')            
        else:
            print(line, end='')
print(line)

# 검색된 라인 변경하기
with fileinput.input(txt_files, inplace=True) as f:
    for line in f:
        if line == '첫번째 라인 입니다\n':
            print('1번째 라인입니다', end='\n')
        else:
            print(line, end='')

# 키워드 포함 변경하기
with fileinput.input(txt_files, inplace=True) as f:
    for line in f:
        if '1번째' in line:
            print('첫번째 줄입니다', end='\n')
        else:
            print(line, end='')

# 텍스트 치환하기
with fileinput.input(txt_files, inplace=True) as f:
    for line in f:
        if '12번째' in line:
            line.replace('12번째', '열두번째', end='')
        else:
            print(line, end='')

import pickle

data = ['apple', 'banana', 'orange']

# 파일 저장
with open('list.pkl', 'wb') as f:
    pickle.dump(data, f)

# 파일 읽기
with open('list.pkl', 'rb') as f:
    data = pickle.load(f)

type(data)

print(data)

# 딕셔너리 저장
data = {}
data[1] = {'id': 1, 'userid': 'apple', 'name': '김사과', 'gender': '여자', 'age': 20}

# 파일저장
with open('dict.pkl', 'wb') as f:
    pickle.dump(data, f)

with open('dict.pkl', 'rb') as f:
    data = pickle.load(f)

print(data)

os.getcwd()

for filename in glob.glob('*.txt'):
    print(filename)

# txt 파일 하위경로에서 찾기:
for filename in glob.glob('**/*.txt'):
    print(filename)

# txt 파일 찾기 : 현재와 하위경로 모두 포함
for filename in glob.glob('**/*.txt', recursive=True):
    print(filename)

# 파일명 글자수로 찾기
for filename in glob.glob('????.*', recursive=True): # 글자수 4개  # ? 의 개수 == 글자수
    print(filename)

for filename in glob.glob('???????.*', recursive=True): # 글자수 7개
    print(filename)

# 문자열 패턴 포함 파일명 찾기
for filename in glob.glob('[a-z]???.*', recursive=True): # [] 안에 범위지정 
    # 모든 파일명부분 물음표칸에 범위지정 가능
    print(filename)

for filename in glob.glob('**/새파일*.*'):
    print(filename)

# 프로젝트라는 단어가 들어가는 파일을 찾기
for filename in glob.glob('**/*프로젝트*.*'):
    print(filename)

import fnmatch

# 파일명은 '새' 로 시작하고 확장명은 .txt를 검색
# 확장자를 제외한 파일명의 길이는 4개이며 파일명의 마지막 문자는 숫자
for filename in os.listdir('./sample'):
    if fnmatch.fnmatch(filename, '새??[0-9].txt'):
        print(filename)

import shutil

# 파일 복사하기
shutil.copy('./sample/새파일1.txt', './sample/새파일1_복사본.txt')

# 파일 이동하기
shutil.move('./sample/새파일1_복사본.txt', './새파일1_복사본.txt') # 윈도우에서는 오른쪽 객체는 ./ 생략가능

# 확장명 바꾸기 # 덮어쓰기
shutil.move('./새파일1_복사본.txt', './새파일1_복사본.py')

shutil.move('./새파일1_복사본.py', './새파일1_복사본.txt')

import zlib

data = 'Hello Python!' * 10000

print(len(data)) # 130000 byte

# 압축하기
compress_data = zlib.compress(data.encode(encoding='utf-8'))
print(len(compress_data))

compress_data

# 압축 해제
org_data = zlib.decompress(compress_data).decode('utf-8')
print(len(org_data))

import gzip

with open('org_data.txt', 'w') as f:
    f.write(data)

# gzip 압축 하기
with gzip.open('compressed.txt.gz', 'wb') as f:
    f.write(data.encode('utf-8'))

# gzip 압축 풀기
with gzip.open('compressed.txt.gz', 'rb') as f:
    org_data = f.read().decode('utf-8')

print(len(org_data))

import zipfile

# 파일 합치고 압축하기
with zipfile.ZipFile('./sample/새파일.zip', 'w') as myzip:
    myzip.write('./sample/새파일1.txt')
    myzip.write('./sample/새파일2.txt')
    myzip.write('./sample/새파일3.txt')
    myzip.write('./sample/새파일4.txt')
    myzip.write('./sample/새파일5.txt')

# 압축 해제하기
with zipfile.ZipFile('./sample/새파일.zip') as myzip:
    myzip.extractall()

import tarfile

with tarfile.open('./sample/새파일.tar', 'w') as mytar:
    mytar.add('./sample/새파일1.txt')
    mytar.add('./sample/새파일2.txt')
    mytar.add('./sample/새파일3.txt')
    mytar.add('./sample/새파일4.txt')
    mytar.add('./sample/새파일5.txt')