openwebrxplus/owrx/controllers/file.py

51 lines
1.7 KiB
Python

from owrx.controllers.template import WebpageController
from owrx.controllers.assets import AssetsController
from owrx.storage import Storage
import re
class FileController(AssetsController):
def getFilePath(self, file):
return Storage.getFilePath(file)
class FilesController(WebpageController):
def template_variables(self):
isimg = re.compile(r'.*\.(png|bmp|gif|jpg)$')
issnd = re.compile(r'.*\.(mp3|wav)$')
files = Storage.getSharedInstance().getStoredFiles()
rows = ""
for i in range(len(files)):
# Start a row
if i % 3 == 0:
rows += '<tr>\n'
# Show images as they are, show document icon for the rest
if isimg.match(files[i]):
shot = "/files/" + files[i]
elif issnd.match(files[i]):
shot = "static/gfx/audio-file.png"
else:
shot = "static/gfx/text-file.png"
# Print out individual tiles
rows += ('<td class="file-tile">' +
('<a href="/files/%s" download="%s">' % (files[i], files[i])) +
('<img src="%s" download="%s">' % (shot, files[i])) +
('<p class="file-title">%s</p>' % files[i]) +
'</a></td>\n')
# Finish a row
if i % 3 == 2:
rows += '</tr>\n'
# Finish final row
if len(files) > 0 and len(files) % 3 != 0:
rows += '</tr>\n'
variables = super().template_variables()
variables["rows"] = rows
return variables
def indexAction(self):
self.serve_template("files.html", **self.template_variables())