Moved all ep_dict code into one function

This commit is contained in:
Jacob Danell 2023-03-02 14:05:36 +01:00 committed by Milan Kolar
parent 93eb9fce86
commit 95e1f95bc1

View file

@ -124,6 +124,11 @@ class Listener:
log.info("Listening to Kitsu events...") log.info("Listening to Kitsu events...")
gazu.events.run_client(self.event_client) gazu.events.run_client(self.event_client)
def get_ep_dict(self, ep_id):
if ep_id and ep_id != "":
return gazu.entity.get_entity(ep_id)
return
# == Project == # == Project ==
def _new_project(self, data): def _new_project(self, data):
"""Create new project into OP DB.""" """Create new project into OP DB."""
@ -179,10 +184,7 @@ class Listener:
# Print message # Print message
ep_id = asset.get("episode_id") ep_id = asset.get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
else:
ep = None
msg = "Asset created: {proj_name} - {ep_name}" \ msg = "Asset created: {proj_name} - {ep_name}" \
"{asset_type_name} - {asset_name}".format( "{asset_type_name} - {asset_name}".format(
@ -233,10 +235,8 @@ class Listener:
) )
# Print message # Print message
ep = None
ep_id = asset["data"]["zou"].get("episode_id") ep_id = asset["data"]["zou"].get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
msg = "Asset deleted: {proj_name} - {ep_name}" \ msg = "Asset deleted: {proj_name} - {ep_name}" \
"{asset_type_name} - {asset_name}".format( "{asset_type_name} - {asset_name}".format(
@ -334,10 +334,8 @@ class Listener:
self._update_sequence(data) self._update_sequence(data)
# Print message # Print message
ep = None
ep_id = sequence.get("episode_id") ep_id = sequence.get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
msg = "Sequence created: {proj_name} - {ep_name}" \ msg = "Sequence created: {proj_name} - {ep_name}" \
"{sequence_name}".format( "{sequence_name}".format(
@ -386,10 +384,8 @@ class Listener:
) )
# Print message # Print message
ep = None
ep_id = sequence["data"]["zou"].get("episode_id") ep_id = sequence["data"]["zou"].get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
gazu_project = gazu.project.get_project( gazu_project = gazu.project.get_project(
sequence["data"]["zou"]["project_id"]) sequence["data"]["zou"]["project_id"])
@ -418,9 +414,8 @@ class Listener:
self._update_shot(data) self._update_shot(data)
# Print message # Print message
ep = None ep_id = shot["episode_id"]
if shot["episode_id"] and shot["episode_id"] != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(shot["episode_id"])
msg = "Shot created: {proj_name} - {ep_name}" \ msg = "Shot created: {proj_name} - {ep_name}" \
"{sequence_name} - {shot_name}".format( "{sequence_name} - {shot_name}".format(
@ -471,10 +466,8 @@ class Listener:
) )
# Print message # Print message
ep = None
ep_id = shot["data"]["zou"].get("episode_id") ep_id = shot["data"]["zou"].get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
msg = "Shot deleted: {proj_name} - {ep_name}" \ msg = "Shot deleted: {proj_name} - {ep_name}" \
"{sequence_name} - {shot_name}".format( "{sequence_name} - {shot_name}".format(
@ -496,10 +489,8 @@ class Listener:
task = gazu.task.get_task(data["task_id"]) task = gazu.task.get_task(data["task_id"])
# Print message # Print message
ep = None
ep_id = task.get("episode_id") ep_id = task.get("episode_id")
if ep_id and ep_id != "": ep = self.get_ep_dict(ep_id)
ep = gazu.asset.get_episode(ep_id)
parent_name = None parent_name = None
ent_type = None ent_type = None