(add): Backup/Restore

api/app/system/backuprestore.py

@@ -0,0 +1,37 @@
+from data.baskets import Basket, BasketRepo
+from data.tickets import Ticket, TicketRepo
+from .prefixes import Prefix, PrefixRepo
+from dataclasses import dataclass, field
+from typing import List
+from fastapi import APIRouter, Header
+from .auth import AuthRepo
+
+chunk_size = 300
+
+@dataclass
+class BackupFile:
+  prefixes: List[Prefix] = field(default_factory=list)
+  baskets: List[Basket] = field(default_factory=list)
+  tickets: List[Ticket] = field(default_factory=list)
+
+@dataclass
+class ReturnMessage:
+  message: str
+
+backuprestore_router = APIRouter(prefix="/api/backuprestore")
+
+@backuprestore_router.get("")
+def get_backup_file(tam_key: str = Header("")):
+  AuthRepo().verify_key(tam_key)
+  return BackupFile(prefixes=PrefixRepo().get_all_prefixes(), baskets=BasketRepo().get_all_baskets(), tickets=TicketRepo().get_all_tickets())
+
+@backuprestore_router.post("")
+def post_backup_file(bf: BackupFile, tam_key: str = Header("")):
+  AuthRepo().verify_key(tam_key)
+  for i in range(0, len(bf.prefixes), chunk_size):
+    PrefixRepo().post_prefixes(bf.prefixes[i:i+chunk_size])
+  for i in range(0, len(bf.baskets), chunk_size):
+    BasketRepo().post_baskets(bf.baskets[i:i+chunk_size])
+  for i in range(0, len(bf.tickets), chunk_size):
+    TicketRepo().post_tickets(bf.tickets[i:i+chunk_size])
+  return ReturnMessage(message="Backup file imported successfully.")