diff options
author | Alexander Gavrilov <angavrilov@gmail.com> | 2022-11-04 14:25:37 +0300 |
---|---|---|
committer | Alexander Gavrilov <angavrilov@gmail.com> | 2022-11-04 14:25:58 +0300 |
commit | 85c414a2023c1fdf16b6f3c9dc462fe242a625b9 (patch) | |
tree | 30698ee3ded397cca66bff5b6071f91483982237 /rigify/base_rig.py | |
parent | 5a818af95080cccf04dfa8317f0e966bff515c64 (diff) |
Rigify: cleanup PyCharm warning highlights in base_rig.py
Also declare stage names for auto-completion.
Diffstat (limited to 'rigify/base_rig.py')
-rw-r--r-- | rigify/base_rig.py | 41 |
1 files changed, 24 insertions, 17 deletions
diff --git a/rigify/base_rig.py b/rigify/base_rig.py index a1f9e952..41430996 100644 --- a/rigify/base_rig.py +++ b/rigify/base_rig.py @@ -1,18 +1,17 @@ # SPDX-License-Identifier: GPL-2.0-or-later import collections +import typing from .utils.errors import RaiseErrorMixin from .utils.bones import BoneDict, BoneUtilityMixin from .utils.mechanism import MechanismUtilityMixin from .utils.metaclass import BaseStagedClass -# Only export certain symbols via 'from base_rig import *' -__all__ = ['BaseRig', 'stage'] -#============================================= +############################################## # Base Rig -#============================================= +############################################## class GenerateCallbackHost(BaseStagedClass, define_stages=True): """ @@ -162,13 +161,13 @@ class BaseRig(GenerateCallbackHost, RaiseErrorMixin, BoneUtilityMixin, Mechanism # Collection of bone names for use in implementing the rig self.bones = BoneDict( # ORG bone names - org = self.find_org_bones(pose_bone), + org=self.find_org_bones(pose_bone), # Control bone names - ctrl = BoneDict(), + ctrl=BoneDict(), # MCH bone names - mch = BoneDict(), + mch=BoneDict(), # DEF bone names - deform = BoneDict(), + deform=BoneDict(), ) # Data useful for complex rig interaction: @@ -234,9 +233,9 @@ class BaseRig(GenerateCallbackHost, RaiseErrorMixin, BoneUtilityMixin, Mechanism """ -#============================================= +############################################## # Rig Utility -#============================================= +############################################## class RigUtility(BoneUtilityMixin, MechanismUtilityMixin): @@ -270,13 +269,21 @@ class RigComponent(LazyRigComponent): self.enable_component() -#============================================= +############################################## # Rig Stage Decorators -#============================================= +############################################## +# Generate @stage.<...> decorators for all valid stages. +@GenerateCallbackHost.stage_decorator_container class stage: - pass - -# Generate @stage.<...> decorators for all valid stages -for name, decorator in GenerateCallbackHost.make_stage_decorators(): - setattr(stage, name, decorator) + # Declare stages for auto-completion - doesn't affect execution. + initialize: typing.Callable + prepare_bones: typing.Callable + generate_bones: typing.Callable + parent_bones: typing.Callable + configure_bones: typing.Callable + preapply_bones: typing.Callable + apply_bones: typing.Callable + rig_bones: typing.Callable + generate_widgets: typing.Callable + finalize: typing.Callable |