added scn migrations

shvatka/core/migration_utils/from_1_to_2/migrators.py

@@ -0,0 +1,79 @@
+from shvatka.core.migration_utils import models_0
+from shvatka.core.models.dto import scn, action
+
+
+def bonus_key_0_to_1(bonus_key: models_0.BonusKey) -> action.BonusKey:
+    return action.BonusKey(text=bonus_key.text, bonus_minutes=bonus_key.bonus_minutes)
+
+
+def hint_0_to_1(hint: models_0.AnyHint) -> scn.AnyHint:
+    match hint:
+        case models_0.TextHint(text=text):
+            return scn.TextHint(text=text)
+        case models_0.GPSHint(latitude=latitude, longitude=longitude):
+            return scn.GPSHint(latitude=latitude, longitude=longitude)
+        case models_0.VenueHint as vh:
+            return scn.VenueHint(
+                latitude=vh.latitude,
+                longitude=vh.longitude,
+                title=vh.title,
+                address=vh.address,
+                foursquare_id=vh.foursquare_id,
+                foursquare_type=vh.foursquare_type,
+            )
+        case models_0.PhotoHint(file_guid=file_guid, caption=caption):
+            return scn.PhotoHint(file_guid=file_guid, caption=caption)
+        case models_0.AudioHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid):
+            return scn.AudioHint(file_guid=file_guid, thumb_guid=thumb_guid, caption=caption)
+        case models_0.VideoHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid):
+            return scn.VideoHint(file_guid=file_guid, thumb_guid=thumb_guid, caption=caption)
+        case models_0.DocumentHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid):
+            return scn.DocumentHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid)
+        case models_0.AnimationHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid):
+            return scn.AnimationHint(file_guid=file_guid, caption=caption, thumb_guid=thumb_guid)
+        case models_0.VoiceHint(file_guid=file_guid, caption=caption):
+            return scn.VoiceHint(file_guid=file_guid, caption=caption)
+        case models_0.VideoNoteHint(file_guid=guid):
+            return scn.VideoNoteHint(file_guid=guid)
+        case models_0.ContactHint as ch:
+            return scn.ContactHint(
+                phone_number=ch.phone_number,
+                first_name=ch.first_name,
+                last_name=ch.last_name,
+                vcard=ch.vcard,
+            )
+        case models_0.StickerHint(file_guid=guid):
+            return scn.StickerHint(file_guid=guid)
+        case _:
+            raise RuntimeError("unknown hint type")
+
+
+def time_hint_0_to_1(time_hint: models_0.TimeHint) -> scn.TimeHint:
+    return scn.TimeHint(
+        time=time_hint.time,
+        hint=[hint_0_to_1(hint) for hint in time_hint.hint],
+    )
+
+
+def hints_0_to_1(hints: models_0.HintsList) -> scn.HintsList:
+    return scn.HintsList([time_hint_0_to_1(hint) for hint in hints])
+
+
+def level_0_to_1(level: models_0.LevelScenario) -> scn.LevelScenario:
+    conditions: list[action.AnyCondition] = [action.KeyWinCondition(set(level.keys))]
+    if level.bonus_keys:
+        conditions.append(
+            action.KeyBonusCondition({bonus_key_0_to_1(b) for b in level.bonus_keys})
+        )
+    return scn.LevelScenario(
+        id=level.id,
+        time_hints=hints_0_to_1(level.time_hints),
+        conditions=scn.Conditions(conditions),
+        __model_version__=1,
+    )
+
+
+def game_0_to_1(game: models_0.GameScenario) -> scn.GameScenario:
+    return scn.GameScenario(
+        name=game.name, levels=[level_0_to_1(lvl) for lvl in game.levels], __model_version__=1
+    )

shvatka/core/migration_utils/models_0/__init__.py

