121 lines
4.1 KiB
Python
Executable File
121 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
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[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:
|
|
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('') # Add a blank line
|
|
# Add shields.io badges
|
|
badges = [
|
|
f' ',
|
|
f'',
|
|
f'',
|
|
f'',
|
|
f'',
|
|
''
|
|
]
|
|
processed_lines.extend(badges)
|
|
else:
|
|
processed_lines.append(line)
|
|
else:
|
|
processed_lines.append(line)
|
|
return processed_lines
|
|
|
|
|
|
def ensure_punctuation(lines):
|
|
"""Ensure all list item descriptions end with a period."""
|
|
punctuated_lines = []
|
|
for line in lines:
|
|
if re.match(r'- \[.*?\]\(.*?\) - .*[^.]$', line):
|
|
line += '.' # Append a period
|
|
punctuated_lines.append(line)
|
|
return punctuated_lines
|
|
|
|
|
|
def remove_extra_parentheses(lines):
|
|
"""Remove extra closing parentheses at the end of image markdown links."""
|
|
corrected_lines = []
|
|
for line in lines:
|
|
corrected_line = re.sub(r'(!\[[^\]]*\]\([^\)]*\))\)+', r'\1', line)
|
|
corrected_lines.append(corrected_line)
|
|
return corrected_lines
|
|
|
|
|
|
def add_awesome_badge(lines):
|
|
"""Ensure the Awesome badge is correctly added."""
|
|
# Check for an existing "Awesome" badge in the lines
|
|
for line in lines:
|
|
if "https://awesome.re" in line:
|
|
return lines # Badge already exists; do nothing
|
|
|
|
# Add the badge at the top if not present
|
|
badge = '[](https://awesome.re)'
|
|
lines.insert(0, badge)
|
|
return lines
|
|
|
|
|
|
def fix_list_item_format(lines):
|
|
"""Ensure list items use a dash separator between link and description."""
|
|
formatted_lines = []
|
|
for line in lines:
|
|
if re.match(r'- \[.*?\]\(.*?\) .*', line):
|
|
line = re.sub(r'(\]\(.*?\)) ', r'\1 - ', line) # Ensure a dash after the link
|
|
formatted_lines.append(line)
|
|
return formatted_lines
|
|
|
|
|
|
def remove_duplicate_links(lines):
|
|
"""Remove duplicate links in the markdown."""
|
|
seen_links = set()
|
|
deduplicated_lines = []
|
|
for line in lines:
|
|
# Check for links in the current line
|
|
if match := re.search(r'\((https?://[^\)]+)\)', line):
|
|
link = match.group(1)
|
|
if link in seen_links and "https://awesome.re" not in line:
|
|
continue # Skip duplicate links (but don't remove Awesome badge)
|
|
seen_links.add(link)
|
|
deduplicated_lines.append(line)
|
|
return deduplicated_lines
|
|
|
|
|
|
def main():
|
|
if len(sys.argv) != 2:
|
|
print("Usage: python script.py <input_markdown_file>")
|
|
sys.exit(1)
|
|
|
|
input_file = sys.argv[1]
|
|
|
|
with open(input_file, 'r', encoding='utf-8') as f:
|
|
lines = [line.rstrip('\n') for line in f]
|
|
|
|
lines = add_awesome_badge(lines)
|
|
lines = add_badges(lines)
|
|
lines = ensure_punctuation(lines)
|
|
lines = fix_list_item_format(lines)
|
|
lines = remove_duplicate_links(lines)
|
|
lines = remove_extra_parentheses(lines)
|
|
|
|
for line in lines:
|
|
print(line)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|