zebra/pyzebra
3
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Convert ReflectionPrinter attrs into properties

Browse Source
This commit is contained in:
usov_i 2020-07-15 15:11:10 +02:00
parent 008068534b
commit def7074a23
1 changed files with 25 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
pyzebra/anatric.py
View File

@ -2,6 +2,20 @@ import subprocess
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
REFLECTION_PRINTER_FORMATS = (
"rafin",
"rafinf",
"rafin2d",
"rafin2di",
"orient",
"shelx",
"jana2k",
"jana2kf",
"raw",
"oksana",
)
def anatric(config_file): def anatric(config_file):
subprocess.run(["anatric", config_file], check=True) subprocess.run(["anatric", config_file], check=True)
@ -15,9 +29,6 @@ class AnatricConfig:
tree = ET.parse(filename) tree = ET.parse(filename)
self._tree = tree self._tree = tree
reflectionPrinter_elem = tree.find("ReflectionPrinter")
self.reflectionPrinter_format = reflectionPrinter_elem.attrib["format"]
alg_elem = tree.find("Algorithm") alg_elem = tree.find("Algorithm")
self.algorithm = alg_elem.attrib["implementation"] self.algorithm = alg_elem.attrib["implementation"]
if self.algorithm == "adaptivemaxcog": if self.algorithm == "adaptivemaxcog":
@ -190,5 +201,16 @@ class AnatricConfig:
def dist1(self, value): def dist1(self, value):
self._tree.find("DataFactory").find("dist1").attrib["value"] = value self._tree.find("DataFactory").find("dist1").attrib["value"] = value
@property
def reflectionPrinter_format(self):
return self._tree.find("ReflectionPrinter").attrib["format"]
@reflectionPrinter_format.setter
def reflectionPrinter_format(self, value):
if value not in REFLECTION_PRINTER_FORMATS:
raise ValueError("Unknown ReflectionPrinter format.")
self._tree.find("ReflectionPrinter").attrib["format"] = value
def save_as(self, filename): def save_as(self, filename):
self._tree.write(filename) self._tree.write(filename)