From a45df1532510c09ea5430a012f3ce5c48711f29c Mon Sep 17 00:00:00 2001
From: Damien Picard <dam.pic@free.fr>
Date: Mon, 29 Mar 2021 11:33:56 +0200
Subject: [PATCH] Add camera rigs: fix poll method when not created with add-on

When creating an armature containing a camera, without the add-on, the
UI panel tries to display, but the data is not set up to use it
properly.
---
 add_camera_rigs/operators.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/add_camera_rigs/operators.py b/add_camera_rigs/operators.py
index 646566ae5..18300e085 100644
--- a/add_camera_rigs/operators.py
+++ b/add_camera_rigs/operators.py
@@ -5,7 +5,10 @@ from bpy.types import Operator
 
 
 def get_rig_and_cam(obj):
-    if obj.type == 'ARMATURE':
+    if (obj.type == 'ARMATURE'
+            and "rig_id" in obj
+            and obj["rig_id"].lower() in {"dolly_rig",
+                                          "crane_rig", "2d_rig"}):
         cam = None
         for child in obj.children:
             if child.type == 'CAMERA':
-- 
GitLab