save_for_regression.py

# when run
# checks if there is uncommitted code
# if there is uncommitted code, ti retuns an error
# if there is no uncommitted code, it saves the dataframe as a parquet file with the commit hash in the name

import pytest
import pandas as pd
from result_data_processor import ResultDataProcessor

import os

import subprocess

def check_git_changes(repo_path):
    try:
        # Change to the repository directory
        original_path = os.getcwd()
        os.chdir(repo_path)

        # Run the git status command
        result = subprocess.run(['git', 'status', '--porcelain'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True)

        # Check the result
        if result.returncode != 0:
            print(f"Error checking git status: {result.stderr}")
            return False, False

        # Check for tracked and untracked changes
        tracked_changes = any(line[:2].strip() != '??' for line in result.stdout.splitlines())
        untracked_changes = any(line[:2] == '??' for line in result.stdout.splitlines())

        return tracked_changes, untracked_changes

    finally:
        # Change back to the original directory
        os.chdir(original_path)

if __name__ == '__main__':
    tracked_changes, untracked_changes = check_git_changes('.')
    if tracked_changes:
        print("There are tracked changes")
    else:
        print("There are no tracked changes")
        df_current = ResultDataProcessor().data
        last_commit = os.popen('git rev-parse HEAD').read().strip()
        print(last_commit)
        # save the current output to a file
        df_current.to_parquet(f'dataframe_history/output_{last_commit}.parquet', index=True)
        print("Saved output to file")
    if untracked_changes:
        print("There are untracked changes")
    else:
        print("There are no untracked changes")