Frame13/ayon-core
1
0
Fork 1
mirror of https://github.com/ynput/ayon-core.git synced 2026-01-02 08:54:53 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

added possibility to have register conditions if action will be registered

Browse source
This commit is contained in:
Jakub Trllo 2019-04-04 11:12:13 +02:00
parent 83f907d771
commit fdf261ed58
1 changed files with 32 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

44
pype/ftrack/lib/ftrack_base_handler.py
View file

@ -46,18 +46,7 @@ class BaseHandler(object):
else:
label = '{} {}'.format(self.label, self.variant)
try:
if hasattr(self, "role_list") and len(self.role_list) > 0:
username = self.session.api_user
user = self.session.query(
'User where username is "{}"'.format(username)
).one()
available = False
for role in user['user_security_roles']:
if role['security_role']['name'] in self.role_list:
available = True
break
if available is False:
raise MissingPermision
self._preregister()
start_time = time.perf_counter()
func(*args, **kwargs)
@ -119,6 +108,37 @@ class BaseHandler(object):
def reset_session(self):
self.session.reset()
def _preregister(self):
# Rolecheck
if hasattr(self, "role_list") and len(self.role_list) > 0:
username = self.session.api_user
user = self.session.query(
'User where username is "{}"'.format(username)
).one()
available = False
for role in user['user_security_roles']:
if role['security_role']['name'] in self.role_list:
available = True
break
if available is False:
raise MissingPermision
# Custom validations
result = self.preregister()
if result is True:
return
msg = "Pre-register conditions were not met"
if isinstance(result, str):
msg = result
raise Exception(msg)
def preregister(self):
'''
Preregister conditions.
Registration continues if returns True.
'''
return True
def register(self):
'''
Registers the action, subscribing the discover and launch topics.