Implement version management with VERSION file and update package structure

Author: kamilstanuch

MANIFEST.in

@@ -0,0 +1,5 @@
+include VERSION
+include README.md
+include requirements.txt
+include requirements-dev.txt
+recursive-include codebase_digest/prompt_library *.md

VERSION

@@ -0,0 +1 @@
+0.1.38

codebase_digest/__init__.py

@@ -1 +1,11 @@
-from .app import main
+import os
+
+def read_version():
+    version_file = os.path.join(os.path.dirname(__file__), '..', 'VERSION')
+    try:
+        with open(version_file) as f:
+            return f.read().strip()
+    except FileNotFoundError:
+        return "unknown"
+
+__version__ = read_version()

setup.py

@@ -1,6 +1,9 @@
 from setuptools import setup, find_packages
 import os
 
+with open('VERSION') as f:
+    version = f.read().strip()
+
 def read_requirements(filename='requirements.txt'):
     """Read the requirements from the given filename"""
     current_dir = os.path.dirname(os.path.abspath(__file__))
@@ -15,7 +18,7 @@ def read_requirements(filename='requirements.txt'):
 
 setup(
     name="codebase-digest",
-    version='0.1.38',
+    version=version,
     author="Kamil Stanuch",
     description="Consolidates and analyzes codebases for insights.",
     long_description=long_description,

update_package.py

@@ -106,29 +106,35 @@ def upload_to_pypi(dist_files, max_attempts=3):
                 return False
 
 def update_version():
+    with open('VERSION', 'r') as f:
+        current_version = f.read().strip()
+    
+    print(f"Current version: {current_version}")
+    while True:
+        new_version = input("Enter new version number (e.g., 0.1.7): ")
+        if re.match(r'^\d+(\.\d+){0,2}(\.?[a-zA-Z0-9]+)?$', new_version):
+            try:
+                version.parse(new_version)
+                break
+            except version.InvalidVersion:
+                print("Invalid version format. Please use a valid version number.")
+        else:
+            print("Invalid version format. Please use a valid version number.")
+    
+    # Update VERSION file
+    with open('VERSION', 'w') as f:
+        f.write(new_version)
+    
+    # Update setup.py
     with open('setup.py', 'r') as f:
-        content = f.readlines()
+        content = f.read()
 
-    for i, line in enumerate(content):
-        if line.strip().startswith('version='):
-            current_version = line.split('=')[1].strip().strip("'").strip('"')
-            print(f"Current version: {current_version}")
-            while True:
-                new_version = input("Enter new version number (e.g., 0.1.7): ")
-                if re.match(r'^\d+(\.\d+){0,2}(\.?[a-zA-Z0-9]+)?$', new_version):
-                    try:
-                        version.parse(new_version)
-                        break
-                    except version.InvalidVersion:
-                        print("Invalid version format. Please use a valid version number.")
-                else:
-                    print("Invalid version format. Please use a valid version number.")
-            content[i] = f"    version='{new_version}',\n"
-            break
+    updated_content = re.sub(r"version=['\"][\d.]+['\"]", f"version='{new_version}'", content)
 
     with open('setup.py', 'w') as f:
-        f.writelines(content)
+        f.write(updated_content)
 
+    print(f"Updated VERSION file and setup.py with new version: {new_version}")
     return new_version
 
 def main():