新增拍照功能

Author: cocpy

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name='tello-python',
-    version='1.0.3',
+    version='1.0.4',
     author='C灵C',
     author_email='c0c@cocpy.com',
     description='Control DJI Tello drone with Python 3',

tello_python/__init__.py

@@ -2,7 +2,7 @@
 __author__ = 'C灵C'
 __liscence__ = 'MIT'
 __copyright__ = 'Copyright 2020 C灵C'
-__version__ = '1.0.3'
+__version__ = '1.0.4'
 __all__ = ['tello', 'stats']
 
 from .tello import Tello

tello_python/tello.py

@@ -29,6 +29,7 @@ def __init__(self, te_ip: str = '192.168.10.1', debug: bool = True):
 
         # 本项目运行时选项
         self.stream_state = False
+        self.camera_state = False
         self.MAX_TIME_OUT = 15.0
         self.debug = debug
 
@@ -82,9 +83,10 @@ def _video_thread(self):
                 break
 
             # 如果按F1键，截图到当前位置
-            if k == 0:
+            if k == 0 or self.camera_state:
                 png_name = datetime.datetime.now().strftime('%Y%m%d_%H%M%S') + '.png'
                 cv2.imwrite(png_name, frame)
+                self.camera_state = False
         cap.release()
         cv2.destroyAllWindows()
 
@@ -101,6 +103,10 @@ def wait(self, delay: float):
     def get_log(self):
         return self.log
 
+    def take_picture(self):
+        """拍照"""
+        self.camera_state = True
+
     # 以下命令强烈建议配合官方SDK食用
     # https://www.ryzerobotics.com/cn/tello/downloads