def list_remote_directory(self):
self.list_entry_model.clear()
try:
ftp = self.ssh_client.open_sftp()
csv_files = []
for csv_file in ftp.listdir_attr(self.filepath):
if fnmatch.fnmatch(csv_file.filename, "*.csv"):
csv_files.append(csv_file.filename + '\t\t\t\t' + "{:.2f}".format((csv_file.st_size / 1000.0)) + "kB")
csv_files.sort()
for i in csv_files:
item = QtGui.QStandardItem(i)
self.list_entry_model.appendRow(item)
self.ui.push_save_file.setEnabled(True)
self.ui.push_delete_file.setEnabled(True)
self.ui.push_plot_file.setEnabled(True)
ftp.close()
except Exception as e:
text = "Error: " + str(e)
mw = ctl_message.Message(text)
ftp.close()
self.close()
def __init__(self):
super().__init__()
self.filepath = '/usr/local/src/pow-edge-app/csv_files/'
with open('target.json', 'r') as outfile:
self.ssh_target = json.load(outfile)
self.ui = file_csv.Ui_FileDialog()
self.ui.setupUi(self)
self.list_entry_model = QtGui.QStandardItemModel()
self.ui.listView.setModel(self.list_entry_model)
self.ui.push_list_files.clicked.connect(self.list_remote_directory)
self.ui.push_save_file.clicked.connect(self.save_file_to_local)
self.ui.push_delete_file.clicked.connect(self.delete_selected_files)
self.ui.push_plot_file.clicked.connect(self.plot_csv_file)
self.ui.push_delete_file.setEnabled(False)
self.ui.push_save_file.setEnabled(False)
self.ui.push_plot_file.setEnabled(False)
self.ssh_client = paramiko.SSHClient()
self.ssh_client.load_system_host_keys()
self.ssh_client.set_missing_host_key_policy(paramiko.AutoAddPolicy())
try:
self.ssh_client.connect(self.ssh_target["ip"], username=self.ssh_target["user"],
password=self.ssh_target["pass"], timeout=3)
except Exception as e:
text = "Connection Error: " + str(e)
mw = ctl_message.Message(text)
self.setModal(True)
self.show()
self.exec_()
def delete_selected_files(self):
remote_files = []
selected_items = self.ui.listView.selectedIndexes()
for sel in selected_items:
remote_files.append(sel.data())
try:
for filename in remote_files:
command = "rm " + self.filepath + filename
self.ssh_client.exec_command(command)
self.list_remote_directory()
raise Exception("File(s) deleted successfully!")
except Exception as e:
text = str(e)
mw = ctl_message.Message(text)
def save_file_to_local(self):
try:
remote_files = []
selected_items = self.ui.listView.selectedIndexes()
for item in selected_items:
remote_files.append(item.data())
if not remote_files:
raise Exception("Please select a file.")
if len(remote_files) > 1:
raise Exception("Please select just one file.")
remote_file = remote_files[0]
sep = '\t'
remote_file = remote_file.split(sep, 1)[0]
remote_csv_filename = self.filepath + remote_file
ftp = self.ssh_client.open_sftp()
remote_csv_file = ftp.file(remote_csv_filename, "r")
remote_file_data = remote_csv_file.read()
remote_file_data = remote_file_data.decode('utf-8')
options = QFileDialog.Options()
fileName, _ = QFileDialog.getSaveFileName(self,"QFileDialog.getSaveFileName()", "","CSV Files (*.csv)", options=options)
if not fileName:
raise Exception("File not saved.")
with open(fileName, "w") as file:
file.write(remote_file_data)
ftp.close()
raise Exception("File saved.")
self.close()
except Exception as e:
text = str(e)
mw = ctl_message.Message(text)
def plot_csv_file(self):
v_ch = range(3)
c_ch = range(3, 6)
d_io = range(6, 10)
color_list = ['red', 'green', 'blue', 'yellow']
try:
remote_files = []
selected_items = self.ui.listView.selectedIndexes()
for item in selected_items:
remote_files.append(item.data())
if not remote_files:
raise Exception("Please select a file.")
if len(remote_files) > 1:
raise Exception("Please select just one file.")
remote_file = remote_files[0]
sep = '\t'
remote_file = remote_file.split(sep, 1)[0]
remote_csv_filename = self.filepath + remote_file
ftp = self.ssh_client.open_sftp()
remote_csv_file = ftp.file(remote_csv_filename, "r")
remote_file_data = remote_csv_file.read()
ftp.close()
remote_file_data = remote_file_data.decode('utf-8')
remote_file_data = remote_file_data.splitlines()
plot_csv_data = []
for channel in remote_file_data:
plot_csv_data.append(ast.literal_eval(channel))
for channel in range(len(plot_csv_data)):
plot_csv_data[channel] = list(plot_csv_data[channel])
fig, axs = plt.subplots(2, sharex=True)
fig.suptitle(remote_file)
axs[0].set_title('Voltage/Digital')
axs[1].set_title('Current/Digital')
axd1 = axs[0].twinx()
axd2 = axs[1].twinx()
for channel in v_ch:
for sample in range(len(plot_csv_data[channel])):
plot_csv_data[channel][sample] *= (25.6 / 32768.0)
time_axis = list(range(len(plot_csv_data[channel])))
for index, time_ in enumerate(time_axis):
time_axis[index] = time_ * 2.0 / float(len(time_axis))
axs[0].plot(time_axis, plot_csv_data[channel], color=color_list[v_ch.index(channel)])
axs[0].axhline(y=0, color='k')
for channel in d_io:
time_axis = list(range(len(plot_csv_data[channel])))
for index, time_ in enumerate(time_axis):
time_axis[index] = time_ * 2.0 / float(len(time_axis))
axd1.plot(time_axis, plot_csv_data[channel], color=color_list[d_io.index(channel)])
for channel in c_ch:
for sample in range(len(plot_csv_data[channel])):
plot_csv_data[channel][sample] *= (2.56 / 32768.0 * 2000.0)
time_axis = list(range(len(plot_csv_data[channel])))
for index, time_ in enumerate(time_axis):
time_axis[index] = time_ * 2.0 / float(len(time_axis))
axs[1].plot(time_axis, plot_csv_data[channel], color=color_list[c_ch.index(channel)])
axs[1].axhline(y=0, color='k')
for channel in d_io:
time_axis = list(range(len(plot_csv_data[channel])))
for index, time_ in enumerate(time_axis):
time_axis[index] = time_ * 2.0 / float(len(time_axis))
axd2.plot(time_axis, plot_csv_data[channel], color=color_list[d_io.index(channel)])
plt.show()
except Exception as e:
text = str(e)
mw = ctl_message.Message(text)