blue
/
awesome-zotero
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Apply Sourcery suggestions to the Python script

This commit is contained in:
Anas Sheashaey 2024-11-11 02:39:30 +02:00
parent a172cdf9b0
commit d8c755ad1c
1 changed files with 6 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
main.py
View File

@ -3,23 +3,22 @@
import re
import sys
def extract_repo(url):
"""Extract the GitHub repository name from the URL."""
match = re.match(r'https://github\.com/([^/]+/[^/]+)', url)
return match.group(1) if match else ''
return match[1] if match else ''
def add_badges(lines):
"""Add GitHub badges to lines containing GitHub repository URLs."""
processed_lines = []
for line in lines:
if 'https://github.com/' in line:
# Extract the GitHub URL
url_match = re.search(r'https://github\.com/\S+', line)
if url_match:
repo_url = url_match.group(0)
if url_match := re.search(r'https://github\.com/\S+', line):
repo_url = url_match[0]
repo_name = extract_repo(repo_url)
processed_lines.append(line)
if repo_name:
processed_lines.append(line)
processed_lines.append('') # Add a blank line
# Add shields.io badges
badges = [
@ -31,8 +30,6 @@ def add_badges(lines):
''
]
processed_lines.extend(badges)
else:
processed_lines.append(line)
else:
processed_lines.append(line)
else:
@ -44,10 +41,7 @@ def style_badges(lines):
def replace_style(match):
alt_text = match.group(1)
url = match.group(2)
if '?' in url:
url += '&style=flat-square'
else:
url += '?style=flat-square'
url += '&style=flat-square' if '?' in url else '?style=flat-square'
return f'![{alt_text}]({url})'
styled_lines = []