wouter/Artemis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add Constants class in utilities

Browse Source
This commit is contained in:
alessandro90
2018-11-04 11:43:27 +01:00
parent a7c44ca981
commit cf8be9f6e3
4 changed files with 73 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
download_window.py
View File

@@ -5,7 +5,7 @@ from threads import DownloadThread, ThreadStatus
Ui_Download_window, _ = uic.loadUiType("download_db_window.ui")
class DownloadWindow(QWidget, Ui_Download_window):
def __init__(self, db_location, data_folder):
def __init__(self):
super().__init__()
self.setupUi(self)
self.setWindowFlags(
@@ -35,7 +35,7 @@ class DownloadWindow(QWidget, Ui_Download_window):
the downloaded file has been discarded.""")
self.bad_file_msg.finished.connect(self.close)
self.download_thread = DownloadThread(db_location, data_folder)
self.download_thread = DownloadThread()
self.download_thread.finished.connect(self.wait_close)
self.cancel_btn.clicked.connect(self.terminate_process)

72
main.py
View File

@@ -22,39 +22,17 @@ from audio_player import AudioPlayer
from double_text_button import DoubleTextButton
from download_window import DownloadWindow
from utilities import Constants
qt_creator_file = "main_window.ui"
Ui_MainWindow, _ = uic.loadUiType(qt_creator_file)
class MyApp(QMainWindow, Ui_MainWindow):
db_location = 'https://aresvalley.com/Storage/Artemis/Database/data.zip'
data_folder = 'Data'
spectra_folder = 'Spectra'
audio_folder = 'Audio'
icons_folder = 'icons_imgs'
Band = namedtuple("Band", ["lower", "upper"])
ELF = Band(0, 30) # Formally it is (3, 30) Hz.
SLF = Band(30, 300)
ULF = Band(300, 3000)
VLF = Band(3000, 30000)
LF = Band(30 * 10**3, 300 * 10**3)
MF = Band(300 * 10 ** 3, 3000 * 10**3)
HF = Band(3 * 10**6, 30 * 10**6)
VHF = Band(30 * 10**6, 300 * 10**6)
UHF = Band(300 * 10**6, 3000 * 10**6)
SHF = Band(3 * 10**9, 30 * 10**9)
EHF = Band(30 * 10**9, 300 * 10**9)
bands = ELF, SLF, ULF, VLF, LF, MF, HF, VHF, UHF, SHF, EHF
active_color = "#39eaff"
inactive_color = "#9f9f9f"
conversion_factors = {"Hz":1, "kHz":1000, "MHz":1000000, "GHz":1000000000}
def __init__(self):
super().__init__()
self.setupUi(self)
self.set_initial_size()
self.download_window = DownloadWindow(self.db_location, self.data_folder)
self.download_window = DownloadWindow()
self.actionExit.triggered.connect(qApp.quit)
self.action_update_database.triggered.connect(self.download_db)
self.db_version = None
@@ -300,8 +278,8 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.stop,
self.volume,
self.audio_progress,
self.data_folder,
self.audio_folder)
Constants.data_folder,
Constants.audio_folder)
BandLabel = namedtuple("BandLabel", ["left", "center", "right"])
self.band_labels = [
@@ -391,7 +369,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
"category_code",
"acf",]
try:
self.db = read_csv(os.path.join(self.data_folder, 'db.csv'),
self.db = read_csv(os.path.join(Constants.data_folder, 'db.csv'),
sep = '*',
header = None,
index_col = 0,
@@ -437,8 +415,8 @@ class MyApp(QMainWindow, Ui_MainWindow):
upper_units = upper_combo_box.currentText()
lower_value = lower_spin_box.value()
upper_value = upper_spin_box.value()
inf_limit = (lower_value * self.conversion_factors[lower_units]) \
// self.conversion_factors[upper_units]
inf_limit = (lower_value * Constants.conversion_factors[lower_units]) \
// Constants.conversion_factors[upper_units]
counter = 0
while inf_limit > upper_spin_box.maximum():
counter += 1
@@ -470,13 +448,13 @@ class MyApp(QMainWindow, Ui_MainWindow):
range_lbl):
activate_low = False
activate_high = False
color = self.inactive_color
color = Constants.inactive_color
title = ''
to_display = ''
if activate_low_btn.isChecked():
to_display += str(lower_spinbox.value()) + ' ' + lower_unit.currentText()
activate_low = True
color = self.active_color
color = Constants.active_color
if lower_confidence.value() != 0:
to_display += ' - ' + str(lower_confidence.value()) + ' %'
else:
@@ -485,7 +463,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
if activate_up_btn.isChecked():
to_display += str(upper_spinbox.value()) + ' ' + upper_unit.currentText()
activate_high = True
color = self.active_color
color = Constants.active_color
if upper_confidence.value() != 0:
to_display += ' + ' + str(upper_confidence.value()) + ' %'
else:
@@ -525,7 +503,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
def update_status_tip(self, available_signals):
if available_signals < self.total_signals:
self.statusbar.setStyleSheet(f'color: {self.active_color}')
self.statusbar.setStyleSheet(f'color: {Constants.active_color}')
else:
self.statusbar.setStyleSheet('color: #ffffff')
self.statusbar.showMessage(f"{available_signals} out of {self.total_signals} signals displayed.")
@@ -591,7 +569,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
band_filter_ok = False
any_checked = False
for btn, band_limits in zip(self.frequency_filters_btns, self.bands):
for btn, band_limits in zip(self.frequency_filters_btns, Constants.bands):
if btn.isChecked():
any_checked = True
if signal_freqs[0] < band_limits.upper and signal_freqs[1] >= band_limits.lower:
@@ -656,9 +634,9 @@ class MyApp(QMainWindow, Ui_MainWindow):
else:
return cat_checked == positive_cases and cat_checked > 0
@classmethod
def filters_ok(cls, spinbox, filter_unit, confidence, sign = 1):
band_filter = spinbox.value() * cls.conversion_factors[filter_unit.currentText()]
@staticmethod
def filters_ok(spinbox, filter_unit, confidence, sign = 1):
band_filter = spinbox.value() * Constants.conversion_factors[filter_unit.currentText()]
return band_filter + sign * (confidence.value() * band_filter) // 100
@pyqtSlot(QListWidgetItem, QListWidgetItem)
@@ -699,9 +677,9 @@ class MyApp(QMainWindow, Ui_MainWindow):
self.description_text.setText(current_signal.at["description"])
for cat, cat_lab in zip(category_code, self.category_labels):
if cat == '0':
cat_lab.setStyleSheet(f"color: {self.inactive_color};")
cat_lab.setStyleSheet(f"color: {Constants.inactive_color};")
elif cat == '1':
cat_lab.setStyleSheet(f"color: {self.active_color};")
cat_lab.setStyleSheet(f"color: {Constants.active_color};")
self.set_band_range(current_signal)
self.audio_widget.set_audio_player(self.current_signal_name)
else:
@@ -713,7 +691,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
for lab in self.property_labels:
lab.setText("N/A")
for lab in self.category_labels:
lab.setStyleSheet(f"color: {self.inactive_color};")
lab.setStyleSheet(f"color: {Constants.inactive_color};")
self.set_band_range()
self.audio_widget.set_audio_player()
@@ -760,20 +738,20 @@ class MyApp(QMainWindow, Ui_MainWindow):
return 10**9
def display_spectrogram(self):
default_pic = os.path.join(self.icons_folder, "nosignalselected.png")
default_pic = os.path.join(Constants.icons_folder, "nosignalselected.png")
item = self.result_list.currentItem()
if item:
spectrogram_name = item.text()
path_spectr = os.path.join(self.data_folder, self.spectra_folder, spectrogram_name + ".png")
path_spectr = os.path.join(Constants.data_folder, Constants.spectra_folder, spectrogram_name + ".png")
if not QFileInfo(path_spectr).exists():
path_spectr = os.path.join(self.icons_folder, "spectrumnotavailable.png")
path_spectr = os.path.join(Constants.icons_folder, "spectrumnotavailable.png")
else:
path_spectr = default_pic
self.spectrogram.setPixmap(QPixmap(path_spectr))
@classmethod
def activate_band_category(cls, band_label, activate = True):
color = cls.active_color if activate else cls.inactive_color
@staticmethod
def activate_band_category(band_label, activate = True):
color = Constants.active_color if activate else Constants.inactive_color
for label in band_label:
label.setStyleSheet(f"color: {color};")
@@ -781,7 +759,7 @@ class MyApp(QMainWindow, Ui_MainWindow):
if current_signal is not None and not self.is_undef_freq(current_signal):
lower_freq = int(current_signal.at["inf_freq"])
upper_freq = int(current_signal.at["sup_freq"])
zipped = list(zip(self.bands, self.band_labels))
zipped = list(zip(Constants.bands, self.band_labels))
for i, w in enumerate(zipped):
band, band_label = w
if lower_freq >= band.lower and lower_freq < band.upper:

25
threads.py
View File

@@ -6,7 +6,7 @@ from shutil import rmtree
import urllib3
from zipfile import ZipFile
from PyQt5.QtCore import QThread
import utilities
from utilities import checksum_ok, Constants
class ThreadStatus(Enum):
ok = auto()
@@ -15,11 +15,9 @@ class ThreadStatus(Enum):
bad_download_err = auto()
class DownloadThread(QThread):
def __init__(self, db_location, path):
def __init__(self):
super().__init__()
self.__db_location = db_location
self.__path = path
self.__status = None
self.__status = ThreadStatus.ok
self.reason = 0
@property
@@ -32,31 +30,24 @@ class DownloadThread(QThread):
def run(self):
try:
db = urllib3.PoolManager().request('GET', self.__db_location)
# db = urllib.request.urlopen(self.__db_location)
db = urllib3.PoolManager().request('GET', Constants.db_location)
# db = urllib.request.urlopen(Constants.db_location)
# raise urllib.error.URLError('Test')
except urllib3.exceptions.MaxRetryError: # No internet connection.
# self.no_connection_error.emit()
self.__status = ThreadStatus.no_connection_err
return
if db.status != 200:
self.reason = db.reason
# self.bad_download_error.emit()
self.__status = ThreadStatus.bad_download_err
return
if not utilities.checksum_ok(db.data, "folder"):
# self.bad_download_error.emit()
if not checksum_ok(db.data, "folder"):
self.__status = ThreadStatus.bad_download_err
return
if os.path.exists(self.__path):
rmtree(self.__path)
if os.path.exists(Constants.data_folder):
rmtree(Constants.data_folder)
try:
# data_folder = db.read()
with ZipFile(BytesIO(db.data)) as zipped:
zipped.extractall()
except:
# self.bad_file_error.emit()
self.__status = ThreadStatus.bad_file_err
return
else:
self.__status = ThreadStatus.ok

40
utilities.py
View File

@@ -1,10 +1,46 @@
from collections import namedtuple
import hashlib
from pandas import read_csv
class _ReadOnlyProperty(object):
def __init__(self, value):
self.value = value
def __get__(self, obj, objtype):
return self.value
def __Set__(self, obj, value):
return NotImplementedError("Cannot change a constant.")
class Constants(object):
db_location = _ReadOnlyProperty('https://aresvalley.com/Storage/Artemis/Database/data.zip')
ref_loc = _ReadOnlyProperty('https://aresvalley.com/Storage/Artemis/Database/data.zip.log')
data_folder = _ReadOnlyProperty('Data')
spectra_folder = _ReadOnlyProperty('Spectra')
audio_folder = _ReadOnlyProperty('Audio')
icons_folder = _ReadOnlyProperty('icons_imgs')
__Band = namedtuple("Band", ["lower", "upper"])
__ELF = __Band(0, 30) # Formally it is (3, 30) Hz.
__SLF = __Band(30, 300)
__ULF = __Band(300, 3000)
__VLF = __Band(3000, 30000)
__LF = __Band(30 * 10**3, 300 * 10**3)
__MF = __Band(300 * 10 ** 3, 3000 * 10**3)
__HF = __Band(3 * 10**6, 30 * 10**6)
__VHF = __Band(30 * 10**6, 300 * 10**6)
__UHF = __Band(300 * 10**6, 3000 * 10**6)
__SHF = __Band(3 * 10**9, 30 * 10**9)
__EHF = __Band(30 * 10**9, 300 * 10**9)
bands = _ReadOnlyProperty((__ELF, __SLF, __ULF, __VLF, __LF, __MF, __HF, __VHF, __UHF, __SHF, __EHF))
active_color = _ReadOnlyProperty("#39eaff")
inactive_color = _ReadOnlyProperty("#9f9f9f")
conversion_factors = _ReadOnlyProperty({"Hz":1, "kHz":1000, "MHz":1000000, "GHz":1000000000})
def checksum_ok(data, what):
code = hashlib.sha256()
code.update(data)
ref_loc = 'https://aresvalley.com/Storage/Artemis/Database/data.zip.log'
if what == "folder":
n = 0
elif what == "db":
@@ -12,7 +48,7 @@ def checksum_ok(data, what):
else:
raise ValueError("Wrong entry name.")
try:
reference = read_csv(ref_loc, delimiter = '*').iat[-1, n]
reference = read_csv(Constants.ref_loc, delimiter = '*').iat[-1, n]
except HTTPError:
return False
return code.hexdigest() == reference