[QT][Python] 함수와 QT 연결하기 (4)

#1. 파일 찾아보기 버튼 - fl_btn
#2. 파일 경로 나타나는 버튼 - fl_lbl
#3. 분석 기능 지정 드롭다운 버튼 - fun_cb
        # 사용자별 대화 빈도
        # 사용자별 이모티콘 사용 빈도
        # 사용자별 활동 시간대 
        # 사용자별 평균 답변 시간
#4. 시간대별 활동 빈도 분석 참여자 선택 텍스트 입력 박스 - filter_te
#5. 분석 결과 이미지 다운로드 버튼 - down_btn
#6. 분석(Start) Run 버튼 - st_btn
#7. 초기화(Reset) Run 버튼 - rst_btn

#8. 결과 화면 - ptr_lbl

 

이제 마지막으로 남은 버튼은 4번과 5번이다.

아무래도 텍스트를 입력하고 이미지를 다운로드 하는 버튼이 가장 어렵다고 느껴졌기 때문에 마지막으로 작업을 했던 것 같다. (하지만 가장 어려운 작업은 아니었던 것 같다.)

---

#4. 시간대별 활동 빈도 분석 참여자 선택 텍스트 입력 박스 - filter_te

텍스트 박스에 이름을 입력하여 참여자별로 시간대별 활동 빈도 시간에 대한 파이차트를 출력하는 것은 하나의 기능에만 해당되는 것이다.

그래서 다양한 방법을 고민하다가 그냥 다른 기능들을 선택하고 분석 버튼을 누를 때 참여자 필터링에 입력값이 있든 없든 상관없이 원래 값이 출력되도록 한다.

하지만 '시간대별 활동 빈도' 기능을 선택하고 이름값을 정확하게 입력하면 제대로된 필터링이 되도록 코드를 짜는 것으로 계획했다.

대신 이름이 틀리거나 빈칸으로 [분석] 버튼을 클릭하면 전체 파이값이 출력되도록 설정했다.

기존의 함수를 정의한 py파일에서도 전체 출력과 개별 출력 값이 모두 나오도록 설정되어있기 때문에 해당 함수를 그대로 활용하면 된다.

 

우선 기존에 작성했던 "사용자별 각 시간대 활동 빈도 분석"의 함수에서  name 필터링 다음에 파이 차트를 출력하는 값이 조금 변경되었다.

    # 파이 차트 생성
    if not pivot_table.empty:
        fig, ax = plt.subplots()
        
        pivot_table.iloc[0].plot(kind='pie', ax=ax, autopct='%1.1f%%', startangle=90, legend=False)
        ax.set_title(f"{names[0]}의 시간대별 활동 빈도")
        ax.set_ylabel('')
        plt.tight_layout()
        
    else:
        pivot_table = activity_by_time.pivot(index='Name', columns='Time_Parts', values='Count_Messages').fillna(0)

        fig, axes = plt.subplots(nrows=1, ncols=len(pivot_table), figsize=(15, 5))
    
        for i, (names, data) in enumerate(pivot_table.iterrows()):
            data.plot(kind='pie', ax=axes[i], autopct='%1.1f%%', startangle=90, legend=False)
            axes[i].set_title(names)
            axes[i].set_ylabel('')
        plt.tight_layout()

원래는 if not 값과 else 값의 순서가 달랐다.

pivot_table 값은 아래와 같은데 

pivot_table = user_data.pivot(index='Name', columns='Time_Parts', values='Count_Messages').fillna(0)

위에 입력된 names 값이 인덱스로 들어가서 값을 출력하는 것을 pivot_table에 넣은 것이다.

만약 pivot_table 값이 비어있지 않다면 Name 값이 입력된 것이기 때문에 개별로 출력되는 코드를 작성했지만 비어있거나 제대로 입력되지 않았다면 다시 pivot_table을 정의하여 반복문에 의해 모든 값이 출력되도록 작성하였다.

 

이 함수를 활용하는 코드는 따로 qt 파일에서 추가할 건 없고 코드 몇 줄만 추가하면 된다.

우선 새로 실행할 때마다 객체가 초기화되어야 하기 때문에 __init__ 함수에 아래의 코드를 추가한다.

        self.filter_te = self.findChild(QTextEdit, 'filter_te')

 

 

그리고 분석을 실행하는 버튼인 [st_btn]을 위해 작성한 함수 start_analyze_data에 filter_te에 작성된 텍스트를 가져와서 입력하는 코드를 작성해준다.

        names = self.filter_te.toPlainText()
        names = names.split()

해당 코드를 추가해주면 start_analyze_data 함수에서 완성되는 코드 값은 아래와 같다.

        if self.selected_value == '사용자별 대화 빈도':
            self.fig = count_messages_by_name(self.refined_data)
            self.plot_pixmap()
        elif self.selected_value == '사용자별 이모티콘 사용 빈도':
            self.fig = count_emoticon_usage(self.refined_data, '이모티콘')
            self.plot_pixmap()
        elif self.selected_value == '사용자별 평균 답변 시간':
            self.fig = calculate_response_time(self.refined_data)
            self.plot_pixmap()
        elif self.selected_value == '사용자별 활동 시간대':
            self.fig = analyze_activity_by_time(self.refined_data, names)
            self.plot_pixmap()

names 값을 정의해줬기 때문에 '사용자별 활동 시간대' 함수에서 names 를 활용할 수 있다. 

그리고 다른 기능 값에서는 정의 안되어있기 때문에 names 값에 영향을 받지 않는다.

 

마지막으로 결과 화면을 리셋할 때 filter_te에 작성된 값도 함께 리셋되어야 하기 때문에 reset_canvas 함수에 

        self.filter_te.clear()

해당 코드를 추가해준다.

 

이렇게하면 필터링 기능에 대한 함수 연결은 완성된다.

 

#5. 분석 결과 이미지 다운로드 버튼 - down_btn

출력 결과 값을 GUI에서 확인하고 이미지로도 다운받을 수 있는 함수를 작성한다.함수명은 download_image 인데 사실 해당 부분은 작성을 아예 몰라서 gpt가 작성해준대로 적용했더니 되었다.

    def download_image(self):
        file_path, _ = QFileDialog.getSaveFileName(self, "이미지 저장", "", "PNG Files (*.png);;All Files (*)")
        if file_path:
            self.fig.savefig(file_path, format='png')

• QFileDialog.getSaveFileName은 사용자가 파일을 저장할 위치와 이름을 선택할 수 있도록 파일 저장 대화 상자를 표시한다.
• "이미지 저장" : 파일 대화 상자 제목으로 사용된다.
• " PNG Files (*.png);;All Files (*)" : 파일 형식 필터를 지정한다. 여기서는 PNG 파일만 선택할 수 있도록 하고, 모든 파일 형식도 선택할 수 있게 한다.
• if file_path를 작성하여 사용자가 파일 경로를 선택했다면 조건문이 실행되도록 설정한다.
• file_path로 지정된 경로에 이미지를 PNG 형식으로 저장한다. 

 

그리고 마지막으로 해당 함수를 qt의 객체인 down_btn 위젯에 연결해준다.

(__init__ 안에)

        self.down_btn.clicked.connect(self.download_image)

 

이렇게하면 모든 함수 입력은 끝난다. 

 

입력한 함수와 메인 선언 함수를 함께 실행하면 동작하는 GUI를 확인할 수 있다.

 

마지막 게시글에 동작하는 결과물 영상을 첨부할 예정이다.