simple-image-hosting 0.1.1__py3-none-any.whl

simple_image/__init__.py

@@ -0,0 +1,3 @@
+from .main import app, create_app
+
+__all__ = ["app", "create_app"]

simple_image/cli.py

@@ -0,0 +1,171 @@
+import argparse
+import getpass
+import os
+import sys
+from pathlib import Path
+
+import uvicorn
+
+from .main import create_app, hash_password
+from .models import DEFAULT_COMPRESS_QUALITY, User, create_session_factory
+
+
+def build_parser() -> argparse.ArgumentParser:
+    parser = argparse.ArgumentParser(prog="simple-image", description="Simple Image server CLI")
+    subparsers = parser.add_subparsers(dest="command", required=True)
+
+    serve = subparsers.add_parser("serve", help="Run HTTP server")
+    serve.add_argument("data_dir", type=Path, help="Runtime data directory (DB + images)")
+    serve.add_argument("--host", default="0.0.0.0", help="Bind host")
+    serve.add_argument("--port", type=int, default=8000, help="Bind port")
+    serve.add_argument("--reload", action="store_true", help="Enable auto reload")
+    serve.add_argument("--api-url", default=None, help="Public API base URL")
+    serve.add_argument("--admin-username", default=None, help="Bootstrap admin username")
+    serve.add_argument("--admin-password", default=None, help="Bootstrap admin password")
+    serve.add_argument(
+        "--database-url",
+        default=None,
+        help="Database URL, e.g. mysql+pymysql://user:pass@host:3306/simple_image",
+    )
+    serve.add_argument(
+        "--compress-quality",
+        type=int,
+        default=None,
+        help="Default upload compress quality (1-95)",
+    )
+    serve.add_argument(
+        "--base-path",
+        default=None,
+        help="Deploy under sub path, e.g. /simple_image",
+    )
+    serve.add_argument(
+        "-d", "--daemon", action="store_true",
+        help="Run server as a background daemon",
+    )
+
+    reset_pwd = subparsers.add_parser(
+        "reset-admin-password", help="Reset admin password"
+    )
+    reset_pwd.add_argument(
+        "data_dir", type=Path, help="Runtime data directory (same as used by serve)"
+    )
+    reset_pwd.add_argument(
+        "--username", default=None,
+        help="Admin username to reset (omit to auto-select the only admin)"
+    )
+    reset_pwd.add_argument(
+        "--database-url", default=None,
+        help="Database URL (omit to use data_dir/database.db)"
+    )
+
+    return parser
+
+
+def _daemonize(data_dir: Path) -> None:
+    """Double-fork to detach from the controlling terminal."""
+    data_dir.mkdir(parents=True, exist_ok=True)
+    pid_file = data_dir / "simple-image.pid"
+    log_file = data_dir / "simple-image.log"
+
+    # First fork – exit parent
+    if os.fork() > 0:
+        raise SystemExit(0)
+
+    os.setsid()
+
+    # Second fork – prevent re-acquiring a terminal
+    if os.fork() > 0:
+        raise SystemExit(0)
+
+    # Redirect stdio to log file
+    log_fd = os.open(str(log_file), os.O_WRONLY | os.O_CREAT | os.O_APPEND, 0o644)
+    os.dup2(log_fd, sys.stdout.fileno())
+    os.dup2(log_fd, sys.stderr.fileno())
+    os.close(log_fd)
+
+    # Write PID file
+    pid_file.write_text(str(os.getpid()))
+
+
+def run_serve(args: argparse.Namespace) -> None:
+    if args.daemon:
+        _daemonize(args.data_dir)
+
+    app = create_app(
+        data_dir=args.data_dir,
+        api_url=args.api_url,
+        admin_username=args.admin_username,
+        admin_password=args.admin_password,
+        default_compress_quality=args.compress_quality,
+        database_url=args.database_url,
+        base_path=args.base_path,
+    )
+    uvicorn.run(app, host=args.host, port=args.port, reload=args.reload)
+
+
+def run_reset_admin_password(args: argparse.Namespace) -> None:
+    db_path = args.data_dir / "database.db"
+    database_url = args.database_url or os.getenv("SIMPLE_IMAGE_DATABASE_URL") or os.getenv("DATABASE_URL")
+    if not database_url:
+        if not db_path.exists():
+            print(f"Error: database not found at {db_path}", file=sys.stderr)
+            raise SystemExit(1)
+        database_url = f"sqlite:///{db_path}"
+
+    use_db_path = None if args.database_url else db_path
+    session_factory = create_session_factory(
+        default_compress_quality=DEFAULT_COMPRESS_QUALITY,
+        database_url=database_url,
+        db_path=use_db_path,
+    )
+    db = session_factory()
+    try:
+        admins = db.query(User).filter(User.is_admin == True).all()
+
+        if not admins:
+            print("Error: no admin user found in the database.", file=sys.stderr)
+            raise SystemExit(1)
+
+        if args.username:
+            target = next((u for u in admins if u.username == args.username), None)
+            if not target:
+                available = ", ".join(u.username for u in admins)
+                print(f"Error: admin '{args.username}' not found. Existing admins: {available}", file=sys.stderr)
+                raise SystemExit(1)
+        elif len(admins) == 1:
+            target = admins[0]
+            print(f"Found admin user: {target.username}")
+        else:
+            print("Multiple admin users found, please specify --username:")
+            for u in admins:
+                print(f"  - {u.username}")
+            raise SystemExit(1)
+
+        new_password = getpass.getpass(f"New password for '{target.username}': ")
+        if not new_password:
+            print("Error: password cannot be empty.", file=sys.stderr)
+            raise SystemExit(1)
+        confirm = getpass.getpass("Confirm new password: ")
+        if new_password != confirm:
+            print("Error: passwords do not match.", file=sys.stderr)
+            raise SystemExit(1)
+
+        target.password_hash = hash_password(new_password)
+        db.commit()
+        print(f"Password for admin '{target.username}' has been reset successfully.")
+    finally:
+        db.close()
+
+
+def main() -> None:
+    parser = build_parser()
+    args = parser.parse_args()
+
+    if args.command == "serve":
+        run_serve(args)
+    elif args.command == "reset-admin-password":
+        run_reset_admin_password(args)
+
+
+if __name__ == "__main__":
+    main()