src: fix staging username and add production user

Fix check to verify `staging` username for setting submitter field. Also, add `production` username for the same. Signed-off-by: Jeny Sadadia <[email protected]>
kernelci · Sep 20, 2024 · e44f391 · e44f391
1 parent b2ec394
commit e44f391
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/send_kcidb.py b/src/send_kcidb.py
@@ -452,7 +452,8 @@ def _run(self, context):
             self.log.info(f"Received an event for node: {node['id']}")
 
             # Submit nodes with service origin only for staging pipeline
-            if self._current_user['username'] == 'staging':
+            if self._current_user['username'] in ('staging.kernelci.org',
+                                                  'production'):
                 if node['submitter'] != 'service:pipeline':
                     self.log.debug(f"Not sending node to KCIDB: {node['id']}")
                     continue

diff --git a/src/trigger.py b/src/trigger.py
@@ -85,7 +85,8 @@ def _run_trigger(self, build_config, force, timeout, trees):
             'timeout': checkout_timeout.isoformat(),
             'treeid': treeid,
         }
-        if self._current_user['username'] == 'staging':
+        if self._current_user['username'] in ('staging.kernelci.org',
+                                              'production'):
             node['submitter'] = 'service:pipeline'
         else:
             node['submitter'] = f"user:{self._current_user['email']}"