From d496e1a7db300d94a18f5e35681c5e99eefd9159 Mon Sep 17 00:00:00 2001 From: Chuck Smith Date: Mon, 15 Jul 2024 07:32:40 -0400 Subject: [PATCH] Update script --- src/update_banner.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/src/update_banner.py b/src/update_banner.py index 93736f51..bfa70f20 100644 --- a/src/update_banner.py +++ b/src/update_banner.py @@ -1,11 +1,17 @@ import re +import os + +# Set the correct path to the README.md file +script_dir = os.path.dirname(__file__) # Get the directory where the script is located +root_dir = os.path.join(script_dir, '..') # Move up to the root directory +readme_path = os.path.join(root_dir, 'README.md') # Path to the README.md in the root # Load the README file -with open('README.md', 'r') as file: +with open(readme_path, 'r', encoding='utf-8') as file: readme_contents = file.read() # Count the list items -list_count = len(re.findall(r'^\s*[-*]\s', readme_contents, re.MULTILINE)) +list_count = len(re.findall(r'^\s*-\s', readme_contents, re.MULTILINE)) # Define the new banner message new_banner = f'## Current List Count: {list_count}' @@ -14,5 +20,5 @@ new_readme_contents = re.sub(r'## Current List Count: \d+', new_banner, readme_contents) # Save the changes back to README.md -with open('README.md', 'w') as file: +with open(readme_path, 'w', encoding='utf-8') as file: file.write(new_readme_contents)