7d47c79898
Message widgets are now inserted into the listView through the `rowsInserted` callback of the messages model. Messages are processed in the GetMessagesTask and GetApplicationMessagesTask when fetching multiple new messages. Single new incoming messages are processed in ProcessMessageTask.
98 lines
2.9 KiB
Python
98 lines
2.9 KiB
Python
import os
|
|
import platform
|
|
import re
|
|
import subprocess
|
|
|
|
from pathlib import Path
|
|
from typing import Iterator, List, Optional
|
|
|
|
from gotify_tray import gotify
|
|
from gotify_tray.database import Downloader
|
|
|
|
|
|
def verify_server(force_new: bool = False, enable_import: bool = True) -> bool:
|
|
from gotify_tray.gui import ServerInfoDialog
|
|
from gotify_tray.database import Settings
|
|
|
|
settings = Settings("gotify-tray")
|
|
|
|
url = settings.value("Server/url", type=str)
|
|
token = settings.value("Server/client_token", type=str)
|
|
|
|
if not url or not token or force_new:
|
|
dialog = ServerInfoDialog(url, token, enable_import)
|
|
if dialog.exec():
|
|
settings.setValue("Server/url", dialog.line_url.text())
|
|
settings.setValue("Server/client_token", dialog.line_token.text())
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return True
|
|
|
|
|
|
def process_messages(messages: List[gotify.GotifyMessageModel]) -> Iterator[gotify.GotifyMessageModel]:
|
|
downloader = Downloader()
|
|
for message in messages:
|
|
if image_url := get_image(message.message):
|
|
downloader.get_filename(image_url)
|
|
yield message
|
|
|
|
|
|
def convert_links(text):
|
|
_link = re.compile(
|
|
r'(?:(https://|http://)|(www\.))(\S+\b/?)([!"#$%&\'()*+,\-./:;<=>?@[\\\]^_`{|}~]*)(\s|$)',
|
|
re.I,
|
|
)
|
|
|
|
def replace(match):
|
|
groups = match.groups()
|
|
protocol = groups[0] or "" # may be None
|
|
www_lead = groups[1] or "" # may be None
|
|
return '<a href="http://{1}{2}" rel="nofollow">{0}{1}{2}</a>{3}{4}'.format(
|
|
protocol, www_lead, *groups[2:]
|
|
)
|
|
|
|
return _link.sub(replace, text)
|
|
|
|
|
|
def get_image(s: str) -> Optional[str]:
|
|
"""If `s` contains only an image URL, this function returns that URL.
|
|
This function also extracts a URL in the `` markdown image format.
|
|
"""
|
|
s = s.strip()
|
|
|
|
# Return True if 's' is a url and has an image extension
|
|
RE = r'(?:(https://|http://)|(www\.))(\S+\b/?)([!"#$%&\'()*+,\-./:;<=>?@[\\\]^_`{|}~]*).(jpg|jpeg|png|bmp|gif)(\s|$)'
|
|
if re.compile(RE, re.I).fullmatch(s) is not None:
|
|
return s
|
|
|
|
# Return True if 's' has the markdown image format
|
|
RE = r'!\[[^\]]*\]\((.*?)\s*("(?:.*[^"])")?\s*\)'
|
|
if re.compile(RE, re.I).fullmatch(s) is not None:
|
|
return re.compile(RE, re.I).findall(s)[0][0]
|
|
|
|
return None
|
|
|
|
|
|
def get_abs_path(s) -> str:
|
|
h = Path(__file__).parent.parent
|
|
p = Path(s)
|
|
return os.path.join(h, p).replace("\\", "/")
|
|
|
|
|
|
def open_file(filename: str):
|
|
if platform.system() == "Linux":
|
|
subprocess.call(["xdg-open", filename])
|
|
elif platform.system() == "Windows":
|
|
os.startfile(filename)
|
|
elif platform.system() == "Darwin":
|
|
subprocess.call(["open", filename])
|
|
|
|
|
|
def get_icon(name: str) -> str:
|
|
if platform.system() == "Darwin":
|
|
name += "-macos"
|
|
|
|
return get_abs_path(f"gotify_tray/gui/images/{name}.png")
|