class ProjectData:
def __init__(self, project):
self.folder = os.path.dirname(project)
self.project_file = project
self.current_file = 0
self.current_label = 0
self.modified = False
self.bad_files = []
self.datafile = None
self.config = None
self.read_conf()
self.read_file()
def read(self):
if self.modified:
self.read_conf()
self.modified = False
self.read_file()
if self.current_label >= len(self.config["labels"]):
self.current_label = 0
def read_conf(self):
try:
with open(self.project_file) as in_file:
self.config = json.load(in_file)
except IOError:
logger.error("Unable to read {}: permission denied".format(
self.project_file))
exit(2)
def read_file(self):
if set(self.config["files"]) == set(self.bad_files):
dialogs.report_no_files()
exit(3)
current = self.config["files"][self.current_file]
file_path = os.path.join(self.folder, current)
if current in self.bad_files or not os.path.exists(file_path):
if not os.path.exists(file_path):
self.datafile = None
self.bad_files.append(current)
dialogs.notify_read_error(current)
self.next_file()
self.read_file()
return
try:
self.datafile = DataFile(file_path, self.config["labels"])
self.insert_header()
except (UnrecognizedFormatError, BadFileError):
self.datafile = None
self.bad_files.append(current)
dialogs.notify_read_error(current)
self.next_file()
self.read_file()
def insert_header(self):
header = self.datafile.get_data_header()
if str(header) not in self.config.keys():
self.config[str(header)] = {
"plot": [[i] for i in self.datafile.get_data_columns()],
"normalize": [],
"functions": []
}
self.modified = True
self.save_config()
def save_file(self):
self.datafile.save()
def save_config(self):
if self.modified:
try:
with open(self.project_file, 'w') as out_file:
json.dump(self.config, out_file)
except IOError:
logger.error("Unable to write {}: permission denied".format(
self.project_file))
exit(2)
self.modified = False
def next_label(self):
self.current_label = (self.current_label + 1) % len(
self.config["labels"])
def prev_label(self):
length = len(self.config["labels"])
self.current_label = (self.current_label - 1 + length) % length
def get_current_label(self):
label = self.config["labels"][self.current_label]
color = self.config["colors"][self.current_label]
return label, color
def get_label_color(self, label):
index = self.config["labels"].index(label)
return self.config["colors"][index]
def get_labels_info(self):
return self.config["labels"], self.config["colors"]
def set_labels_info(self, names, colors):
self.config["labels"] = names
self.config["colors"] = colors
self.modified = True
if self.current_label >= len(names):
self.current_label = 0
def get_plot_info(self):
header = self.datafile.get_data_header()
conf = self.config[str(header)]
return conf["plot"], conf["normalize"]
def set_plot_info(self, plot_set, normalize):
header = self.datafile.get_data_header()
conf = self.config[str(header)]
conf["plot"] = plot_set
conf["normalize"] = normalize
self.modified = True
def next_file(self):
self.current_file = (self.current_file + 1) % len(self.config["files"])
def prev_file(self):
self.current_file = (self.current_file - 1 + len(
self.config["files"])) % len(self.config["files"])
def get_functions(self):
header = self.datafile.get_data_header()
return self.config[str(header)]["functions"]
def add_function(self, fs):
header = self.datafile.get_data_header()
new_header = header + [fs.name]
self.config[str(new_header)] = self.config[str(header)]
self.config[str(new_header)]["functions"].append(fs.name)
self.datafile.add_function(fs)
self.modified = True
def remove_function(self, index):
header = self.datafile.get_data_header()
conf = self.config[str(header)]
f_name = conf["functions"][index]
f_col = self.datafile.get_data_columns()[header.index(f_name)]
for plot in conf["plot"]:
if f_col in plot:
plot.remove(f_col)
for i, col in enumerate(plot):
if col > f_col:
plot[i] = col - 1
self.datafile.remove_function(f_name)
del conf["functions"][index]
self.modified = True
class FilesData:
def __init__(self, files):
self.files_list = files
self.config_list = []
self.current_file = 0
self.current_label = 0
self.modified = False
self.bad_files = []
self.datafile = None
self.config = None
self.init()
self.read()
def init(self):
for file in self.files_list:
conf = file + ".json"
file_conf = conf if os.path.exists(conf) else None
self.config_list.append(file_conf)
# noinspection PyTypeChecker
def read(self):
conf_path = self.config_list[self.current_file]
if conf_path:
self.read_conf()
self.read_file()
else:
self.config = {
"labels": ["Label"],
"colors": ["#1f77b4"],
}
self.read_file()
self.config["plot"] = [[i]
for i in self.datafile.get_data_columns()]
self.config["normalize"] = []
self.config["functions"] = []
if self.current_label >= len(self.config["labels"]):
self.current_label = 0
self.modified = False
def read_conf(self):
conf_path = self.config_list[self.current_file]
try:
with open(conf_path) as in_file:
self.config = json.load(in_file)
except IOError:
logger.error(
"Unable to read {}: permission denied".format(conf_path))
exit(2)
def read_file(self):
if set(self.files_list) == set(self.bad_files):
dialogs.report_no_files()
exit(3)
current = self.files_list[self.current_file]
if current in self.bad_files:
self.next_file()
self.read_file()
return
try:
self.datafile = DataFile(current, self.config["labels"])
except (UnrecognizedFormatError, BadFileError):
self.datafile = None
self.bad_files.append(current)
dialogs.notify_read_error(os.path.basename(current))
self.next_file()
self.read_file()
def save_file(self):
self.datafile.save()
def save_config(self):
if self.modified:
conf_path = self.files_list[self.current_file] + ".json"
try:
with open(conf_path, 'w') as out_file:
json.dump(self.config, out_file)
except IOError:
logger.error(
"Unable to write {}: permission denied".format(conf_path))
exit(2)
self.config_list[self.current_file] = conf_path
self.modified = False
def next_label(self):
self.current_label = (self.current_label + 1) % len(
self.config["labels"])
def prev_label(self):
length = len(self.config["labels"])
self.current_label = (self.current_label - 1 + length) % length
def get_current_label(self):
label = self.config["labels"][self.current_label]
color = self.config["colors"][self.current_label]
return label, color
def get_label_color(self, label):
index = self.config["labels"].index(label)
return self.config["colors"][index]
def get_labels_info(self):
return self.config["labels"], self.config["colors"]
def set_labels_info(self, names, colors):
self.config["labels"] = names
self.config["colors"] = colors
self.modified = True
if self.current_label >= len(names):
self.current_label = 0
def get_plot_info(self):
return self.config["plot"], self.config["normalize"]
def set_plot_info(self, plot_set, normalize):
self.config["plot"] = plot_set
self.config["normalize"] = normalize
self.modified = True
def next_file(self):
self.current_file = (self.current_file + 1) % len(self.files_list)
def prev_file(self):
self.current_file = (self.current_file - 1 +
len(self.files_list)) % len(self.files_list)
def get_functions(self):
return self.config["functions"]
def add_function(self, fs):
self.datafile.add_function(fs)
self.config["functions"].append(fs.name)
self.modified = True
def remove_function(self, index):
f_name = self.config["functions"][index]
header = self.datafile.get_data_header()
f_col = self.datafile.get_data_columns()[header.index(f_name)]
for plot in self.config["plot"]:
if f_col in plot:
plot.remove(f_col)
for i, col in enumerate(plot):
if col > f_col:
plot[i] = col - 1
self.datafile.remove_function(f_name)
del self.config["functions"][index]
self.modified = True
