mirror of
https://github.com/olivierkes/manuskript.git
synced 2024-06-01 10:39:34 +12:00
106 lines
2.4 KiB
Python
106 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
# --!-- coding: utf8 --!--
|
|
|
|
import collections
|
|
import os
|
|
|
|
from manuskript.io.textFile import TextFile
|
|
|
|
|
|
class Status:
|
|
|
|
def __init__(self, host, index: int, name: str):
|
|
self.host = host
|
|
self.ID = index
|
|
|
|
self.name = name
|
|
|
|
def __str__(self):
|
|
return self.name
|
|
|
|
def load(self):
|
|
self.host.load()
|
|
|
|
def save(self):
|
|
self.host.save()
|
|
|
|
|
|
class StatusHost:
|
|
|
|
def __init__(self, path):
|
|
self.file = TextFile(os.path.join(path, "status.txt"))
|
|
self.statuses = collections.OrderedDict()
|
|
|
|
def addStatus(self, name: str = None) -> Status:
|
|
if name is None:
|
|
name = "New Status"
|
|
|
|
status = Status(self, 1+len(self.statuses), name)
|
|
self.statuses[name] = status
|
|
return status
|
|
|
|
def removeStatus(self, name: str) -> Status:
|
|
return self.statuses.pop(name)
|
|
|
|
def renameStatus(self, oldName: str, newName: str):
|
|
match = False
|
|
drop = [oldName]
|
|
|
|
for name, status in self.statuses.items():
|
|
if match:
|
|
drop.append(name)
|
|
elif name == oldName:
|
|
status.name = newName
|
|
match = True
|
|
|
|
if not match:
|
|
return
|
|
|
|
statuses = list()
|
|
for name in drop:
|
|
statuses.append(self.statuses.pop(name))
|
|
|
|
for status in statuses:
|
|
self.statuses[status.name] = status
|
|
|
|
def getStatus(self, name: str) -> Status:
|
|
return self.statuses.get(name)
|
|
|
|
def getStatusByID(self, ID: int | None) -> Status | None:
|
|
if ID is None:
|
|
return None
|
|
|
|
index = 1
|
|
for status in self.statuses.values():
|
|
if index == ID:
|
|
assert status.ID == ID
|
|
return status
|
|
|
|
index += 1
|
|
|
|
return None
|
|
|
|
def __iter__(self):
|
|
return self.statuses.values().__iter__()
|
|
|
|
def load(self):
|
|
try:
|
|
text = self.file.load()
|
|
self.statuses.clear()
|
|
except FileNotFoundError:
|
|
self.statuses.clear()
|
|
return
|
|
|
|
if (len(text) <= 1) or (text[len(text) - 1] != "\n"):
|
|
return
|
|
|
|
text = text[:-1]
|
|
if len(text) <= 0:
|
|
return
|
|
|
|
for name in text.split("\n"):
|
|
self.addStatus(name)
|
|
|
|
def save(self):
|
|
self.file.save("\n".join(self.statuses.keys()) + "\n")
|