create-case 0.1.2__tar.gz → 0.1.5__tar.gz

{create_case-0.1.2 → create_case-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: create-case
-Version: 0.1.2
+Version: 0.1.5
 Summary: 
 Author: 6vi7ms
 Author-email: 6vi7ms@gmail.com
@@ -10,4 +10,6 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Programming Language :: Python :: 3.14
 Description-Content-Type: text/markdown
 
-
+```sh
+create-case -h
+```

create_case-0.1.5/README.md

@@ -0,0 +1,3 @@
+```sh
+create-case -h
+```

{create_case-0.1.2 → create_case-0.1.5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "create-case"
-version = "0.1.2"
+version = "0.1.5"
 description = ""
 authors = [
     {name = "6vi7ms",email = "6vi7ms@gmail.com"}

create_case-0.1.5/src/create_case/main.py

@@ -0,0 +1,125 @@
+import os
+import re
+import argparse
+from datetime import datetime
+
+CASE_STRUCTURE = ["Pictures", "Readers", "Result", "Extractions", "Exports"]
+CASE_PATTERN = re.compile(r"^F-\d{4}-\d{3}$")
+
+
+def create_structure(base_path, case_name, evidence_count=0):
+    for folder in CASE_STRUCTURE:
+        folder_path = os.path.join(base_path, folder)
+        os.makedirs(folder_path, exist_ok=True)
+
+        # Result ไม่มี Evidence Folder
+        if folder == "Result":
+            continue
+
+        if evidence_count > 0:
+            for ev in range(1, evidence_count + 1):
+                ev_folder = f"{case_name} EV{ev:02d}"
+                os.makedirs(
+                    os.path.join(folder_path, ev_folder),
+                    exist_ok=True
+                )
+
+
+def print_structure(case_name, evidence_count=0):
+    print("\nFolder structure:")
+
+    for i, folder in enumerate(CASE_STRUCTURE):
+        prefix = "└─" if i == len(CASE_STRUCTURE) - 1 else "├─"
+        print(f"{prefix} {folder}")
+
+        if folder != "Result" and evidence_count > 0:
+            for ev in range(1, evidence_count + 1):
+                sub_prefix = (
+                    "   └─"
+                    if ev == evidence_count
+                    else "   ├─"
+                )
+                print(f"{sub_prefix} {case_name} EV{ev:02d}")
+
+
+def main():
+    try:
+        parser = argparse.ArgumentParser(
+            description="Create forensic case folder structure"
+        )
+
+        parser.add_argument(
+            "case",
+            nargs="?",
+            type=int,
+            help="Case number (e.g. 1 -> F-2026-001)"
+        )
+
+        parser.add_argument(
+            "-y",
+            "--year",
+            type=int,
+            default=datetime.now().year,
+            help="Case year"
+        )
+
+        parser.add_argument(
+            "-e",
+            "--evidence",
+            type=int,
+            default=0,
+            help="Number of evidence folders to create"
+        )
+
+        args = parser.parse_args()
+
+        if args.case is None:
+            # ใช้ Current Folder เป็น Case Folder
+            case_name = os.path.basename(os.getcwd())
+
+            if not CASE_PATTERN.match(case_name):
+                print(
+                    f"Error: Current folder '{case_name}' "
+                    f"is not a valid case folder."
+                )
+                return 1
+
+            base_path = os.getcwd()
+
+            create_structure(
+                base_path,
+                case_name,
+                args.evidence
+            )
+
+            print("\nUpdated structure in:")
+            print(f"  {os.path.abspath(base_path)}")
+
+        else:
+            case_name = f"F-{args.year}-{args.case:03d}"
+            base_path = os.path.abspath(case_name)
+
+            os.makedirs(base_path, exist_ok=True)
+
+            create_structure(
+                base_path,
+                case_name,
+                args.evidence
+            )
+
+            print("\nCreated case folder:")
+            print(f"  {base_path}")
+
+        print_structure(
+            case_name,
+            args.evidence
+        )
+
+        return 0
+
+    finally:
+        input("\nPress Enter to exit...")
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

create_case-0.1.2/src/create_case/main.py

@@ -1,100 +0,0 @@
-import os
-import string
-import argparse
-from datetime import datetime
-import getpass
-import ctypes
-
-
-CONFIG = {"local": ["Pictures", "Readers", "Exports"], "external": ["Extractions", "Readers"], "nas": ["Extractions", "Pictures", "Readers", "Exports"], "onedrive": ["Pictures", "Exports"]}
-
-
-def list_drives():
-    drives = []
-    for letter in string.ascii_uppercase:
-        drive = f"{letter}:\\"
-        if os.path.exists(drive):
-            drives.append(drive)
-    return drives
-
-
-def is_external_drive(drive_path):
-    DRIVE_REMOVABLE = 2
-    drive_type = ctypes.windll.kernel32.GetDriveTypeW(drive_path)
-    return drive_type == DRIVE_REMOVABLE
-
-
-def ask_for_drive():
-    drives = list_drives()
-    for idx, drv in enumerate(drives, 1):
-        print(f"{idx}. {drv}")
-
-    choice = input("Select drive: ")
-    try:
-        return drives[int(choice) - 1]
-    except:
-        print("Invalid choice")
-        return None
-
-
-def create_folders(base_path, folder_list):
-    for folder in folder_list:
-        path = os.path.join(base_path, folder)
-        try:
-            os.makedirs(path, exist_ok=True)
-        except Exception as e:
-            print(f"Failed to create {path}: {e}")
-
-
-def create_drive_case(case_folder):
-    drive = ask_for_drive()
-    if not drive:
-        return
-
-    if is_external_drive(drive):
-        folder_list = CONFIG["external"]
-    else:
-        folder_list = CONFIG["local"]
-
-    base_path = os.path.join(drive, case_folder)
-    create_folders(base_path, folder_list)
-
-
-def create_nas(year, case):
-    base = rf"\\192.168.9.10\Case Archive\Case-Forensic\{year}\F-{year}-{case}"
-    create_folders(base, CONFIG["nas"])
-
-
-def create_onedrive(year, case):
-    user = getpass.getuser()
-    base = rf"C:\Users\{user}\OneDrive\Documents\Forensic Reports\{year}\F-{year}-{case}"
-    create_folders(base, CONFIG["onedrive"])
-
-
-def main():
-    parser = argparse.ArgumentParser()
-    parser.add_argument("case", nargs="?", type=int, default=1)
-    parser.add_argument("-y", "--year", type=int, default=datetime.now().year)
-    args = parser.parse_args()
-
-    year = args.year
-    case = str(args.case).zfill(3)
-    case_folder_local = f"F-{year}-{case}"
-
-    print("1. OneDrive")
-    print("2. NAS")
-    print("3. External Drive Scan")
-    choice = input("Choose the location to create the case folder: ").strip()
-
-    if choice == "1":
-        create_onedrive(year, case)
-    elif choice == "2":
-        create_nas(year, case)
-    elif choice == "3":
-        create_drive_case(case_folder_local)
-    
-    print("Goodbye!")
-
-
-if __name__ == "__main__":
-    main()