PR2: Add new preference and AnnouncementChecker class

docs/make_bulletin_json.py

@@ -0,0 +1,40 @@
+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"):
+        if 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(output_json_file, "w", encoding="utf-8") as json_file:
+        json.dump(bulletin_json, json_file, ensure_ascii=False, indent=4)
+
+
+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(app=self)
 
         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,17 @@
 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
 
 
 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, os.pardir))
+)
+BULLETIN_JSON = os.path.join(BASE_DIR, "..", "docs", "bulletin.json")
 
 
 @attr.s(auto_attribs=True)
@@ -146,6 +152,54 @@ def get_release(self, version: str) -> Release:
         )
 
 
+@attr.s(auto_attribs=True)
+class AnnouncementChecker:
+    """Checker for new announcements on the bulletin page of sleap."""
+
+    app: "MainWindow"
+    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.app.state["announcement last seen date"]
+        return _previous_announcement_date
+
+    def _read_bulletin_data(self):
+        """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 = {}
+
+    @property
+    def new_announcement(self) -> bool:
+        self._read_bulletin_data()
+        if (
+            self._latest_data
+            and self._latest_data["date"] != self.previous_announcement_date
+        ):
+            return True
+        return False
+
+    def get_latest_announcement(self) -> Optional[Tuple[str, str]]:
+        """Return latest announcements on the releases page not seen by user."""
+        if self.new_announcement:
+            return (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.app.state["announcement last seen date"] = announcement[0]
+        self.app.state["announcement"] = announcement[1]
+
+
 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": None,
+        "announcement": None,
     }
     _filename = "preferences.yaml"
 

tests/gui/test_web.py

@@ -1,6 +1,15 @@
 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
+from sleap.gui.app import create_app
+import json
+import os
 
 
 def test_release_from_json():
@@ -72,6 +81,37 @@ def test_release_checker():
     assert checker.releases[1] != rls_test
 
 
+def test_announcementchecker():
+
+    BULLETIN_JSON_PATH = "D:\\TalmoLab\\sleap\\tests\\data\\announcement_checker_bulletin\\test_bulletin.json"
+    app = create_app()
+    app.state = {}
+    app.state["announcement last seen date"] = "10/10/2023"
+    checker = AnnouncementChecker(app=app, 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 == ("10/12/2023", "New announcement")
+
+    checker.update_announcement()
+    assert app.state["announcement last seen date"] == "10/12/2023"
+    assert app.state["announcement"] == "New announcement"
+
+    # Delete the JSON file
+    os.remove(BULLETIN_JSON_PATH)
+
+
 def test_get_analytics_data():
     analytics_data = get_analytics_data()
     assert "platform" in analytics_data