Duo-Guardian / scripts /fix_ci.py
Daksh C Jain
added split nodes
392da6f
import os
import gitlab
from dotenv import load_dotenv
load_dotenv()
TEST_PROJECT_ID = 80447540
gl_token = os.getenv("GITLAB_TOKEN")
gl = gitlab.Gitlab('https://gitlab.com', private_token=gl_token)
project = gl.projects.get(TEST_PROJECT_ID)
ci_config = """
stages:
- analyze
context_brain_job:
stage: analyze
image: python:3.10-slim
script:
- apt-get update && apt-get install -y git
- git clone https://oauth2:${GITLAB_TOKEN}@gitlab.com/gitlab-ai-hackathon/participants/35466397.git context_brain_repo
- cd context_brain_repo
- pip install -r requirements.txt
- python src/main.py
- mv gl-code-quality-report.json ../gl-code-quality-report.json
artifacts:
when: always
reports:
codequality: gl-code-quality-report.json
rules:
- if: $CI_MERGE_REQUEST_IID
"""
try:
project.commits.create({
'branch': 'main',
'commit_message': 'Add pipeline to main',
'actions': [{'action': 'update', 'file_path': '.gitlab-ci.yml', 'content': ci_config}]
})
print("Added to main")
except Exception as e:
print("Check 1:", e)
pass
try:
project.commits.create({
'branch': 'feature/agentic-testing-v1',
'commit_message': 'Add pipeline to branch',
'actions': [{'action': 'create', 'file_path': '.gitlab-ci.yml', 'content': ci_config}]
})
print("Added to branch")
except Exception as e:
print("Check 2:", e)
pass