def on_mechanism_storage_paste_clicked(self):
"""Add the storage data from string."""
expr, ok = QInputDialog.getText(self, "Storage",
"Please input expression:")
if not ok:
return
try:
#Put the expression into parser to see if it is legal.
PMKS_parser.parse(expr)
except:
QMessageBox.warning(self, "Loading failed",
"Your expression is in an incorrect format.")
return
name, ok = QInputDialog.getText(self, "Storage", "Please input name tag:")
if not ok:
return
if not name:
nameList = [
self.mechanism_storage.item(i).text()
for i in range(self.mechanism_storage.count())
]
i = 0
while "Prototype_{}".format(i) in nameList:
i += 1
name = "Prototype_{}".format(i)
_addStorage(self, name, expr, clear=False)
def on_action_Import_PMKS_server_triggered(self):
"""Load PMKS URL and turn it to expression."""
URL, ok = QInputDialog.getText(self, "PMKS URL input",
"Please input link string:")
if not ok:
return
if not URL:
QMessageBox.warning(self, "Loading failed",
"Your link is in an incorrect format.")
return
try:
for s in URL.split('?')[-1].split('&'):
if 'mech=' in s:
expr = s.replace('mech=', '').split('|')
break
textList = [s for s in expr if s not in ('', " ", '\n')]
expression = []
while textList:
item = textList.pop(0).split(',')[:-1]
for i, e in enumerate(reversed(item)):
if e in ['R', 'P', 'RP']:
t = -(i + 1)
break
links = item[:t]
item = item[t:]
expression.append("J[{}, P[{}], L[{}]]".format(
"{}:{}".format(item[0], item[-1]) if item[0] != 'R' else 'R',
", ".join((item[1], item[2])), ", ".join(links)))
expression = "M[{}]".format(", ".join(expression))
except:
QMessageBox.warning(self, "Loading failed",
"Your link is in an incorrect format.")
else:
self.parseExpression(expression)
def __rename(self):
"""Show up a string input to change the data name."""
row = self.collections_list.currentRow()
if not row > -1:
return
name, ok = QInputDialog.getText(
self,
"Profile name",
"Please enter the profile name:"
)
if not ok:
return
if not name:
QMessageBox.warning(
self,
"Profile name",
"Can not use blank string to rename."
)
return
item = self.collections_list.item(row)
self.collections[name] = self.collections.pop(item.text())
item.setText(name)
self.unsave_func()
def __copy(self):
"""Ask a name to copy a data."""
row = self.collections_list.currentRow()
if not row > -1:
return
name, ok = QInputDialog.getText(
self,
"Profile name",
"Please enter a new profile name:"
)
if not ok:
return
if not name:
QMessageBox.warning(
self,
"Profile name",
"Can not use blank string to rename."
)
return
name_old = self.collections_list.item(row).text()
self.collections[name] = self.collections[name_old].copy()
self.collections_list.addItem(name)
self.unsave_func()
def pasteStorage(self):
"""Add the storage data from string."""
expr, ok = QInputDialog.getMultiLineText(self, "Storage",
"Please input expression:")
if not ok:
return
try:
# Put the expression into parser to see if it is legal.
parse_params(expr)
except Exception as e:
print(e)
QMessageBox.warning(self, "Loading failed",
"Your expression is in an incorrect format.")
return
name, ok = QInputDialog.getText(self, "Storage",
"Please input name tag:")
if not ok:
return
name_list = [
self.mechanism_storage.item(i).text()
for i in range(self.mechanism_storage.count())
]
i = 0
name = name or f"Prototype_{i}"
while name in name_list:
name = f"Prototype_{i}"
i += 1
self.__add_storage(name, expr)
def on_save_button_clicked(self):
"""Save the profile to database."""
if self.profile_name:
name = self.profile_name
ok = True
else:
name, ok = QInputDialog.getText(self, "Profile name",
"Please enter the profile name:")
if ok:
i = 0
while (name not in self.collections) and (not name):
name = "Structure_{}".format(i)
self.collections[name] = self.get_currentMechanismParams()
self.profile_name = name
self.unsaveFunc()
def __import_pmks_url(self):
"""Load PMKS URL and turn it to expression."""
url, ok = QInputDialog.getText(
self,
"PMKS URL input",
"Please input link string:"
)
if not ok:
return
if not url:
QMessageBox.warning(
self,
"Loading failed",
"Your link is in an incorrect format."
)
return
try:
for s in url.split('?')[-1].split('&'):
if 'mech=' in s:
expr = s.replace('mech=', '').split('|')
break
else:
raise ValueError
text_list = [s for s in expr if s not in ('', " ", '\n')]
expr.clear()
while text_list:
item = text_list.pop(0).split(',')[:-1]
for i, e in enumerate(reversed(item)):
if e in {'R', 'P', 'RP'}:
t = -(i + 1)
break
else:
raise ValueError
links = item[:t]
item = item[t:]
type_text = f"{item[0]}:{item[-1]}" if item[0] != 'R' else 'R'
links_text = ", ".join(links)
expr.append(f"J[{type_text}, P[{item[1]}, {item[2]}], L[{links_text}]]")
expr = "M[" + ", ".join(expr) + "]"
except (ValueError, IndexError):
QMessageBox.warning(
self,
"Loading failed",
"Your link is in an incorrect format."
)
else:
self.parse_expression(expr)
def __save(self):
"""Save the profile to database."""
if self.profile_name.text():
name = self.profile_name.text()
ok = True
else:
name, ok = QInputDialog.getText(self, "Profile name",
"Please enter the profile name:")
if not ok:
return
i = 0
while (name not in self.collections) and (not name):
name = f"Structure_{i}"
self.collections[name] = self.__get_current_mechanism_params()
self.profile_name.setText(name)
self.unsave_func()
def on_save_button_clicked(self):
"""Save as new profile to collection widget."""
if not self.mech_params:
return
name, ok = QInputDialog.getText(self, "Profile name",
"Please enter the profile name:")
if not ok:
return
i = 0
while (name not in self.collections) and (not name):
name = "Structure_{}".format(i)
mech_params = deepcopy(self.mech_params)
for key in ['Driver', 'Follower', 'Target']:
for name in mech_params[key]:
mech_params[key][name] = None
self.collections[name] = mech_params
self.unsaveFunc()
def on_record_start_toggled(self, toggled):
"""Save to file path data."""
if toggled:
self.MainCanvas.recordStart(int(360 /
self.record_interval.value()))
return
path = self.MainCanvas.getRecordPath()
name, ok = QInputDialog.getText(self, "Recording completed!",
"Please input name tag:")
if (not name) or (name in self.pathData):
i = 0
while "Record_{}".format(i) in self.pathData:
i += 1
QMessageBox.information(self, "Record",
"The name tag is being used or empty.")
name = "Record_{}".format(i)
self.addPath(name, path)
def __add_from_edges(self):
"""Add collection by input string."""
edges_str = ""
while not edges_str:
edges_str, ok = QInputDialog.getText(
self, "Add by edges", "Please enter a connection expression:\n"
"Example: [(0, 1), (1, 2), (2, 3), (3, 0)]")
if not ok:
return
try:
edges = eval(edges_str)
if any(len(edge) != 2 for edge in edges):
raise ValueError("wrong format")
except (SyntaxError, ValueError) as error:
QMessageBox.warning(self, str(error), f"Error: {error}")
return
else:
self.add_collection(edges)
def on_add_by_edges_button_clicked(self):
"""Add collection by input string."""
edgesSTR = ""
while not edgesSTR:
edgesSTR, ok = QInputDialog.getText(
self, "Add by edges",
"Please enter a connection expression:\n" +
"Example: [(0, 1), (1, 2), (2, 3), (3, 0)]")
if not ok:
return
try:
edges = eval(edgesSTR)
if any(len(edge) != 2 for edge in edges):
raise SyntaxError("Wrong format")
except Exception as e:
QMessageBox.warning(self, str(e), "Error: {}".format(e))
return
else:
self.addCollection(edges)
def __save_profile(self):
"""Save as new profile to collection widget."""
if not self.mech_params:
return
name, ok = QInputDialog.getText(self, "Profile name",
"Please enter the profile name:")
if not ok:
return
i = 0
while (not name) and (name not in self.collections):
name = f"Structure_{i}"
i += 1
mech_params = deepcopy(self.mech_params)
for key in ('Placement', 'Target'):
for mp in mech_params[key]:
mech_params[key][mp] = None
self.collections[name] = mech_params
self.workbook_no_save()
def __start_record(self, toggled: bool):
"""Save to file path data."""
if toggled:
self.MainCanvas.record_start(int(
self.dial_spinbox.maximum() / self.record_interval.value()
))
return
path = self.MainCanvas.get_record_path()
name, ok = QInputDialog.getText(
self,
"Recording completed!",
"Please input name tag:"
)
i = 0
name = name or f"Record_{i}"
while name in self.__path_data:
name = f"Record_{i}"
i += 1
QMessageBox.information(
self,
"Record",
"The name tag is being used or empty."
)
self.add_path(name, path)
def save(self, fileName: str, isBranch=False):
"""Save database.
+ Append to new branch function.
"""
author_name = self.FileAuthor.text()
if not author_name:
author_name = self.FileAuthor.placeholderText()
branch_name = '' if isBranch else self.branch_current.text()
commit_text = self.FileDescription.text()
while not author_name:
author_name, ok = QInputDialog.getText(
self, "Author", "Please enter author's name:",
QLineEdit.Normal, "Anonymous")
if not ok:
return
while not branch_name.isidentifier():
branch_name, ok = QInputDialog.getText(
self, "Branch", "Please enter a branch name:",
QLineEdit.Normal, "master")
if not ok:
return
while not commit_text:
commit_text, ok = QInputDialog.getText(self, "Commit",
"Please add a comment:",
QLineEdit.Normal,
"Update mechanism.")
if not ok:
return
if ((fileName != self.fileName.absoluteFilePath())
and os.path.isfile(fileName)):
os.remove(fileName)
print("The original file has been overwritten.")
self.connectDatabase(fileName)
isError = False
with db.atomic():
if author_name in (user.name for user in UserModel.select()):
author_model = (UserModel.select().where(
UserModel.name == author_name).get())
else:
author_model = UserModel(name=author_name)
if branch_name in (branch.name for branch in BranchModel.select()):
branch_model = (BranchModel.select().where(
BranchModel.name == branch_name).get())
else:
branch_model = BranchModel(name=branch_name)
pointData = self.pointDataFunc()
linkcolor = {
vlink.name: vlink.colorSTR
for vlink in self.linkDataFunc()
}
args = {
'author':
author_model,
'description':
commit_text,
'mechanism':
compress("M[{}]".format(", ".join(vpoint.expr
for vpoint in pointData))),
'linkcolor':
compress(linkcolor),
'storage':
compress(self.storageDataFunc()),
'pathdata':
compress(self.pathDataFunc()),
'collectiondata':
compress(self.CollectDataFunc()),
'triangledata':
compress(self.TriangleDataFunc()),
'inputsdata':
compress(self.InputsDataFunc()),
'algorithmdata':
compress(self.AlgorithmDataFunc()),
'branch':
branch_model
}
try:
args['previous'] = (CommitModel.select().where(
CommitModel.id == self.commit_current_id.value()).get())
except CommitModel.DoesNotExist:
args['previous'] = None
new_commit = CommitModel(**args)
try:
author_model.save()
branch_model.save()
new_commit.save()
except Exception as e:
print(str(e))
db.rollback()
isError = True
else:
self.history_commit = (CommitModel.select().order_by(
CommitModel.id))
if isError:
os.remove(fileName)
print("The file was removed.")
return
self.read(fileName)
print("Saving \"{}\" successful.".format(fileName))
size = QFileInfo(fileName).size()
print(
"Size: {}".format("{} MB".format(round(size / 1024 /
1024, 2)) if size / 1024 //
1024 else "{} KB".format(round(size / 1024, 2))))
def save(self, file_name: str, is_branch: bool = False):
"""Save database, append commit to new branch function."""
author_name = self.FileAuthor.text(
) or self.FileAuthor.placeholderText()
branch_name = '' if is_branch else self.branch_current.text()
commit_text = self.FileDescription.text()
while not author_name:
author_name, ok = QInputDialog.getText(
self, "Author", "Please enter author's name:",
QLineEdit.Normal, "Anonymous")
if not ok:
return
while not branch_name.isidentifier():
branch_name, ok = QInputDialog.getText(
self, "Branch", "Please enter a branch name:",
QLineEdit.Normal, "master")
if not ok:
return
while not commit_text:
commit_text, ok = QInputDialog.getText(self, "Commit",
"Please add a comment:",
QLineEdit.Normal,
"Update mechanism.")
if not ok:
return
if (file_name !=
self.file_name.absoluteFilePath()) and isfile(file_name):
os_remove(file_name)
print("The original file has been overwritten.")
self.__connect_database(file_name)
is_error = False
with _db.atomic():
if author_name in (user.name for user in UserModel.select()):
author_model = (UserModel.select().where(
UserModel.name == author_name).get())
else:
author_model = UserModel(name=author_name)
if branch_name in (branch.name for branch in BranchModel.select()):
branch_model = (BranchModel.select().where(
BranchModel.name == branch_name).get())
else:
branch_model = BranchModel(name=branch_name)
args = {
'author':
author_model,
'description':
commit_text,
'mechanism':
_compress(self.__point_expr_func()),
'linkcolor':
_compress(self.__link_expr_func()),
'storage':
_compress(list(self.__storage_data_func())),
'pathdata':
_compress(self.__path_data_func()),
'collectiondata':
_compress(self.__collect_data_func()),
'triangledata':
_compress(self.__triangle_data_func()),
'inputsdata':
_compress(
tuple((b, d) for b, d, a in self.__inputs_data_func())),
'algorithmdata':
_compress(self.__algorithm_data_func()),
'branch':
branch_model,
}
try:
args['previous'] = (CommitModel.select().where(
CommitModel.id == self.commit_current_id.value()).get())
except CommitModel.DoesNotExist:
args['previous'] = None
new_commit = CommitModel(**args)
try:
author_model.save()
branch_model.save()
new_commit.save()
except Exception as e:
print(str(e))
_db.rollback()
is_error = True
else:
self.history_commit = CommitModel.select().order_by(
CommitModel.id)
if is_error:
os_remove(file_name)
print("The file was removed.")
return
self.read(file_name)
print(f"Saving \"{file_name}\" successful.")
size = QFileInfo(file_name).size()
print("Size: " + (f"{size / 1024 / 1024:.02f} MB" if size / 1024 //
1024 else f"{size / 1024:.02f} KB"))
