From 741147015fe440c9b047bbfe4b3e7bec1beb6e1c Mon Sep 17 00:00:00 2001
From: dave <dave@davepedu.com>
Date: Sat, 17 Aug 2019 10:50:31 -0700
Subject: [PATCH] cache layer

---
 mediaweb/__init__.py | 62 +++++++++++++++++++++++++++++++++++++++-----
 templates/index.html |  8 +++++-
 2 files changed, 63 insertions(+), 7 deletions(-)

diff --git a/mediaweb/__init__.py b/mediaweb/__init__.py
index 080d688..2473d10 100644
--- a/mediaweb/__init__.py
+++ b/mediaweb/__init__.py
@@ -4,11 +4,55 @@ import logging
 from jinja2 import Environment, FileSystemLoader, select_autoescape
 from deluge_client import DelugeRPCClient
 from urllib.parse import urlparse
+from pprint import pprint
+from threading import Thread
+from time import sleep
+from queue import Queue
+from dataclasses import dataclass, field
 
 
 APPROOT = os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))
 
 
+@dataclass
+class Cache:
+    torrents: dict = field(default_factory=dict)
+
+
+class ClientCache(object):
+    def __init__(self, client):
+        self.client = client
+        self.data = Cache()
+        self.q = Queue()
+
+        self.background_t = Thread(target=self.background, daemon=True)
+        self.background_t.start()
+
+        self.timer_t = Thread(target=self.timer, daemon=True)
+        self.timer_t.start()
+
+    def refresh(self):
+        self.q.put(None)
+
+    def background(self):
+        while True:
+            self.q.get()  # block until we need to do something
+            logging.info("performing background tasks...")
+
+            self.data.torrents = self.client.core.get_torrents_status({"label": "sickrage"},
+                                                                      ['name', 'label', 'save_path', 'is_seed',
+                                                                       'is_finished', 'progress'])
+
+            self.q.task_done()
+            logging.info("background tasks complete")
+
+    def timer(self):
+        while True:
+            self.refresh()
+            logging.info("sleeping...")
+            sleep(300)  # TODO configurable task interval
+
+
 class MediaWeb(object):
     def __init__(self, rpc, templater, uioptions):
         self.tpl = templater
@@ -25,17 +69,21 @@ class MediaWeb(object):
         return {}
 
     @cherrypy.expose
-    def index(self):
-        torrents = self.rpc.core.get_torrents_status({"label": "sickrage"}, ['name', 'label', 'save_path', 'is_seed', 'is_finished', 'progress'])
-        return self.render("index.html", torrents=torrents)
+    def index(self, action=None):
+        if action:
+            if action == "update":
+                self.rpc.refresh()
+            raise cherrypy.HTTPRedirect("/")
+        return self.render("index.html", torrents=self.rpc.data.torrents)
 
     @cherrypy.expose
     def move(self, thash, dest=None, otherdest=None):
-        torrent = self.rpc.core.get_torrent_status(thash, [])  # TODO reduce to needed fields
+        torrent = self.rpc.client.core.get_torrent_status(thash, [])  # TODO reduce to needed fields
 
         if cherrypy.request.method == "POST" and (dest or otherdest):
             target = otherdest or dest
-            self.rpc.core.move_storage([thash], target)
+            self.rpc.client.core.move_storage([thash], target)
+            self.rpc.refresh()
             raise cherrypy.HTTPRedirect("/")
 
         return self.render("moveform.html", torrent=torrent)
@@ -83,7 +131,9 @@ def main():
     assert uri.scheme == "deluge"
     rpc = DelugeRPCClient(uri.hostname, uri.port if uri.port else 58846, uri.username, uri.password, decode_utf8=True)
 
-    web = MediaWeb(rpc, tpl, uioptions)
+    rpc_cache = ClientCache(rpc)
+
+    web = MediaWeb(rpc_cache, tpl, uioptions)
     cherrypy.tree.mount(web, '/', {'/': {'tools.auth_basic.on': True,
                                          'tools.auth_basic.realm': 'mediaweb',
                                          'tools.auth_basic.checkpassword': validate_password, }})
diff --git a/templates/index.html b/templates/index.html
index fb26960..bff82ea 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -2,6 +2,12 @@
 {% block body %}
 
 <div class="torrents">
+    <div class="toolbar">
+        <form action="/" method="post">
+            <input name="action" type="submit" value="refresh">
+            <input name="action" type="submit" value="update">
+        </form>
+    </div>
     <h2>Completed</h2>
     <table>
         <tr>
@@ -32,7 +38,7 @@
             <th>hash</th>
             <th>name</th>
             <th>path</th>
-            <th>percent</th>
+            <th>progress</th>
             <th>actions</th>
         </tr>
     {% for torid, tor in torrents.items() %}{% if not tor.is_finished %}