update-case-drive 0.1.0__py3-none-any.whl

update_case_drive/main.py

@@ -0,0 +1,155 @@
+import ctypes
+import os
+
+import psutil
+import requests
+import typer
+
+app = typer.Typer()
+
+
+def get_windows_volume_name(drive_letter):
+    """ฟังก์ชันสำหรับดึง Volume Name (ชื่อไดรฟ์) จาก Windows อัตโนมัติ"""
+    volume_name_buffer = ctypes.create_unicode_buffer(261)
+    file_system_buffer = ctypes.create_unicode_buffer(261)
+    serial_number = ctypes.c_ulong(0)
+    max_component_len = ctypes.c_ulong(0)
+    file_system_flags = ctypes.c_ulong(0)
+
+    try:
+        rc = ctypes.windll.kernel32.GetVolumeInformationW(
+            ctypes.c_wchar_p(drive_letter),
+            volume_name_buffer,
+            ctypes.sizeof(volume_name_buffer),
+            ctypes.byref(serial_number),
+            ctypes.byref(max_component_len),
+            ctypes.byref(file_system_flags),
+            file_system_buffer,
+            ctypes.sizeof(file_system_buffer),
+        )
+        if rc:
+            return volume_name_buffer.value
+    except (OSError, ctypes.ArgumentError) as e:
+        print(f"Error occurred while fetching volume name for {drive_letter}: {e}")
+    return ""
+
+
+@app.command()
+def browse(
+    mode: str = typer.Option(
+        "online",
+        help="โหมดการเชื่อมต่อ: 'local' สำหรับ localhost หรือ 'online' สำหรับ server จริง",
+    ),
+):
+    """แสดงรายการไดรฟ์เฉพาะที่มีการตั้งชื่อ Volume Name (ยกเว้น Drive C)
+
+    ให้ผู้ใช้เลือก แล้วส่งข้อมูลไป Django API ทันที
+    """
+    # กำหนด URL ตามโหมดที่เลือก
+    if mode.lower() == "local":
+        api_url = "http://127.0.0.1:9747/drives/api/receive/"
+        typer.echo("🛠️ โหมด: Local (http://127.0.0.1:9747/)")
+    else:
+        api_url = "https://www.policeadmin.com/storage/drives/api/receive/"
+        typer.echo("🌐 โหมด: Online (https://www.policeadmin.com/)")
+
+    # 1. ดึงรายชื่อพาร์ทิชัน/ไดรฟ์ทั้งหมดในเครื่อง
+    partitions = psutil.disk_partitions(all=True)
+    drive_list = []
+
+    typer.echo("🖥️ รายชื่อไดรฟ์ที่มีการตั้งชื่อในระบบ (ไม่รวม Drive C):")
+    for partition in partitions:
+        mountpoint = partition.mountpoint
+
+        # ข้าม Drive C (รองรับทั้ง C:\ และรูปแบบอื่นๆ ที่ขึ้นต้นด้วย C)
+        if mountpoint.lower().startswith("c:") or "c:\\" in mountpoint.lower():
+            continue
+
+        # ดึงชื่อ Volume Name
+        vol_name = get_windows_volume_name(mountpoint)
+
+        # กรองเฉพาะไดรฟ์ที่มีการตั้งชื่อ (Volume Name) เท่านั้น ถ้าไม่มีให้ข้ามไป
+        if not vol_name:
+            continue
+
+        drive_list.append(mountpoint)
+        label_display = f" [ชื่อ: {vol_name}]"
+
+        # เช็คขนาดความจุ
+        size_info = ""
+        try:
+            usage = psutil.disk_usage(mountpoint)
+            total_gb = usage.total / (1024**3)
+            free_gb = usage.free / (1024**3)
+            size_info = f" (ความจุ: {total_gb:.2f} GB, ว่าง: {free_gb:.2f} GB)"
+        except OSError:
+            size_info = " (ไม่สามารถอ่านข้อมูลความจุได้)"
+
+        current_idx = len(drive_list) - 1
+        typer.echo(f"  [{current_idx}] {mountpoint}{label_display}{size_info}")
+
+    if not drive_list:
+        typer.echo("❌ ไม่พบไดรฟ์ที่มีการตั้งชื่อ Volume Name ในระบบ (นอกจาก Drive C)")
+        raise typer.Exit()
+
+    # 2. ให้ผู้ใช้เลือกหมายเลขไดรฟ์
+    selected_idx = typer.prompt("\nกรุณากรอกหมายเลขไดรฟ์ที่ต้องการเปิดดู", type=int)
+
+    if selected_idx < 0 or selected_idx >= len(drive_list):
+        typer.echo("❌ หมายเลขไดรฟ์ไม่ถูกต้อง")
+        raise typer.Exit()
+
+    selected_drive = drive_list[selected_idx]
+
+    # 3. ดึงชื่ออัตโนมัติจากไดรฟ์ที่เลือก
+    storage_name = get_windows_volume_name(selected_drive)
+
+    typer.echo(f"\n📂 กำลังเปิดดูและรวบรวมข้อมูลโฟลเดอร์ในไดรฟ์: {selected_drive} (ชื่อที่ใช้: {storage_name})\n")
+
+    # 4. อ่านขนาดความจุ และรายชื่อโฟลเดอร์ในไดรฟ์ที่เลือก
+    total_gb = 0.0
+    try:
+        usage = psutil.disk_usage(selected_drive)
+        total_gb = round(usage.total / (1024**3), 2)
+    except OSError:
+        typer.echo("⚠️ ไม่สามารถอ่านขนาดความจุของไดรฟ์ได้ (จะส่งขนาดเป็น 0.0)")
+
+    folders = []
+    try:
+        items = os.listdir(selected_drive)
+        folders = [os.path.join(selected_drive, item) for item in items if os.path.isdir(os.path.join(selected_drive, item))]
+
+        if not folders:
+            typer.echo("📁 ไม่พบโฟลเดอร์ย่อยในไดรฟ์นี้")
+        else:
+            typer.echo(f"พบโฟลเดอร์ทั้งหมด {len(folders)} โฟลเดอร์:")
+            for folder in folders:
+                typer.echo(f"  - 📁 {folder}")
+
+    except PermissionError:
+        typer.echo(f"❌ ไม่มีสิทธิ์เข้าถึงไดรฟ์ {selected_drive} (Permission Denied)")
+    except OSError as e:
+        typer.echo(f"❌ เกิดข้อผิดพลาดในการอ่านโฟลเดอร์: {e}")
+
+    # 5. เตรียม Payload ส่งข้อมูลไปยัง Django API
+    payload = {
+        "name": storage_name,  # ชื่อไดรฟ์ที่ดึงจาก Windows อัตโนมัติ (เช่น SC01)
+        "drive_path": selected_drive,  # เก็บ path จริงไว้ด้วย (เช่น D:\)
+        "size_gb": total_gb,
+        "folders": folders,
+    }
+
+    typer.echo(f"\n🚀 กำลังส่งข้อมูลไปยัง Django API ({api_url})...")
+    try:
+        response = requests.post(api_url, json=payload, timeout=10)
+        if response.status_code == 201:
+            typer.echo("✅ ส่งข้อมูลเข้า Django Database สำเร็จ!")
+            typer.echo(f"   Response: {response.json()}")
+        else:
+            typer.echo(f"⚠️ เกิดข้อผิดพลาดจาก Server (Status {response.status_code}): {response.text}")
+    except requests.exceptions.RequestException as e:
+        typer.echo(f"❌ ไม่สามารถเชื่อมต่อ API ได้: {e}")
+
+
+if __name__ == "__main__":
+    app()

