IPCam.py 2.6 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182
  1. from PySide6.QtWidgets import QLabel
  2. from PySide6.QtCore import Qt
  3. from threading import Thread
  4. from PySide6.QtGui import QImage, QPixmap
  5. import cv2
  6. from module import create_filename
  7. from module.ImageTool import cupture_face
  8. from logger import logger
  9. class IPCam(Thread):
  10. lnk_connect: str = None
  11. qLabel: QLabel = None
  12. net_work_error: bool = False
  13. def __init__(self, parent=None) -> None:
  14. Thread.__init__(self, parent)
  15. self.status = True
  16. self.cap = True
  17. def run(self) -> None:
  18. logger.info("srarting take image from ip cam")
  19. self.net_work_error = False
  20. try:
  21. self.cap = cv2.VideoCapture()
  22. if 'rtsp' in self.lnk_connect:
  23. self.cap.open(self.lnk_connect)
  24. while self.status:
  25. if 'http' in self.lnk_connect:
  26. self.cap.open(self.lnk_connect)
  27. if not self.cap.isOpened():
  28. raise ConnectionError
  29. ret, frame = self.cap.read()
  30. if not ret: continue
  31. color_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
  32. h, w, ch = color_frame.shape
  33. img = QImage(color_frame.data, w, h,
  34. ch * w, QImage.Format_RGB888)
  35. label_geometry = self.qLabel.geometry()
  36. h, w = label_geometry.height()-20, label_geometry.width()-20
  37. self.__scaled_img = QPixmap.fromImage(
  38. img.scaled(h, w, Qt.KeepAspectRatio))
  39. self.qLabel.setPixmap(self.__scaled_img)
  40. # "http://admin:admin102030@192.168.1.108/cgi-bin/snapshot.cgi?2"\
  41. except ConnectionError:
  42. logger.error("Нет сети или не правильная строка подключения")
  43. logger.info("stop take image from ip cam")
  44. self.status = False
  45. self.qLabel.setText(
  46. 'Нет сети или не правильная строка подключения')
  47. except RuntimeError:
  48. self.status = False
  49. def stop_cam(self):
  50. logger.info("stop take image from ip cam")
  51. self.cap.release()
  52. self.status = False
  53. cv2.destroyAllWindows()
  54. self.join()
  55. def cupture_image(self, qLabel: QLabel) -> str:
  56. name_file = create_filename()
  57. qLabel.setPixmap(self.__scaled_img)
  58. self.__scaled_img.save(name_file, 'jpg')
  59. self.stop_cam()
  60. face_file_name = create_filename('face')
  61. cupture_face(name_file, face_file_name)
  62. return name_file