2021-05-09 11:15:14 +12:00
|
|
|
#!/usr/bin/env python
|
|
|
|
# --!-- coding: utf8 --!--
|
|
|
|
|
2022-11-19 11:32:08 +13:00
|
|
|
import collections
|
2021-05-09 11:15:14 +12:00
|
|
|
import os
|
|
|
|
|
2023-02-05 04:11:18 +13:00
|
|
|
from manuskript.data.abstractData import AbstractData
|
2021-05-09 11:15:14 +12:00
|
|
|
from manuskript.data.color import Color
|
|
|
|
from manuskript.io.mmdFile import MmdFile
|
|
|
|
|
|
|
|
|
|
|
|
class Label:
|
|
|
|
|
2022-11-20 02:59:51 +13:00
|
|
|
def __init__(self, host, index: int, name: str, color: Color):
|
2021-05-09 11:15:14 +12:00
|
|
|
self.host = host
|
2022-11-20 02:59:51 +13:00
|
|
|
self.ID = index
|
2021-05-09 11:15:14 +12:00
|
|
|
|
|
|
|
self.name = name
|
|
|
|
self.color = color
|
|
|
|
|
|
|
|
def __str__(self):
|
|
|
|
return self.name
|
|
|
|
|
|
|
|
def load(self):
|
|
|
|
self.host.load()
|
|
|
|
|
|
|
|
def save(self):
|
|
|
|
self.host.save()
|
|
|
|
|
|
|
|
|
2023-02-05 04:11:18 +13:00
|
|
|
class LabelHost(AbstractData):
|
2021-05-09 11:15:14 +12:00
|
|
|
|
|
|
|
def __init__(self, path):
|
2023-02-05 04:11:18 +13:00
|
|
|
AbstractData.__init__(self, os.path.join(path, "labels.txt"))
|
|
|
|
self.file = MmdFile(self.dataPath, 21)
|
2022-11-19 11:32:08 +13:00
|
|
|
self.labels = collections.OrderedDict()
|
2021-05-09 11:15:14 +12:00
|
|
|
|
2023-03-22 11:37:17 +13:00
|
|
|
def changePath(self, path: str):
|
|
|
|
AbstractData.changePath(self, os.path.join(path, "labels.txt"))
|
|
|
|
self.file = MmdFile(self.dataPath, 21)
|
|
|
|
|
2022-11-20 02:36:57 +13:00
|
|
|
def addLabel(self, name: str = None, color: Color = None) -> Label:
|
2022-11-19 11:32:08 +13:00
|
|
|
if name is None:
|
|
|
|
name = "New Label"
|
|
|
|
|
2022-11-20 02:59:51 +13:00
|
|
|
label = Label(self, 1+len(self.labels), name, color)
|
2022-11-19 11:32:08 +13:00
|
|
|
self.labels[name] = label
|
|
|
|
return label
|
2021-05-09 11:15:14 +12:00
|
|
|
|
2022-11-20 02:36:57 +13:00
|
|
|
def removeLabel(self, name: str) -> Label:
|
|
|
|
return self.labels.pop(name)
|
2021-05-09 11:15:14 +12:00
|
|
|
|
|
|
|
def renameLabel(self, oldName: str, newName: str):
|
2022-11-20 02:36:57 +13:00
|
|
|
match = False
|
|
|
|
drop = [oldName]
|
2021-05-09 11:15:14 +12:00
|
|
|
|
2022-11-20 02:36:57 +13:00
|
|
|
for name, label in self.labels.items():
|
|
|
|
if match:
|
|
|
|
drop.append(name)
|
|
|
|
elif name == oldName:
|
|
|
|
label.name = newName
|
|
|
|
match = True
|
|
|
|
|
|
|
|
if not match:
|
|
|
|
return
|
|
|
|
|
|
|
|
labels = list()
|
|
|
|
for name in drop:
|
|
|
|
labels.append(self.labels.pop(name))
|
|
|
|
|
|
|
|
for label in labels:
|
|
|
|
self.labels[label.name] = label
|
|
|
|
|
|
|
|
def getLabel(self, name: str) -> Label:
|
2021-05-09 11:15:14 +12:00
|
|
|
return self.labels.get(name)
|
|
|
|
|
2022-11-20 02:59:51 +13:00
|
|
|
def getLabelByID(self, ID: int | None) -> Label | None:
|
|
|
|
if ID is None:
|
|
|
|
return None
|
2022-11-20 02:36:57 +13:00
|
|
|
|
2022-11-20 02:59:51 +13:00
|
|
|
index = 1
|
2022-11-20 02:36:57 +13:00
|
|
|
for label in self.labels.values():
|
2022-11-20 02:59:51 +13:00
|
|
|
if index == ID:
|
|
|
|
assert label.ID == ID
|
2022-11-20 02:36:57 +13:00
|
|
|
return label
|
|
|
|
|
|
|
|
index += 1
|
|
|
|
|
|
|
|
return None
|
|
|
|
|
2021-05-09 11:15:14 +12:00
|
|
|
def __iter__(self):
|
|
|
|
return self.labels.values().__iter__()
|
|
|
|
|
|
|
|
def load(self):
|
2023-02-05 04:11:18 +13:00
|
|
|
self.labels.clear()
|
|
|
|
AbstractData.load(self)
|
|
|
|
|
2021-05-09 11:15:14 +12:00
|
|
|
try:
|
|
|
|
metadata, _ = self.file.loadMMD(True)
|
|
|
|
except FileNotFoundError:
|
2023-02-05 04:11:18 +13:00
|
|
|
self.complete(False)
|
2021-05-09 11:15:14 +12:00
|
|
|
return
|
|
|
|
|
|
|
|
for (name, value) in metadata.items():
|
|
|
|
if value is None:
|
|
|
|
continue
|
|
|
|
|
|
|
|
self.addLabel(name, Color.parse(value))
|
|
|
|
|
2023-02-05 04:11:18 +13:00
|
|
|
self.complete()
|
|
|
|
|
2021-05-09 11:15:14 +12:00
|
|
|
def save(self):
|
2023-02-05 04:11:18 +13:00
|
|
|
AbstractData.save(self)
|
2022-11-19 11:32:08 +13:00
|
|
|
metadata = collections.OrderedDict()
|
2021-05-09 11:15:14 +12:00
|
|
|
|
2021-05-09 12:14:36 +12:00
|
|
|
for (name, label) in self.labels.items():
|
|
|
|
metadata[name] = label.color
|
2021-05-09 11:15:14 +12:00
|
|
|
|
|
|
|
self.file.save((metadata, None))
|
2023-02-05 04:11:18 +13:00
|
|
|
self.complete()
|