Add database artifact logic to dev action and main workflow

Author: flufflycthu1u

.github/workflows/dev.yml

@@ -7,6 +7,9 @@ on:
       - '!master'
   workflow_dispatch:
 
+permissions:
+      actions: read
+
 jobs:
   run-cute-pets:
     runs-on: ubuntu-latest
@@ -21,6 +24,27 @@ jobs:
       - name: Install dependencies
         run: pip install --break-system-packages -r requirements.txt
 
+      - name: Get Previous Run ID
+        id: get_id
+        env:
+          GH_TOKEN: ${{ secrets.GITHUB_TOKEN }}
+        run: |
+          # Fetches the ID of the last completed run for the current workflow
+          PREVIOUS_RUN_ID=$(gh run list --workflow "${{ github.workflow }}" \
+          --status success \
+          --limit 1 \
+          --json databaseId \
+          --jq '.[0].databaseId')
+          echo "previous_run_id=$PREVIOUS_RUN_ID" >> "$GITHUB_OUTPUT"
+
+      - name: Download previous database artifact
+        uses: actions/download-artifact@v8
+        with:
+          name: database.json
+          github-token: ${{ secrets.GITHUB_TOKEN }}
+          run-id: ${{ steps.get_id.outputs.previous_run_id }}
+        continue-on-error: true 
+
       - name: Call RescueGroups API
         env:
           CUTEPETSBOSTON_RESCUEGROUPS_API_KEY: ${{ secrets.CUTEPETSBOSTON_RESCUEGROUPS_API_KEY }}
@@ -31,3 +55,10 @@ jobs:
         run: |
           #In order to create posts on the test accounts remove the --debugposters debug flag
           python ./main.py --debugsources --debugposters
+
+      - name: Upload database artifact
+        uses: actions/upload-artifact@v7
+        with:
+          path: database.json
+          retention-days: 14
+          archive: false

main.py

@@ -1,5 +1,10 @@
 import random
 import argparse
+import json
+import sys
+import traceback
+from pathlib import Path
+from datetime import datetime, timezone, timedelta
 
 def main():
     parser = argparse.ArgumentParser()
@@ -74,10 +79,39 @@ def run(sources, posters):
 
 
 def pick_pet(pets):
-    eligible = [pet for pet in pets if pet.image_url and pet.adoption_url]
-    if not eligible:
-        return None
-    return random.choice(eligible)
+    Path("database.json").touch(exist_ok=True)
+    # Open file
+    with open("database.json", "r+") as f:
+        # Load json
+        try:
+            data = json.load(f)
+        except (json.JSONDecodeError, ValueError) as e:
+            print(f"{type(e).__name__}:{e}", file=sys.stderr)
+            traceback.print_exc()
+            data = {}
+        
+        if "posted_pets" in data:
+            posted_pet_ids =  {pet.pet_id for pet in data["posted_pets"]}
+        else:
+            pet_ids = {}
+            data["posted_pets"] = []
+        # Check pet has an image, adoption url, and has not been posted
+        eligible = [pet for pet in pets if pet.image_url and pet.adoption_url and pet.pet_id not in posted_pet_ids]
+        if not eligible:
+            return None
+        
+        selected_pet = random.choice(eligible)
+        # Add pet ID to list of posted pets
+        data["posted_pets"].append({"name": selected_pet.name, "pet_id": selected_pet.pet_id, "time": datetime.now(timezone.utc).isoformat()})
+        # Remove old pets
+        cutoff = datetime.now() - timedelta(weeks=12)
+        new_pets = [item for item in data["posted_pets"] if datetime.fromisoformat(item['time']) > cutoff]
+        data["posted_pets"] = new_pets
+        # Export json
+        f.seek(0)
+        json.dump(data, f)
+        f.truncate()
+        return selected_pet
 
 
 if __name__ == "__main__":