openpyxl 연습 - 2 (+엑셀 슬라이서)

검색어 중심으로 CVE 검색해 놓은 걸 팀장님이 대분류 Vendor, 소분류 Product로 다시 정리해달라고 하셨다. 이걸 하려고 보니까 역시 양이 방대해서 CVE 조사할때 크롤링 프로그램 만든 것 처럼 엑셀 자동화 프로그램으로 이걸 수행하면 어떨까 싶어서 회사 와서 초반에 공부했던 openpyxl 모듈을 사용하여 코딩을 해봤다.

 

일단 내가 생각한 건 Vendor별로 분류를 하는거니까 같은 Vendor를 갖고있는 모든 CVE를 다 가져와야 하므로 key변수에 찾고자하는 Vendor명을 입력하고 이것과 같으면 메모장에 해당 CVE의 ID를 write하여 저장하는 식으로 코딩을 해야겠다고 생각했다.

 

 

먼저 내가 짠 전체 코드다.

 

import openpyxl
from openpyxl import Workbook, load_workbook
import sys

key = 'd-link'
vendor = ''
product = ''
cve = ''
cnt = 0
filename = input('Enter Output File Name (Default : search.txt) : ') or 'search.txt'

wb = Workbook()

if not filename.endswith('.txt'):
    filename += '.txt'

try :
    f = open(filename, 'wt')
    for i in range(54, 4031):
        wb = load_workbook('sheet.xlsx')
        sheet = wb.active

        vendor = sheet.cell(i, 7).value
        cve = sheet.cell(i, 5).value

        if vendor != key:
           continue
        else:
            cnt += 1
            if cve == None:
                f.write('\n')
                cve = 'none'
            else:
                f.write(cve + '\n')

            print(cnt, ': ', cve)


    print('---------- END ----------')
    print('total: ', cnt)
    f.close

except KeyboardInterrupt:
    print('End the process, the program has exited...')
    sys.exit()

 

이렇게 해서 잘 돌아가긴 하는데 속도가 너어ㅓ어어ㅓ무 느렸다. 근데 진짜 이걸 그냥 쌩으로 내가 분류해서 정리하면 백퍼 쌩노가다인데다가 비효율적이고 시간도 엄청 오래걸릴게 뻔했다.

 

 

 최대한 효율적으로 하고 싶어서 머리를 굴렸는데, 아무리 엑셀 자동화 모듈을 사용한다고해도 이렇게 검색을 하는 프로그램은 검색하는데 시간이 걸릴 수 밖에 없기 때문에 속도 면에서는 메리트가 없었다.  계속 같은 위치에 무언가를 복사 붙여넣기를 한다거나 하는 반복적인 '문서작업'을 자동화로 해주는데에는 장점이 될 수 있겠지만, 이렇게 이미 존재하는 파일에 있는 data들 속에서 원하는 정보를 찾는 것은 프로그램으로 하기에는 무리일 것 같았다..

 

 

실행화면... 글쓰기 시작 전부터 실행시켜놨는데 아직도 하고있다...

 

 

 

암튼 그래서 내가 찾아낸건 바로 '슬라이서'이다!

 

 

이렇게 슬라이서가 알아서 이 sheet에 있는 모든 Vendor행의 목록들을 보여준다. 그럼 난 내가 찾고자 하는 Vendor에 체크를 하고, 슬라이서 하나를 더 생성해서

 

 

이렇게 Apple Vendor에 해당하는 Product도 선택하면 분류를 잘 할 수 있었다. 

 

어떻게 보면 이렇게 하는 것도 노가다를 완전히 안해도 되게 만들어주진 않았지만 그래도 훨씬 편하고 빠르게 작업을 할 수 있었다.