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