1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
from typing import BinaryIO, Protocol
from abc import abstractmethod
from glob import glob
import json
from pathlib import Path
from . import SoloTool
class SessionManager(Protocol):
@abstractmethod
def getSessions(self) -> list[str]:
raise NotImplementedError
@abstractmethod
def loadSession(self, key: str, player=None) -> SoloTool:
raise NotImplementedError
@abstractmethod
def saveSession(self, soloTool: SoloTool, key: str) -> None:
raise NotImplementedError
@abstractmethod
def addSong(self, name: str, content: BinaryIO) -> None:
raise NotImplementedError
class _FileSystemSessionManager(SessionManager):
def __init__(self, songPool: str, sessionPath: str):
self._songPool = Path(songPool)
self._sessionPath = Path(sessionPath)
def getSessions(self) -> list[str]:
return [Path(f).stem for f in glob(f"{self._sessionPath}/*.json")]
def loadSession(self, key: str, player=None) -> SoloTool:
with open(self._sessionPath / f"{key}.json", "r") as f:
session = json.load(f)
st = SoloTool(self._songPool, player=player)
for i, entry in enumerate(session):
songPath = entry["path"]
keyPoints = entry["key_points"]
st.addSong(songPath)
st._keyPoints[i] = keyPoints
return st
def saveSession(self, soloTool: SoloTool, key: str) -> None:
session = []
for i, song in enumerate(soloTool.songs):
entry = {
"path": song,
"key_points" : soloTool._keyPoints[i]
}
session.append(entry)
with open(self._sessionPath / f"{key}.json", "w") as f:
json.dump(session, f)
def addSong(self, name: str, content: BinaryIO) -> None:
from shutil import copyfileobj
newSong = self._songPool / name
with open(newSong, "wb") as f:
copyfileobj(content, f)
class _FileBrowserSessionManager(SessionManager):
def __init__(self, songPool: str, sessionPath: str):
pass
def getSessions(self) -> list[str]:
pass
def loadSession(self, key: str, player=None) -> SoloTool:
pass
def saveSession(self, soloTool: SoloTool, key: str) -> None:
pass
def addSong(self, name: str, content: BinaryIO) -> None:
pass
def getSessionManager(songPool: str, sessionPath: str) -> SessionManager:
from re import search
match = search(r"^([a-z0-9]+://)", sessionPath)
if not match or match.group(0) == "file://":
return _FileSystemSessionManager(songPool, sessionPath)
elif match.group(0) in ["http://", "https://"]:
return _FileBrowserSessionManager(songPool, sessionPath)
|
