extract-sqlite 0.1.0__tar.gz → 0.1.1__tar.gz

{extract_sqlite-0.1.0 → extract_sqlite-0.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: extract-sqlite
-Version: 0.1.0
+Version: 0.1.1
 Summary: Add your description here
 Author: 6vi7ms
 Author-email: 6vi7ms <6vi7ms@gmail.com>

{extract_sqlite-0.1.0 → extract_sqlite-0.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "extract-sqlite"
-version = "0.1.0"
+version = "0.1.1"
 description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"

{extract_sqlite-0.1.0 → extract_sqlite-0.1.1}/pyproject.toml.orig

@@ -1,6 +1,6 @@
 [project]
 name = "extract-sqlite"
-version = "0.1.0"
+version = "0.1.1"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{extract_sqlite-0.1.0 → extract_sqlite-0.1.1}/src/extract_sqlite/main.py

@@ -1,16 +1,21 @@
 import sqlite3
+from pathlib import Path
 
 import click
 import pandas as pd
 
 
 @click.command()
-@click.option("--db", "-d", required=True, type=click.Path(exists=True), help="เส้นทางไปยังไฟล์ SQLite database")
-@click.option("--output", "-o", default="output.xlsx", help="ชื่อไฟล์ Excel ที่ต้องการส่งออก (ค่าเริ่มต้น: output.xlsx)")
+@click.argument("db", type=click.Path(exists=True))
+@click.option("--output", "-o", default=None, help="ชื่อไฟล์ Excel ที่ต้องการส่งออก (ค่าเริ่มต้น: ใช้ชื่อเดียวกับไฟล์ Database)")
 def main(db, output):
-    """ส่งออกข้อมูลทุกตารางจาก SQLite database ไปเป็นไฟล์ Excel โดยแยกตามชื่อ Sheet"""
-    click.echo(f"กำลังเชื่อมต่อกับฐานข้อมูล: {db}")
-    conn = sqlite3.connect(db)
+    db_path = Path(db)
+
+    if output is None:
+        output = str(db_path.with_suffix(".xlsx"))
+
+    click.echo(f"กำลังเชื่อมต่อกับฐานข้อมูล: {db_path}")
+    conn = sqlite3.connect(db_path)
 
     try:
         tables_query = "SELECT name FROM sqlite_master WHERE type='table' AND name NOT LIKE 'sqlite_%';"
@@ -25,7 +30,7 @@ def main(db, output):
         with pd.ExcelWriter(output, engine="openpyxl") as writer:
             for table in tables:
                 df = pd.read_sql(f"SELECT * FROM [{table}]", conn)
-                sheet_name = table[:31]  # Excel จำกัดชื่อ Sheet ที่ 31 ตัวอักษร
+                sheet_name = table[:31]
                 df.to_excel(writer, sheet_name=sheet_name, index=False)
                 click.echo(f" - ส่งออกตาราง '{table}' -> Sheet '{sheet_name}' ({len(df)} แถว)")