diff --git a/.gitignore b/.gitignore index d3a72f2..0de428e 100644 --- a/.gitignore +++ b/.gitignore @@ -15,3 +15,4 @@ __pycache__/ /src/lib/db/*imports*/ /tmp/ /venv*/ +runme_custom.sh diff --git a/runme_custom.sh b/runme_custom.sh new file mode 100755 index 0000000..bd9a5cb --- /dev/null +++ b/runme_custom.sh @@ -0,0 +1,4 @@ +#!/bin/bash -e +cd "$(dirname "$0")" +source "./venv/bin/activate" || source "./venv/Scripts/activate" || : +python -O "./src/main.py" --no-edgetpu --no-tflite $* diff --git a/src/components/__init__.py b/src/components/__init__.py index b19b0f1..16f3aa6 100644 --- a/src/components/__init__.py +++ b/src/components/__init__.py @@ -1,3 +1,5 @@ +import sys + from .archive_synchronizer import ArchiveSynchronizer from .consumer import Consumer from .galaxy_camera import GalaxyCamera @@ -11,5 +13,6 @@ from .tecna_marposs_provaset_t3 import TecnaMarpossProvasetT3 from .tecna_screwdriver import TecnaScrewdriver from .test_component import TestComponent from .uvc_camera import UVCCamera -from .vision import Vision +if "--vision" in sys.argv: + from .vision import Vision from .vision_saver import VisionSaver diff --git a/src/main.py b/src/main.py index c23e80a..9cb7b16 100644 --- a/src/main.py +++ b/src/main.py @@ -55,13 +55,16 @@ try: from components import (ArchiveSynchronizer, GalaxyCamera, Multicomp730424, NeoPixels, Os_Label_Printer, RemoteAPI, TecnaMarpossProvasetT3, TecnaScrewdriver, - UVCCamera, Vision, VisionSaver) + UVCCamera, VisionSaver) from lib.db import Users from lib.helpers import ConfigReader from PyQt5.QtCore import QObject, QThread, pyqtSignal from PyQt5.QtWidgets import QApplication, QMessageBox from ui import About, Archive, Login, Main_Window, Test, Users_Management + if "--vision" in sys.argv: + from components import Vision + class Main(QObject): do = pyqtSignal(dict) @@ -87,9 +90,11 @@ try: "screwdriver": {"c": TecnaScrewdriver, "k": {"paused": True}}, "tecna_t3": {"c": TecnaMarpossProvasetT3, "k": {"paused": True}}, "uvc_camera": {"c": UVCCamera, "k": {"paused": True}}, - "vision_saver": {"c": VisionSaver, "t": False}, - "vision": {"c": Vision, "k": {"paused": True}}, + "vision_saver": {"c": VisionSaver, "t": False} } + if "--vision" in sys.argv: + self.components_specs["vision"] = {"c": Vision, "k": {"paused": True}} + for component_name in list(self.components_specs): if self.config.get("hardware_config", {}).get(component_name, None) != "present": self.components_specs.pop(component_name, None)