gpt-pr 0.3.0__tar.gz → 0.4.1__tar.gz

{gpt-pr-0.3.0 → gpt-pr-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.3.0
+Version: 0.4.1
 Summary: Automate your GitHub workflow with GPT-PR: an OpenAI powered library for streamlined PR generation.
 Home-page: http://github.com/alissonperez/gpt-pr
 Author: Alisson R. Perez

{gpt-pr-0.3.0 → gpt-pr-0.4.1}/gpt_pr.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: gpt-pr
-Version: 0.3.0
+Version: 0.4.1
 Summary: Automate your GitHub workflow with GPT-PR: an OpenAI powered library for streamlined PR generation.
 Home-page: http://github.com/alissonperez/gpt-pr
 Author: Alisson R. Perez

{gpt-pr-0.3.0 → gpt-pr-0.4.1}/gpt_pr.egg-info/SOURCES.txt

@@ -11,12 +11,14 @@ gpt_pr.egg-info/not-zip-safe
 gpt_pr.egg-info/requires.txt
 gpt_pr.egg-info/top_level.txt
 gptpr/__init__.py
+gptpr/checkversion.py
 gptpr/config.py
 gptpr/consolecolor.py
 gptpr/gh.py
 gptpr/gitutil.py
 gptpr/main.py
 gptpr/prdata.py
+gptpr/test_checkversion.py
 gptpr/test_config.py
 gptpr/test_prdata.py
 gptpr/version.py

gpt-pr-0.4.1/gptpr/checkversion.py

@@ -0,0 +1,90 @@
+import requests
+import os
+import json
+import tempfile
+from gptpr.version import __version__
+from datetime import datetime, timedelta
+
+from gptpr import consolecolor as cc
+
+
+PACKAGE_NAME = 'gpt-pr'
+CACHE_FILE = os.path.join(os.path.expanduser("~"), '.gpt_pr_update_cache.json')
+CACHE_DURATION = timedelta(days=1)
+
+
+def cache_daily_version(func):
+    def wrapper(*args, **kwargs):
+        cache = load_cache()
+        last_checked = cache.get('last_checked')
+
+        if last_checked:
+            last_checked = datetime.fromisoformat(last_checked)
+
+            if datetime.now() - last_checked < CACHE_DURATION:
+                # Use cached version info
+                latest_version = cache.get('latest_version')
+                if latest_version:
+                    return latest_version
+
+        latest_version = func(*args, **kwargs)
+        cache = {
+            'last_checked': datetime.now().isoformat(),
+            'latest_version': latest_version
+        }
+        save_cache(cache)
+
+        return latest_version
+
+    return wrapper
+
+
+def get_cache_file_path():
+    temp_dir = tempfile.gettempdir()
+    return os.path.join(temp_dir, f'{PACKAGE_NAME}_update_cache.json')
+
+
+@cache_daily_version
+def get_latest_version():
+    url = f'https://pypi.org/pypi/{PACKAGE_NAME}/json'
+
+    try:
+        response = requests.get(url)
+        response.raise_for_status()
+        data = response.json()
+        return data['info']['version']
+    except requests.exceptions.RequestException as e:
+        print(f"Error fetching latest version info: {e}")
+        return None
+
+
+def load_cache():
+    cache_file = get_cache_file_path()
+    if os.path.exists(cache_file):
+        with open(cache_file, 'r') as file:
+            return json.load(file)
+
+    return {}
+
+
+def save_cache(data):
+    cache_file = get_cache_file_path()
+    with open(cache_file, 'w') as file:
+        file.write(json.dumps(data))
+
+
+def check_for_updates():
+    latest_version = get_latest_version()
+
+    if latest_version and latest_version != __version__:
+        print('')
+        print(cc.yellow(
+            f'A new version of {PACKAGE_NAME} is available ({latest_version}). '
+            f'You are using version {__version__}. Please update by running'),
+            cc.green(f'pip install --upgrade {PACKAGE_NAME}.'))
+        print('')
+
+
+if __name__ == "__main__":
+    check_for_updates()
+    # Your CLI code here

{gpt-pr-0.3.0 → gpt-pr-0.4.1}/gptpr/main.py

@@ -7,6 +7,7 @@ from gptpr.prdata import get_pr_data
 from gptpr.version import __version__
 from gptpr.config import config, config_command_example, CONFIG_README_SECTION
 from gptpr import consolecolor as cc
+from gptpr.checkversion import check_for_updates
 
 
 def run(base_branch='main', yield_confirmation=False, version=False):
@@ -81,10 +82,14 @@ def print_config():
 
 
 def main():
+    check_for_updates()
+
     fire.Fire(run)
 
 
 def run_config():
+    check_for_updates()
+
     fire.Fire({
         'set': set_config,
         'get': get_config,

gpt-pr-0.4.1/gptpr/test_checkversion.py

@@ -0,0 +1,126 @@
+import pytest
+import requests
+import json
+from datetime import datetime
+from unittest.mock import patch, mock_open
+
+from gptpr.version import __version__
+from gptpr.checkversion import (get_latest_version, load_cache,
+                                save_cache, check_for_updates,
+                                CACHE_DURATION)
+
+
+@pytest.fixture
+def mock_requests_get(mocker):
+    return mocker.patch('requests.get')
+
+
+@pytest.fixture
+def mock_os_path_exists(mocker):
+    return mocker.patch('os.path.exists')
+
+
+@pytest.fixture
+def mock_open_file(mocker):
+    return mocker.patch('builtins.open', mock_open())
+
+
+@pytest.fixture
+def mock_datetime(mocker):
+    return mocker.patch('gptpr.checkversion.datetime')
+
+
+def test_get_latest_version(mock_requests_get, mock_os_path_exists):
+    mock_os_path_exists.return_value = False
+    mock_response = mock_requests_get.return_value
+    mock_response.raise_for_status.return_value = None
+    mock_response.json.return_value = {'info': {'version': '2.0.0'}}
+
+    assert get_latest_version() == '2.0.0'
+
+
+def test_get_latest_version_error(mock_requests_get, mock_os_path_exists):
+    mock_os_path_exists.return_value = False
+    mock_requests_get.side_effect = requests.exceptions.RequestException
+
+    assert get_latest_version() is None
+
+
+def test_load_cache(mock_os_path_exists, mock_open_file):
+    mock_os_path_exists.return_value = True
+    mock_open_file.return_value.read.return_value = json.dumps({
+        'last_checked': datetime.now().isoformat(),
+        'latest_version': '2.0.0'
+    })
+
+    cache = load_cache()
+    assert cache['latest_version'] == '2.0.0'
+
+
+def test_load_cache_no_file(mock_os_path_exists):
+    mock_os_path_exists.return_value = False
+
+    cache = load_cache()
+    assert cache == {}
+
+
+def test_save_cache(mock_open_file):
+    data = {
+        'last_checked': datetime.now().isoformat(),
+        'latest_version': '2.0.0'
+    }
+
+    save_cache(data)
+    mock_open_file.return_value.write.assert_called_once_with(json.dumps(data))
+
+
+def test_check_for_updates_new_version(mocker, mock_datetime, mock_requests_get, mock_open_file):
+    # Set up mocks
+    last_checked_str = (datetime(2024, 1, 1) - CACHE_DURATION).isoformat()
+    mock_datetime.now.return_value = datetime(2024, 1, 2)
+    mock_datetime.fromisoformat.return_value = datetime.fromisoformat(last_checked_str)
+    mock_open_file.return_value.read.return_value = json.dumps({
+        'last_checked': last_checked_str,
+        'latest_version': '1.0.0'
+    })
+    mock_requests_get.return_value.raise_for_status.return_value = None
+    mock_requests_get.return_value.json.return_value = {'info': {'version': '2.0.0'}}
+
+    # Capture the print statements
+    with patch('builtins.print') as mocked_print:
+        check_for_updates()
+        assert mocked_print.call_count == 3
+
+
+def test_check_for_updates_no_new_version(mocker, mock_datetime, mock_requests_get, mock_open_file):
+    # Set up mocks
+    last_checked_str = (datetime(2024, 1, 1) - CACHE_DURATION).isoformat()
+    mock_datetime.now.return_value = datetime(2024, 1, 2)
+    mock_datetime.fromisoformat.return_value = datetime.fromisoformat(last_checked_str)
+    mock_open_file.return_value.read.return_value = json.dumps({
+        'last_checked': (datetime(2024, 1, 1) - CACHE_DURATION).isoformat(),
+        'latest_version': __version__
+    })
+    mock_requests_get.return_value.raise_for_status.return_value = None
+    mock_requests_get.return_value.json.return_value = {'info': {'version': __version__}}
+
+    # Capture the print statements
+    with patch('builtins.print') as mocked_print:
+        check_for_updates()
+        assert mocked_print.call_count == 0
+
+
+def test_check_for_updates_cache_valid(mock_datetime, mock_open_file):
+    # Set up mocks
+    last_checked_str = datetime(2024, 1, 2).isoformat()
+    mock_datetime.now.return_value = datetime(2024, 1, 2)
+    mock_datetime.fromisoformat.return_value = datetime.fromisoformat(last_checked_str)
+    mock_open_file.return_value.read.return_value = json.dumps({
+        'last_checked': last_checked_str,
+        'latest_version': __version__
+    })
+
+    # Capture the print statements
+    with patch('builtins.print') as mocked_print:
+        check_for_updates()
+        assert mocked_print.call_count == 0

gpt-pr-0.4.1/gptpr/version.py

@@ -0,0 +1 @@
+__version__ = "0.4.1"

gpt-pr-0.3.0/gptpr/version.py

@@ -1 +0,0 @@
-__version__ = "0.3.0"