PR2: Add new preference and AnnouncementChecker class

.github/workflows/website.yml

@@ -9,6 +9,7 @@ on:
       - main
       - develop
       - shrivaths/changelog-announcement-1
+      - shrivaths/changelog-announcement-2
     paths:
       - "docs/**"
       - "README.rst"
@@ -48,6 +49,7 @@ jobs:
         run: |
           cd docs
           python make_api_doctree.py
+          python make_bulletin_json.py
           make html
 
       - name: Deploy (sleap.ai)

docs/make_bulletin_json.py

@@ -0,0 +1,43 @@
+import json
+import os
+
+# Set the file paths
+input_md_file = os.path.join(os.path.dirname(__file__), "bulletin.md")
+output_json_file = os.path.join(os.path.dirname(__file__), "bulletin.json")
+
+
+def generate_json_file():
+    with open(input_md_file, "r", encoding="utf-8") as md_file:
+        markdown_content = md_file.read()
+    bulletin_json = []
+    content = ""
+
+    # Initialize title and date with default values
+    title = "DEFAULT_TITLE"
+    date = "DEFAULT_DATE"
+
+    for line in markdown_content.split("\n"):
+        # Skip if the line begins with #
+        if line.startswith("# "):
+            continue
+        elif line.startswith("---"):
+            bulletin_json.append({"title": title, "date": date, "content": content})
+            content = ""
+            # Reset title and date to their default values after each section
+            title = "DEFAULT_TITLE"
+            date = "DEFAULT_DATE"
+        elif line.startswith("## "):
+            title = line[3:].strip()
+        elif line.startswith("_"):
+            date = line[1 : len(line) - 1].strip()
+        else:
+            content += line + "\n"
+    # Append last section
+    bulletin_json.append({"title": title, "date": date, "content": content})
+
+    with open("bulletin.json", "w") as f:
+        f.write(repr(bulletin_json))
+
+
+if __name__ == "__main__":
+    generate_json_file()

sleap/gui/app.py

@@ -69,7 +69,7 @@
 from sleap.gui.overlays.tracks import TrackListOverlay, TrackTrailOverlay
 from sleap.gui.shortcuts import Shortcuts
 from sleap.gui.state import GuiState
