Spaces:
Running
Running
| import gradio as gr | |
| import requests | |
| import git | |
| import os | |
| import subprocess | |
| import zipfile | |
| import io | |
| from getpass import getpass | |
| import radon.complexity as radon_complexity | |
| import pylint.lint | |
| from pylint.reporters.text import TextReporter | |
| # Function to download repo as a zip file from GitHub API | |
| def download_repo(github_url): | |
| repo_name = github_url.split('/')[-1] | |
| repo_owner = github_url.split('/')[-2] | |
| api_url = f"https://github.com/{repo_owner}/{repo_name}/archive/refs/heads/main.zip" | |
| # Download the zip file | |
| response = requests.get(api_url) | |
| # Extract the zip file | |
| zip_file = zipfile.ZipFile(io.BytesIO(response.content)) | |
| zip_file.extractall(f"/content/repos/{repo_name}") | |
| return f"/content/repos/{repo_name}" | |
| # Function to fetch the GitHub repo using SSH or token for private repos | |
| def fetch_repo(github_url, token=None): | |
| try: | |
| # Check if the repository is private or public | |
| if token: | |
| repo_name = github_url.split('/')[-1] | |
| repo_owner = github_url.split('/')[-2] | |
| clone_url = f"https://{token}@github.com/{repo_owner}/{repo_name}.git" | |
| else: | |
| repo_name = github_url.split('/')[-1] | |
| repo_owner = github_url.split('/')[-2] | |
| clone_url = f"git@github.com:{repo_owner}/{repo_name}.git" | |
| # Create a temporary directory to clone the repo | |
| temp_dir = f"/content/repos/{repo_name}" | |
| # Check if repo already exists | |
| if not os.path.exists(temp_dir): | |
| # If SSH method fails, fallback to downloading as a zip | |
| try: | |
| git.Repo.clone_from(clone_url, temp_dir) | |
| except Exception as e: | |
| return download_repo(github_url) | |
| return temp_dir | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Function to calculate cyclomatic complexity using Radon | |
| def analyze_complexity(repo_path): | |
| result = subprocess.run(['radon', 'cc', repo_path, '--json'], stdout=subprocess.PIPE) | |
| complexity_data = result.stdout.decode('utf-8') | |
| return complexity_data | |
| # Function to run pylint for linting | |
| def analyze_linting(repo_path): | |
| pylint_output = [] | |
| pylint_runner = pylint.lint.Run([repo_path], reporter=TextReporter(pylint_output)) | |
| return "\n".join(pylint_output) | |
| # Function to check test coverage | |
| def analyze_coverage(repo_path): | |
| result = subprocess.run(['pytest', '--cov', repo_path], stdout=subprocess.PIPE) | |
| coverage_data = result.stdout.decode('utf-8') | |
| return coverage_data | |
| # Function to check for code duplication with jscpd | |
| def analyze_duplication(repo_path): | |
| result = subprocess.run(['jscpd', repo_path, '--reporters', 'text'], stdout=subprocess.PIPE) | |
| duplication_data = result.stdout.decode('utf-8') | |
| return duplication_data | |
| # Main function to analyze the repository | |
| def analyze_code(github_url, token=None): | |
| try: | |
| repo_path = fetch_repo(github_url, token) | |
| # Perform analysis | |
| complexity = analyze_complexity(repo_path) | |
| linting = analyze_linting(repo_path) | |
| coverage = analyze_coverage(repo_path) | |
| duplication = analyze_duplication(repo_path) | |
| # Combine results | |
| results = f"### Cyclomatic Complexity Analysis\n{complexity}\n\n" | |
| results += f"### Linting Results\n{linting}\n\n" | |
| results += f"### Test Coverage\n{coverage}\n\n" | |
| results += f"### Code Duplication\n{duplication}\n" | |
| return results | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Create Gradio Interface | |
| def gradio_interface(): | |
| gr.Interface( | |
| fn=analyze_code, | |
| inputs=[ | |
| gr.Textbox(label="GitHub Repository URL", placeholder="Enter the URL of the GitHub repo"), | |
| gr.Textbox(label="GitHub Token (Optional)", type="password") | |
| ], | |
| outputs="text", | |
| live=True, | |
| title="GitHub Code Quality Analyzer", | |
| description="Enter a GitHub repository URL to analyze its code quality (complexity, linting, coverage, duplication)." | |
| ).launch(share=True) | |
| gradio_interface() | |