@@ -0,0 +1,39 @@
+from .file_content import (
+    FileMeta,
+    FileContentLink,
+    TgLink,
+    ParsedTgLink,
+    SavedFileMeta,
+    StoredFileMeta,
+    FileMetaLightweight,
+    UploadedFileMeta,
+    VerifiableFileMeta,
+)
+from .game import (
+    GameScenario,
+    FullGameScenario,
+    ParsedGameScenario,
+    ParsedCompletedGameScenario,
+    RawGameScenario,
+    UploadedGameScenario,
+)
+from .hint_part import (
+    AnyHint,
+    BaseHint,
+    FileMixin,
+    TextHint,
+    GPSHint,
+    VenueHint,
+    AudioHint,
+    VideoHint,
+    DocumentHint,
+    AnimationHint,
+    VoiceHint,
+    VideoNoteHint,
+    StickerHint,
+    PhotoHint,
+    ContactHint,
+)
+from .level import LevelScenario, SHKey, BonusKey, HintsList
+from .parsed_zip import ParsedZip
+from .time_hint import TimeHint

shvatka/core/migration_utils/models_0/file_content.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+from dataclasses import dataclass, field
+
+from shvatka.core.models import dto
+from shvatka.core.models.enums.hint_type import HintType
+
+
+@dataclass
+class TgLink:
+    file_id: str
+    """telegram file_id"""
+    content_type: HintType
+    """type of content"""
+
+
+@dataclass
+class FileContentLink:
+    file_path: str
+    """path to file in file system"""
+
+
+@dataclass
+class FileMetaLightweight:
+    guid: str
+    """GUID for filename in file storage, DB and in archive"""
+    original_filename: str
+    """Filename from user before renamed to guid"""
+    extension: str
+    """extension with leading dot: ".zip" ".tar.gz" etc"""
+    content_type: HintType | None = field(kw_only=True, default=None)
+    """type of content"""
+
+    @property
+    def local_file_name(self):
+        return self.guid + (self.extension or "")
+
+    @property
+    def public_filename(self):
+        return self.original_filename + (self.extension or "")
+
+
+@dataclass
+class StoredFileMeta(FileMetaLightweight):
+    file_content_link: FileContentLink
+
+
+@dataclass
+class UploadedFileMeta(FileMetaLightweight):
+    tg_link: TgLink | None = None
+
+
+@dataclass
+class FileMeta(StoredFileMeta):
+    tg_link: TgLink
+
+
+@dataclass
+class VerifiableFileMeta(FileMeta):
+    author_id: int
+
+
+@dataclass
+class SavedFileMeta(VerifiableFileMeta):
+    id: int
+    author: dto.Player
+
+
+@dataclass
+class ParsedTgLink(TgLink):
+    filename: str | None = None

shvatka/core/migration_utils/models_0/game.py

@@ -0,0 +1,47 @@
+from dataclasses import dataclass
+from datetime import datetime
+from typing import BinaryIO, Sequence, Literal
+
+from shvatka.core.models import enums
+from shvatka.core.models.dto.export_stat import GameStat
+from . import UploadedFileMeta, FileMetaLightweight
+from .file_content import FileMeta
+from .level import LevelScenario
+
+
+@dataclass
+class GameScenario:
+    name: str
+    levels: list[LevelScenario]
+    __model_version__: Literal[0]
+
+
+@dataclass
+class FullGameScenario(GameScenario):
+    files: Sequence[FileMeta]
+
+
+@dataclass
+class UploadedGameScenario(GameScenario):
+    files: list[UploadedFileMeta]
+
+
+@dataclass
+class ParsedGameScenario(GameScenario):
+    files: list[FileMetaLightweight]
+
+
+@dataclass
+class ParsedCompletedGameScenario(ParsedGameScenario):
+    id: int
+    start_at: datetime
+    files_contents: dict[str, BinaryIO]
+    stat: GameStat
+    status: enums.GameStatus = enums.GameStatus.complete
+
+
+@dataclass
+class RawGameScenario:
+    scn: dict
+    files: dict[str, BinaryIO]
+    stat: dict | None = None