56 lines
1.9 KiB
Python
56 lines
1.9 KiB
Python
"""Import an already unpacked game from a folder on the server.
|
|
|
|
python -m app.import_dir /games/Some-Game [--title "Name"] [--owner admin]
|
|
|
|
The folder is only read; the converted copy goes to the blob store like an upload.
|
|
"""
|
|
|
|
import argparse
|
|
import asyncio
|
|
from pathlib import Path
|
|
|
|
from sqlalchemy import select
|
|
|
|
from app.db import SessionLocal
|
|
from app.models import Game, GameVersion, Role, User
|
|
from app.worker import convert
|
|
|
|
|
|
async def main() -> None:
|
|
ap = argparse.ArgumentParser()
|
|
ap.add_argument("path", type=Path)
|
|
ap.add_argument("--title")
|
|
ap.add_argument("--owner", help="username; defaults to the first admin")
|
|
args = ap.parse_args()
|
|
if not args.path.is_dir():
|
|
raise SystemExit(f"{args.path} is not a directory")
|
|
|
|
from app.converter.preflight import inspect_directory
|
|
from app.converter.compatibility import CompatibilityError
|
|
try:
|
|
report = await asyncio.to_thread(inspect_directory, args.path)
|
|
except CompatibilityError as exc:
|
|
raise SystemExit(str(exc)) from None
|
|
|
|
async with SessionLocal() as db:
|
|
stmt = select(User).join(Role).order_by(User.id)
|
|
stmt = stmt.where(User.username == args.owner) if args.owner else stmt.where(Role.name == "admin")
|
|
owner = await db.scalar(stmt.limit(1))
|
|
if owner is None:
|
|
raise SystemExit("owner not found")
|
|
game = Game(title=args.title or args.path.name, engine=report["engine"], created_by_id=owner.id)
|
|
db.add(game)
|
|
await db.flush()
|
|
version = GameVersion(game_id=game.id, owner_id=owner.id, name="Оригинал",
|
|
upload_name=args.path.name, status="importing")
|
|
db.add(version)
|
|
await db.commit()
|
|
game_id, version_id = game.id, version.id
|
|
|
|
ok = await convert(version_id, args.path, args.path.name)
|
|
print(f"game {game_id}, version {version_id}: {'ready' if ok else 'failed'}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|