-from sleap.gui.web import ReleaseChecker, ping_analytics
+from sleap.gui.web import ReleaseChecker, AnnouncementChecker, ping_analytics
 from sleap.gui.widgets.docks import (
     InstancesDock,
     SkeletonDock,
@@ -158,6 +158,8 @@ def __init__(
         self.state["share usage data"] = prefs["share usage data"]
         self.state["skeleton_preview_image"] = None
         self.state["skeleton_description"] = "No skeleton loaded yet"
+        self.state["announcement last seen date"] = prefs["announcement last seen date"]
+        self.state["announcement"] = prefs["announcement"]
         if no_usage_data:
             self.state["share usage data"] = False
         self.state["clipboard_track"] = None
@@ -168,6 +170,7 @@ def __init__(
         self.state.connect("show non-visible nodes", self.plotFrame)
 
         self.release_checker = ReleaseChecker()
+        self.announcement_checker = AnnouncementChecker(state=self.state)
 
         if self.state["share usage data"]:
             ping_analytics()
@@ -223,6 +226,8 @@ def closeEvent(self, event):
         prefs["color predicted"] = self.state["color predicted"]
         prefs["trail shade"] = self.state["trail_shade"]
         prefs["share usage data"] = self.state["share usage data"]
+        prefs["announcement last seen date"] = self.state["announcement last seen date"]
+        prefs["announcement"] = self.state["announcement"]
 
         # Save preferences.
         prefs.save()

sleap/gui/web.py

@@ -3,11 +3,16 @@
 import attr
 import pandas as pd
 import requests
-from typing import List, Dict, Any
+from typing import List, Dict, Any, Optional, Tuple
+import json
+import os
+from datetime import datetime
 
 
 REPO_ID = "talmolab/sleap"
 ANALYTICS_ENDPOINT = "https://analytics.sleap.ai/ping"
+BASE_DIR = os.path.dirname(os.path.abspath(os.path.join(__file__, os.path.pardir)))
+BULLETIN_JSON = os.path.join(BASE_DIR, "..", "docs", "bulletin.json")
 
 
 @attr.s(auto_attribs=True)
@@ -146,6 +151,64 @@ def get_release(self, version: str) -> Release:
         )
 
 
+@attr.s(auto_attribs=True)
+class AnnouncementChecker:
+    """Checker for new announcements on the bulletin page of sleap."""
+
+    state: "GuiState"
+    bulletin_json_path: str = BULLETIN_JSON
+    _previous_announcement_date: str = None
+    _latest_data: Optional[Dict[str, str]] = None
+
+    @property
+    def previous_announcement_date(self):
+        _previous_announcement_date = self.state["announcement last seen date"]
+        return _previous_announcement_date
+
+    def _read_bulletin_data(self) -> Optional[Dict]:
+        """Reads the bulletin data from the JSON file."""
+        try:
+            with open(self.bulletin_json_path, "r") as jsf:
+                data = json.load(jsf)
+                self._latest_data = data[0]
+        except FileNotFoundError:
+            self._latest_data = None
+
+    def new_announcement_available(self) -> bool:
+        """Check if latest announcement is available."""
+        self._read_bulletin_data()
+        if self.previous_announcement_date and self._latest_data:
+            latest_date = datetime.strptime(self._latest_data["date"], "%m/%d/%Y")
+            previous_date = datetime.strptime(
+                self.previous_announcement_date, "%m/%d/%Y"
+            )
+            if latest_date > previous_date:
+                return True
+            else:
+                return False
+        else:
+            return True
+
+    def get_latest_announcement(self) -> Optional[Tuple[str, str, str]]:
+        """Return latest announcements on the releases page not seen by user."""
+        if self.new_announcement_available():
+            return (
+                self._latest_data["title"],
+                self._latest_data["date"],
+                self._latest_data["content"],
+            )
+        return None
+
+    def update_announcement(self):
+        """Update the last seen date of announcement in preferences."""
+        announcement = self.get_latest_announcement()
+        if announcement is None:
+            return
+        self.state["announcement last seen date"] = announcement[1]
+        new_announcement = "\n".join(announcement[2].split("\n"))
+        self.state["announcement"] = "## " + announcement[0] + "\n" + new_announcement
+
+
 def get_analytics_data() -> Dict[str, Any]:
     """Gather data to be transmitted to analytics backend."""
     import os

sleap/prefs.py

@@ -5,6 +5,7 @@
 """
 
 from sleap import util
+from datetime import date
 
 
 class Preferences(object):
@@ -28,6 +29,8 @@ class Preferences(object):
         "node label size": 12,
         "show non-visible nodes": True,
         "share usage data": True,
+        "announcement last seen date": False,
+        "announcement": False,
     }
     _filename = "preferences.yaml"
 

tests/conftest.py

@@ -11,3 +11,4 @@
 from tests.fixtures.datasets import *
 from tests.fixtures.videos import *
 from tests.fixtures.models import *
+from tests.fixtures.announcements import *

tests/data/announcement_checker_bulletin/test_bulletin.json

@@ -0,0 +1 @@
+[{"title": "title1", "date": "10/09/2023", "content": "New announcement"}, {"title": "title2", "date": "10/07/2023", "content": "Old Announcment"}]

tests/fixtures/announcements.py

@@ -0,0 +1,8 @@
+import pytest
+
+TEST_BULLETIN_JSON = "tests/data/announcement_checker_bulletin/test_bulletin.json"
+
+
+@pytest.fixture
+def bulletin_json_path():
+    return TEST_BULLETIN_JSON

tests/gui/test_web.py

@@ -1,6 +1,16 @@
 import pandas as pd
-from sleap.gui.web import ReleaseChecker, Release, get_analytics_data, ping_analytics
+from sleap.gui.web import (
+    ReleaseChecker,
+    Release,
+    AnnouncementChecker,
+    get_analytics_data,
+    ping_analytics,
+)
 import pytest
+import json
+import os
+from sleap.gui.commands import CommandContext
+from sleap.io.dataset import Labels
 
 
 def test_release_from_json():
@@ -72,6 +82,49 @@ def test_release_checker():
     assert checker.releases[1] != rls_test
 
 
+def test_announcementchecker(bulletin_json_path):
+
+    labels = Labels()
+    context = CommandContext.from_labels(labels=labels)
+    context.state = {}
+    context.state["announcement last seen date"] = "10/10/2023"
+    checker = AnnouncementChecker(
+        state=context.state, bulletin_json_path=bulletin_json_path
+    )
+
+    # Check if the announcement checker gets the correct date from the app
+    assert checker.previous_announcement_date == "10/10/2023"
+
+    # Create dummy JSON file to check
+    bulletin_data = [
+        {"title": "title1", "date": "10/12/2023", "content": "New announcement"},
+        {"title": "title2", "date": "10/07/2023", "content": "Old Announcment"},
+    ]
+    with open(bulletin_json_path, "w") as test_file:
+        json.dump(bulletin_data, test_file)
+
+    # Check if latest announcement is fetched
+    announcement = checker.get_latest_announcement()
+    assert announcement == ("title1", "10/12/2023", "New announcement")
+
+    # Check if announcement is updated
+    checker.update_announcement()
+    assert context.state["announcement last seen date"] == "10/12/2023"
+    assert context.state["announcement"] == "## title1\nNew announcement"
+
+    # Create dummy JSON file
+    bulletin_data = [
+        {"title": "title1", "date": "10/09/2023", "content": "New announcement"},
+        {"title": "title2", "date": "10/07/2023", "content": "Old Announcment"},
+    ]
+    with open(bulletin_json_path, "w") as test_file:
+        json.dump(bulletin_data, test_file)
+
+    # Check to ensure no new announcement is created
+    announcement = checker.get_latest_announcement()
+    assert announcement == None
+
+
 def test_get_analytics_data():
     analytics_data = get_analytics_data()
     assert "platform" in analytics_data