74 lines
2.9 KiB
Python
74 lines
2.9 KiB
Python
from PySide6.QtWidgets import QPushButton
|
|
from pathlib import Path
|
|
from core.database import DatabaseManager
|
|
from core.media import MediaRepository
|
|
from ui.widgets.color_list_widget import ColorListWidget
|
|
from ui.widgets.thumbnail_list_widget import ThumbnailListWidget
|
|
from ui.widgets.split_view_widget import SplitView
|
|
from .camera_controller import CameraController
|
|
|
|
|
|
class MainController:
|
|
def __init__(self, view):
|
|
self.db = DatabaseManager()
|
|
self.db.connect()
|
|
self.media_repo = MediaRepository(self.db)
|
|
self.media_repo.sync_media()
|
|
|
|
self.camera_controller = CameraController()
|
|
|
|
self.view = view
|
|
self.color_list: ColorListWidget = view.color_list_widget
|
|
self.thumbnail_list: ThumbnailListWidget = view.thumbnail_widget
|
|
self.split_view: SplitView = view.preview_widget
|
|
|
|
self.photo_button: QPushButton = view.photo_button
|
|
self.photo_button.clicked.connect(self.take_photo)
|
|
|
|
self.color_list.colorSelected.connect(self.on_color_selected)
|
|
self.color_list.editColor.connect(self.on_edit_color)
|
|
self.thumbnail_list.selectedThumbnail.connect(self.on_thumbnail_selected)
|
|
|
|
self.camera_controller.errorOccurred.connect(self.split_view.widget_start.set_info_text)
|
|
self.camera_controller.frameReady.connect(self.split_view.set_live_image)
|
|
self.split_view.widget_start.camera_start_btn.clicked.connect(self.camera_controller.start)
|
|
|
|
def start_camera(self):
|
|
pass
|
|
|
|
def load_colors(self) -> None:
|
|
colors = self.db.get_all_colors()
|
|
print("Loaded colors:", colors)
|
|
self.color_list.set_colors(colors)
|
|
|
|
|
|
def on_color_selected(self, color_name: str):
|
|
print(f"Wybrano kolor: {color_name}")
|
|
color_id = self.db.get_color_id(color_name)
|
|
if color_id is not None:
|
|
media_items = self.db.get_media_for_color(color_id)
|
|
print(f"Media dla koloru {color_name} (ID: {color_id}):", media_items)
|
|
|
|
self.thumbnail_list.list_widget.clear()
|
|
for media in media_items:
|
|
if media['file_type'] == 'photo':
|
|
file_name = Path(media['media_path']).name
|
|
self.thumbnail_list.add_thumbnail(media['media_path'], file_name, media['id'])
|
|
else:
|
|
print(f"Nie znaleziono koloru o nazwie: {color_name}")
|
|
|
|
def on_edit_color(self, color_name: str):
|
|
print(f"Edycja koloru: {color_name}")
|
|
|
|
def on_thumbnail_selected(self, media_id: int):
|
|
media = self.db.get_media(media_id)
|
|
if media:
|
|
print(f"Wybrano miniaturę o ID: {media_id}, ścieżka: {media['media_path']}")
|
|
self.split_view.set_reference_image(media['media_path'])
|
|
else:
|
|
print(f"Nie znaleziono mediów o ID: {media_id}")
|
|
|
|
def take_photo(self):
|
|
print("Robienie zdjęcia...")
|
|
self.split_view.toglle_live_view()
|