update_case_drive-0.1.0.dist-info/METADATA

@@ -0,0 +1,12 @@
+Metadata-Version: 2.3
+Name: update-case-drive
+Version: 0.1.0
+Summary: Add your description here
+Author: 6vi7ms
+Author-email: 6vi7ms <6vi7ms@gmail.com>
+Requires-Dist: psutil>=7.2.2
+Requires-Dist: requests>=2.34.2
+Requires-Dist: typer>=0.27.3
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+

update_case_drive-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+update_case_drive/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+update_case_drive/main.py,sha256=Hq5NA_CpbQHfwvv7CNdmmQfKdHDFWfd8Bmf3lSz-35A,7555
+update_case_drive-0.1.0.dist-info/WHEEL,sha256=7hzKWg-J8I3Buqyw5tBii5z_MmAVsDYXXijd_QAtNZ8,81
+update_case_drive-0.1.0.dist-info/entry_points.txt,sha256=CuZJjb6LSoffsvj2zoD2vmoLxh6TIJaYNjiyk433ZdE,66
+update_case_drive-0.1.0.dist-info/METADATA,sha256=GlO3X4n_wSQbxndJk6bP3Lw9-MT-qWEzy_HIXKchuLc,306
+update_case_drive-0.1.0.dist-info/RECORD,,

update_case_drive-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: uv 0.12.18
+Root-Is-Purelib: true
+Tag: py3-none-any

update_case_drive-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+update-case-drive = update_case_drive.main:app
+