openwebrx-clone/owrx/controllers/settings/bookmarks.py

122 lines
4.6 KiB
Python
Raw Normal View History

from owrx.controllers.template import WebpageController
from owrx.controllers.admin import AuthorizationMixin
2021-02-13 17:35:15 +00:00
from owrx.bookmarks import Bookmark, Bookmarks
2021-02-13 22:53:16 +00:00
from owrx.modes import Modes
2021-02-13 23:41:03 +00:00
import json
import logging
logger = logging.getLogger(__name__)
class BookmarksController(AuthorizationMixin, WebpageController):
2021-02-13 17:35:15 +00:00
def template_variables(self):
variables = super().template_variables()
variables["bookmarks"] = self.render_table()
return variables
def render_table(self):
bookmarks = Bookmarks.getSharedInstance()
2021-02-13 22:53:16 +00:00
def render_mode(m):
return """
<option value={mode}>{name}</option>
""".format(
mode=m.modulation,
name=m.name,
)
2021-02-13 17:35:15 +00:00
return """
2021-02-14 15:21:09 +00:00
<table class="table">
2021-02-13 17:35:15 +00:00
<tr>
<th>Name</th>
<th class="frequency">Frequency</th>
<th>Modulation</th>
2021-02-13 17:41:42 +00:00
<th>Actions</th>
2021-02-13 17:35:15 +00:00
</tr>
{bookmarks}
2021-02-13 22:53:16 +00:00
<tr class="inputs" style="display:none;">
<td><input class="form-control form-control-sm" type="text" name="name"></td>
<td><input class="form-control form-control-sm" type="number" step="1" name="frequency"></td>
<td><select class="form-control form-control-sm" name="modulation">{options}</select></td>
<td></td>
</tr>
2021-02-13 17:35:15 +00:00
</table>
""".format(
2021-02-14 13:48:32 +00:00
bookmarks="".join(self.render_bookmark(b) for b in bookmarks.getBookmarks()),
2021-02-13 22:53:16 +00:00
options="".join(render_mode(m) for m in Modes.getAvailableModes()),
2021-02-13 17:35:15 +00:00
)
2021-02-14 13:48:32 +00:00
def render_bookmark(self, bookmark: Bookmark):
2021-02-13 22:53:16 +00:00
mode = Modes.findByModulation(bookmark.getModulation())
2021-02-13 17:35:15 +00:00
return """
2021-02-14 13:48:32 +00:00
<tr data-id="{id}">
2021-02-13 17:35:15 +00:00
<td>{name}</td>
<td class="frequency">{frequency}</td>
2021-02-13 22:53:16 +00:00
<td data-value="{modulation}">{modulation_name}</td>
2021-02-13 17:41:42 +00:00
<td>
2021-02-14 15:51:16 +00:00
<button class="btn btn-sm btn-danger bookmark-delete">delete</button>
2021-02-13 17:41:42 +00:00
</td>
2021-02-13 17:35:15 +00:00
</tr>
""".format(
2021-02-13 17:41:42 +00:00
id=id(bookmark),
2021-02-13 17:35:15 +00:00
name=bookmark.getName(),
frequency=bookmark.getFrequency(),
2021-02-13 22:53:16 +00:00
modulation=bookmark.getModulation() if mode is None else mode.modulation,
modulation_name=bookmark.getModulation() if mode is None else mode.name,
2021-02-13 17:35:15 +00:00
)
2021-02-13 23:41:03 +00:00
def _findBookmark(self, bookmark_id):
bookmarks = Bookmarks.getSharedInstance()
try:
return next(b for b in bookmarks.getBookmarks() if id(b) == bookmark_id)
except StopIteration:
return None
def update(self):
bookmark_id = int(self.request.matches.group(1))
bookmark = self._findBookmark(bookmark_id)
if bookmark is None:
self.send_response("{}", content_type="application/json", code=404)
return
try:
data = json.loads(self.get_body())
for key in ["name", "frequency", "modulation"]:
if key in data:
value = data[key]
if key == "frequency":
value = int(value)
setattr(bookmark, key, value)
Bookmarks.getSharedInstance().store()
self.send_response("{}", content_type="application/json", code=200)
except json.JSONDecodeError:
self.send_response("{}", content_type="application/json", code=400)
2021-02-14 15:21:09 +00:00
def new(self):
bookmarks = Bookmarks.getSharedInstance()
try:
data = json.loads(self.get_body())
# sanitize
data = {k: data[k] for k in ["name", "frequency", "modulation"]}
bookmark = Bookmark(data)
bookmarks.addBookmark(bookmark)
bookmarks.store()
self.send_response(json.dumps({"bookmark_id": id(bookmark)}), content_type="application/json", code=200)
except json.JSONDecodeError:
self.send_response("{}", content_type="application/json", code=400)
2021-02-14 15:51:16 +00:00
def delete(self):
bookmark_id = int(self.request.matches.group(1))
bookmark = self._findBookmark(bookmark_id)
if bookmark is None:
self.send_response("{}", content_type="application/json", code=404)
return
bookmarks = Bookmarks.getSharedInstance()
bookmarks.removeBookmark(bookmark)
bookmarks.store()
self.send_response("{}", content_type="application/json", code=200)
def indexAction(self):
self.serve_template("settings/bookmarks.html", **self.template_variables())