From 122a4dc9db074f7bd14421e1d8b9244a05318da7 Mon Sep 17 00:00:00 2001
From: Michael reda <eng.michaelreda@gmail.com>
Date: Wed, 5 Apr 2023 12:14:06 +0200
Subject: [PATCH] add sync to specific projects or listen only

---
 openpype/modules/kitsu/kitsu_module.py        | 16 ++++++++++--
 .../modules/kitsu/utils/update_op_with_zou.py | 25 +++++++++++++++++--
 2 files changed, 37 insertions(+), 4 deletions(-)

diff --git a/openpype/modules/kitsu/kitsu_module.py b/openpype/modules/kitsu/kitsu_module.py
index b91373af20..f4e3dd5691 100644
--- a/openpype/modules/kitsu/kitsu_module.py
+++ b/openpype/modules/kitsu/kitsu_module.py
@@ -124,19 +124,31 @@ def push_to_zou(login, password):
 
 
 @cli_main.command()
+@click.option("-prjs", "--projects", envvar="SYNC_PROJECTS", help="Sync specific kitsu projects")
 @click.option("-l", "--login", envvar="KITSU_LOGIN", help="Kitsu login")
 @click.option(
     "-p", "--password", envvar="KITSU_PWD", help="Password for kitsu username"
 )
-def sync_service(login, password):
+def sync_service(login, password, projects="^"):
     """Synchronize openpype database from Zou sever database.
 
     Args:
         login (str): Kitsu user login
         password (str): Kitsu user password
+        projects (str): specific kitsu projects
+
+    SYNC_PROJECTS:
+    *: all projects
+    ^: dont sync any project just listen
+    "project01 project02 ...": to choose custom projects
+
+
     """
     from .utils.update_op_with_zou import sync_all_projects
     from .utils.sync_service import start_listeners
 
-    sync_all_projects(login, password)
+    projects = projects.strip()
+    projects = projects.split(' ')
+
+    sync_all_projects(login, password, specific_projects=projects)
     start_listeners(login, password)
diff --git a/openpype/modules/kitsu/utils/update_op_with_zou.py b/openpype/modules/kitsu/utils/update_op_with_zou.py
index 4f4f0810bc..a397198a13 100644
--- a/openpype/modules/kitsu/utils/update_op_with_zou.py
+++ b/openpype/modules/kitsu/utils/update_op_with_zou.py
@@ -359,7 +359,7 @@ def write_project_to_op(project: dict, dbcon: AvalonMongoDB) -> UpdateOne:
 
 
 def sync_all_projects(
-    login: str, password: str, ignore_projects: list = None
+    login: str, password: str, ignore_projects: list = None, specific_projects: list = None
 ):
     """Update all OP projects in DB with Zou data.
 
@@ -367,6 +367,7 @@ def sync_all_projects(
         login (str): Kitsu user login
         password (str): Kitsu user password
         ignore_projects (list): List of unsynced project names
+        specific_projects (list): List of synced project names
     Raises:
         gazu.exception.AuthFailedException: Wrong user login and/or password
     """
@@ -381,7 +382,27 @@ def sync_all_projects(
     dbcon = AvalonMongoDB()
     dbcon.install()
     all_projects = gazu.project.all_projects()
-    for project in all_projects:
+
+
+    project_to_sync = []
+    if specific_projects == ['*']:
+        project_to_sync = all_projects
+
+    elif specific_projects == ['^']:
+        return
+
+    elif isinstance(specific_projects, list):
+        all_kitsu_projects = {p['name']: p for p in all_projects}
+        for proj_name in specific_projects:
+            if proj_name in all_kitsu_projects:
+                project_to_sync.append(all_kitsu_projects[proj_name])
+            else:
+                log.info(f'`{proj_name}` project does not exists in kitsu.'
+                         f' Please make sure you write the project correctly.')
+    else:
+        return
+
+    for project in project_to_sync:
         if ignore_projects and project["name"] in ignore_projects:
             continue
         sync_project_from_kitsu(dbcon, project)