Excelシートで出力できるように

dislocker.py

@@ -6,8 +6,8 @@ from psycopg2 import sql
 import hashlib
 import string
 import random
-import csv
 from datetime import datetime
+from openpyxl import Workbook
 
 class DL():
     def __init__(self):
@@ -203,6 +203,7 @@ class Bot(discord.Client):
             csv_file_path = dislocker.export_dir_path + "pc_usage_history.csv"
             main_table = "pc_usage_history"
             related_table = "club_member"
+            excel_file_path = dislocker.export_dir_path + "pc_usage_history.xlsx"
         
             # メインテーブルの列情報を取得（user_idを除く）
             cursor.execute(sql.SQL("SELECT * FROM {} LIMIT 0").format(sql.Identifier(main_table)))
@@ -227,17 +228,36 @@ class Bot(discord.Client):
             
             rows = cursor.fetchall()
             
-            with open(csv_file_path, 'w', newline='', encoding='utf-8-sig') as csvfile:
+                    # Excelワークブックを作成
-                csvwriter = csv.writer(csvfile)
+            wb = Workbook()
-                csvwriter.writerow(column_names)
+            ws = wb.active
-                
+            
-                for row in rows:
+            # 列名を書き込み
-                    # nameを2番目に移動
+            ws.append(column_names)
-                    formatted_row = [self.format_datetime(row[0])] + [row[-1]] + [self.format_datetime(field) if field is not None else '' for field in row[1:-1]]
+            
-                    csvwriter.writerow(formatted_row)
+            # データを書き込み
-
+            for row in rows:
-            print(f"テーブル '{main_table}' の内容を '{csv_file_path}' に出力しました。")
+                # nameを2番目に移動
-            result = {"result": "ok", "file_path": csv_file_path}
+                formatted_row = [self.format_datetime(row[0])] + [row[-1]] + [self.format_datetime(field) if field is not None else '' for field in row[1:-1]]
+                ws.append(formatted_row)
+            
+            # 列幅を自動調整
+            for col in ws.columns:
+                max_length = 0
+                column = col[0].column_letter
+                for cell in col:
+                    try:
+                        if len(str(cell.value)) > max_length:
+                            max_length = len(str(cell.value))
+                    except:
+                        pass
+                adjusted_width = (max_length + 2) * 1.2
+                ws.column_dimensions[column].width = adjusted_width
+            
+            # Excelファイルを保存
+            wb.save(excel_file_path)
+            
+            print(f"テーブル '{main_table}' の内容を '{excel_file_path}' に出力しました。")
             
         except (Exception, psycopg2.Error) as error:
             print("使用履歴のエクスポート時にエラーが発生しました\nエラー内容\n", str(error))

temp/xcel.py

@@ -0,0 +1,97 @@
+import psycopg2
+import csv
+from psycopg2 import sql
+from datetime import datetime
+from openpyxl import Workbook
+from openpyxl.utils import get_column_letter
+
+def format_datetime(value):
+    if isinstance(value, datetime):
+        return value.strftime('%Y-%m-%d %H:%M:%S')
+    return value
+
+def export_table_to_excel(host, port, database, user, password, main_table, related_table, excel_file_path):
+    try:
+        conn = psycopg2.connect(
+            host=host,
+            port=port,
+            database=database,
+            user=user,
+            password=password
+        )
+        
+        cur = conn.cursor()
+        
+        # メインテーブルの列情報を取得（member_idを除く）
+        cur.execute(sql.SQL("SELECT * FROM {} LIMIT 0").format(sql.Identifier(main_table)))
+        main_columns = [desc[0] for desc in cur.description if desc[0] != 'member_id']
+        
+        # クエリを作成（列名を明確に指定）
+        query = sql.SQL("""
+            SELECT {main_columns}, {related_table}.name
+            FROM {main_table}
+            LEFT JOIN {related_table} ON {main_table}.member_id = {related_table}.id
+        """).format(
+            main_columns=sql.SQL(', ').join([sql.SQL("{}.{}").format(sql.Identifier(main_table), sql.Identifier(col)) for col in main_columns]),
+            main_table=sql.Identifier(main_table),
+            related_table=sql.Identifier(related_table)
+        )
+        
+        cur.execute(query)
+        
+        # 列名を再構成（nameを2番目に配置）
+        column_names = [main_columns[0], 'name'] + main_columns[1:]
+        
+        rows = cur.fetchall()
+        
+        # Excelワークブックを作成
+        wb = Workbook()
+        ws = wb.active
+        
+        # 列名を書き込み
+        ws.append(column_names)
+        
+        # データを書き込み
+        for row in rows:
+            # nameを2番目に移動
+            formatted_row = [format_datetime(row[0])] + [row[-1]] + [format_datetime(field) if field is not None else '' for field in row[1:-1]]
+            ws.append(formatted_row)
+        
+        # 列幅を自動調整
+        for col in ws.columns:
+            max_length = 0
+            column = col[0].column_letter
+            for cell in col:
+                try:
+                    if len(str(cell.value)) > max_length:
+                        max_length = len(str(cell.value))
+                except:
+                    pass
+            adjusted_width = (max_length + 2) * 1.2
+            ws.column_dimensions[column].width = adjusted_width
+        
+        # Excelファイルを保存
+        wb.save(excel_file_path)
+        
+        print(f"テーブル '{main_table}' の内容を '{excel_file_path}' に出力しました。")
+        
+    except (Exception, psycopg2.Error) as error:
+        print("エラーが発生しました:", error)
+    
+    finally:
+        if conn:
+            cur.close()
+            conn.close()
+            print("データベース接続を閉じました。")
+
+# 使用例
+host = "192.168.1.220"
+port = "12245"
+database = "dislocker"
+user = "suti7"
+password = "Testing1234"
+main_table = "pc_usage_history"
+related_table = "club_member"
+excel_file_path = "output.xlsx"
+
+export_table_to_excel(host, port, database, user, password, main_table, related_table, excel_file_path)