Kims12 commited on
Commit
244704f
·
verified ·
1 Parent(s): 5e76f66

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -4
app.py CHANGED
@@ -19,25 +19,31 @@ def analyze_reviews(file):
19
  end_date = datetime(current_year, 12, 31)
20
 
21
  logging.info(f"데이터 필터링: {start_date}부터 {end_date}까지.")
22
- # B열을 날짜로 변환
23
- df['B'] = pd.to_datetime(df.iloc[:, 1])
24
 
25
  # 최근 3년 데이터 필터링
26
  mask = (df['B'] >= start_date) & (df['B'] <= end_date)
27
  filtered_df = df.loc[mask]
28
  logging.info(f"필터링된 데이터 행 수: {len(filtered_df)}.")
29
 
 
 
 
 
30
  # 년월별 리뷰 건수 집계
31
  filtered_df['Year-Month'] = filtered_df['B'].dt.strftime('%Y-%m')
32
  review_counts = filtered_df.groupby('Year-Month').size().reset_index(name='리뷰건수')
33
  logging.info("년월별 리뷰 건수 집계 완료.")
34
 
35
  # 새로운 시트에 작성
36
- with pd.ExcelWriter(file.name, engine='openpyxl', mode='a') as writer:
 
 
37
  review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['A열', 'B열'])
38
  logging.info("새로운 시트 '월별 리뷰건수' 작성 완료.")
39
 
40
- return file.name
41
  except Exception as e:
42
  logging.error(f"오류 발생: {e}")
43
  return None
 
19
  end_date = datetime(current_year, 12, 31)
20
 
21
  logging.info(f"데이터 필터링: {start_date}부터 {end_date}까지.")
22
+ # B열을 날짜로 변환하고 timezone 정보 제거
23
+ df['B'] = pd.to_datetime(df.iloc[:, 1]).dt.tz_convert(None) if df.iloc[:, 1].dtype == 'datetime64[ns, UTC]' else pd.to_datetime(df.iloc[:, 1])
24
 
25
  # 최근 3년 데이터 필터링
26
  mask = (df['B'] >= start_date) & (df['B'] <= end_date)
27
  filtered_df = df.loc[mask]
28
  logging.info(f"필터링된 데이터 행 수: {len(filtered_df)}.")
29
 
30
+ if filtered_df.empty:
31
+ logging.warning("필터링된 데이터가 없습니다.")
32
+ return None
33
+
34
  # 년월별 리뷰 건수 집계
35
  filtered_df['Year-Month'] = filtered_df['B'].dt.strftime('%Y-%m')
36
  review_counts = filtered_df.groupby('Year-Month').size().reset_index(name='리뷰건수')
37
  logging.info("년월별 리뷰 건수 집계 완료.")
38
 
39
  # 새로운 시트에 작성
40
+ output_file = "분석_결과.xlsx"
41
+ with pd.ExcelWriter(output_file, engine='openpyxl') as writer:
42
+ df.to_excel(writer, index=False, sheet_name='원본 데이터') # 원본 데이터를 새 파일에 저장
43
  review_counts.to_excel(writer, sheet_name='월별 리뷰건수', index=False, header=['A열', 'B열'])
44
  logging.info("새로운 시트 '월별 리뷰건수' 작성 완료.")
45
 
46
+ return output_file
47
  except Exception as e:
48
  logging.error(f"오류 발생: {e}")
49
  return None